import { computed, onBeforeUnmount, onMounted } from 'vue';
import { getWebsocketUrl } from '@/utils/websocket/config'
import { useConnectWebSocket } from '@/utils/websocket/connect-websocket'
import { useStore } from 'vuex'
import dayjs from 'dayjs'
import { aLinkDownloadUtil } from '@/utils/util'
import EventBus from '@/utils/eventBus'
import { getStore, setStore } from '@/utils/store'

export const useGlobalWS = () => {
	const store = useStore()
	const loginUserInfo = computed(() => store.state.user.userInfo)
	const downloadProgress = computed(() => store.state.common.downloadProgress)
	let currentWebSocket = null

	// 下载进度处理
	function downloadProcessing(payload) {
		const {type,status,progress,download_url} = payload?.data || {}
		let setProgress = 0
		if (status === 'PENDING'){
			setProgress = 1
		}else if(status === 'CANCELLED'){
			setProgress = 100
		}else if(status !== 'COMPLETED' && progress === 100){
			setProgress = 99
		}else{
			setProgress = progress
		}
		store.commit('setDownloadProgress', {
			...downloadProgress.value,
			[type]: setProgress,
		})
		if (status === 'COMPLETED' && ['htsjzx','htlsrwxq'].includes(type)) {
			const name = `数据中心-${dayjs().format('YYYYMMDDHHmmss')}.zip`
			const currentUrl = getStore({name: 'downloadUrl'})
			if (currentUrl === download_url && download_url!== undefined) return
			aLinkDownloadUtil(download_url, name)
			setStore({ name: 'downloadUrl', content: download_url })
		}
	}

	const logOut = () => {
		store.commit('SET_THEME_NAME', '')
		store.dispatch('LogOut').then(() => {
			router.push({ path: '/login' })
			setTimeout(() => location.reload())
		})
	}
	function messageHandler(payload) {
		switch (payload.biz_code) {
			case 'JOB_ISREFRESH':
				store.commit('jobUpdateKeyAdd')
				break
			case 'DEVICE_ISREFRESH':
				store.commit('deviceUpdateKeyAdd')
				break
			case 'DOWNLOAD_PROGRESS':
				downloadProcessing(payload)
				break
			case 'LOGOUT_USER':
				logOut()
				break
		}
	}

	onMounted(() => {
		let webSocketUrl = getWebsocketUrl() + `&model_type=3&workspace-id=${loginUserInfo.value.user_id}`
		currentWebSocket = useConnectWebSocket((result)=>{
			const payload = JSON.parse(result)
			messageHandler(payload)
		}, webSocketUrl)
		EventBus.on('useGlobalWS-messageHandler',messageHandler)
	})

	onBeforeUnmount(() => {
		currentWebSocket?.close()
		EventBus.off('useGlobalWS-messageHandler',messageHandler)
	})
	return {}
}