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 { computed, onBeforeUnmount, onMounted } from 'vue';
import { setStore,getStore } from '@/utils/store';
import { getDownloadStatusApi } from '@/api/dataCenter/dataCenter';

export const useDownloadWs = () => {
	const store = useStore()
	const loginUserInfo = computed(() => store.state.user.userInfo)
	const downloadProgress = computed(() => store.state.common.downloadProgress)
	let currentWebSocket = null
	function messageHandler(payload) {
		const {type,status,progress,download_url} = payload
		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 })
		}
	}

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

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