192 lines
		
	
	
		
			8.2 KiB
		
	
	
	
		
			HTML
		
	
	
	
	
	
			
		
		
	
	
			192 lines
		
	
	
		
			8.2 KiB
		
	
	
	
		
			HTML
		
	
	
	
	
	
<!DOCTYPE html>
 | 
						|
<html lang="en">
 | 
						|
 | 
						|
<head>
 | 
						|
    <meta charset="UTF-8">
 | 
						|
    <title>webRTC</title>
 | 
						|
</head>
 | 
						|
 | 
						|
<body>
 | 
						|
    <div>
 | 
						|
        <h1>webRTC</h1>
 | 
						|
        <p>选择音乐使频道内所有设备同步播放 chrome://webrtc-internals/</p>
 | 
						|
    </div>
 | 
						|
    <script type="module">
 | 
						|
        import IndexedDB from './database.js'
 | 
						|
        import MusicList from './music.js'
 | 
						|
        import ClientList from './client.js'
 | 
						|
 | 
						|
        // 读取本地音乐列表(本地缓存)
 | 
						|
        const database = new IndexedDB('musicDatabase', 1, 'musicObjectStore')
 | 
						|
        await database.open()
 | 
						|
        const list = await database.getAll()
 | 
						|
 | 
						|
        // 读取本地用户名(本地缓存)
 | 
						|
        const name = localStorage.getItem('username') ?? '游客'
 | 
						|
 | 
						|
        // 初始化客户端列表
 | 
						|
        const clientList = new ClientList({ name })
 | 
						|
 | 
						|
        // 初始化音乐列表(加入本地缓存)
 | 
						|
        const musicList = new MusicList({
 | 
						|
            list,
 | 
						|
            onplay: item => {
 | 
						|
                console.log('播放音乐', item.name)
 | 
						|
            },
 | 
						|
            onstop: item => {
 | 
						|
                console.log('停止音乐', item.name)
 | 
						|
            },
 | 
						|
            onlike: (item, list) => {
 | 
						|
                console.log('喜欢音乐', item)
 | 
						|
                if (item.arrayBuffer) {
 | 
						|
                    database.add(item)
 | 
						|
                    clientList.send('base', JSON.stringify({
 | 
						|
                        type: 'set_music_list',
 | 
						|
                        list: list.map(({ id, name, size, type }) => ({ id, name, size, type }))
 | 
						|
                    }))
 | 
						|
                }
 | 
						|
            },
 | 
						|
            onunlike: (item, list) => {
 | 
						|
                console.log('取消喜欢', item)
 | 
						|
                if (item.arrayBuffer) {
 | 
						|
                    database.delete(item.id)
 | 
						|
                    clientList.send('base', JSON.stringify({
 | 
						|
                        type: 'set_music_list',
 | 
						|
                        list: list.map(({ id, name, size, type }) => ({ id, name, size, type }))
 | 
						|
                    }))
 | 
						|
                }
 | 
						|
            },
 | 
						|
            onban: item => {
 | 
						|
                console.log('禁止音乐', item)
 | 
						|
            },
 | 
						|
            onunban: item => {
 | 
						|
                console.log('解禁音乐', item)
 | 
						|
            },
 | 
						|
            onremove: item => {
 | 
						|
                console.log('移除音乐', item)
 | 
						|
                database.delete(item.id)
 | 
						|
            },
 | 
						|
            onadd: (item, list) => {
 | 
						|
                console.log('添加音乐', item.name)
 | 
						|
            },
 | 
						|
            onupdate: item => {
 | 
						|
                console.log('更新音乐', item)
 | 
						|
                database.put(item)
 | 
						|
            },
 | 
						|
            onerror: error => {
 | 
						|
                console.log('音乐列表错误', error)
 | 
						|
            },
 | 
						|
            onload: async item => {
 | 
						|
                console.log('加载音乐', item)
 | 
						|
                return await new Promise((resolve) => {
 | 
						|
                    var buffer = new ArrayBuffer(0)
 | 
						|
                    var count = 0
 | 
						|
                    clientList.setChannel(`music-data-${item.id}`, {
 | 
						|
                        onmessage: async (event, client) => {
 | 
						|
                            buffer = appendBuffer(buffer, event.data)
 | 
						|
                            console.log('收到音乐数据 chunk', count, buffer.byteLength)
 | 
						|
                            count++
 | 
						|
                            if (buffer.byteLength >= item.size) {
 | 
						|
                                console.log('音乐数据接收完毕')
 | 
						|
                                item.arrayBuffer = buffer
 | 
						|
                                event.target.close() // 关闭信道
 | 
						|
                                resolve(item)
 | 
						|
                            }
 | 
						|
                        }
 | 
						|
                    })
 | 
						|
                    console.log('TODO: 重复请求改为指向目标主机(需要主机各自保存列表/收到列表时)')
 | 
						|
                    const client = clientList.clientlist.find(client => {
 | 
						|
                        return client.musicList.find(x => x.id === item.id)
 | 
						|
                    })
 | 
						|
                    clientList.sendto(client.id, 'base', JSON.stringify({
 | 
						|
                        type: 'get_music_data', id: item.id, channel: `music-data-${item.id}`
 | 
						|
                    }))
 | 
						|
                })
 | 
						|
            }
 | 
						|
        })
 | 
						|
 | 
						|
        // 缓冲分片发送
 | 
						|
        const CHUNK_SIZE = 1024 * 128 // 每个块的大小为128KB
 | 
						|
        const THRESHOLD = 1024 * 1024 // 缓冲区的阈值为1MB
 | 
						|
        const DELAY = 100             // 延迟500ms
 | 
						|
 | 
						|
        // 将两个ArrayBuffer合并成一个
 | 
						|
        function appendBuffer(buffer1, buffer2) {
 | 
						|
            const tmp = new Uint8Array(buffer1.byteLength + buffer2.byteLength)
 | 
						|
            tmp.set(new Uint8Array(buffer1), 0)
 | 
						|
            tmp.set(new Uint8Array(buffer2), buffer1.byteLength)
 | 
						|
            return tmp.buffer
 | 
						|
        }
 | 
						|
        // 只有一个基本信道, 用于交换和调度信息
 | 
						|
        clientList.setChannel('base', {
 | 
						|
            onopen: async event => {
 | 
						|
                console.log('打开信道', event.target.label)
 | 
						|
                // 要求对方发送音乐列表
 | 
						|
                clientList.send('base', JSON.stringify({ type: 'get_music_list' }))
 | 
						|
            },
 | 
						|
            onmessage: async (event, client) => {
 | 
						|
                const { type, id, channel, list } = JSON.parse(event.data)
 | 
						|
                if (type === 'get_music_list') {
 | 
						|
                    const ms = musicList.list.filter(item => item.arrayBuffer)
 | 
						|
                    console.log(client.name, '请求音乐列表:', ms)
 | 
						|
                    clientList.send('base', JSON.stringify({
 | 
						|
                        type: 'set_music_list',
 | 
						|
                        list: ms.map(({ id, name, size, type }) => ({ id, name, size, type }))
 | 
						|
                    }))
 | 
						|
                    return
 | 
						|
                }
 | 
						|
                if (type === 'set_music_list') {
 | 
						|
                    console.log(client.name, '发来音乐列表:', `x${JSON.parse(event.data).list.length}`)
 | 
						|
                    console.log('将列表保存到本机记录:', client)
 | 
						|
                    client.musicList = list
 | 
						|
                    list.forEach(item => musicList.add(item))
 | 
						|
                    return
 | 
						|
                }
 | 
						|
                if (type === 'get_music_data') {
 | 
						|
                    // 建立一个信道, 用于传输音乐数据(接收方已经准备好摘要信息)
 | 
						|
                    console.log(client.name, '建立一个信道, 用于传输音乐数据', musicList.list)
 | 
						|
                    musicList.list.filter(item => item.id === id).forEach(item => {
 | 
						|
                        const ch = client.webrtc.createDataChannel(channel, { reliable: true })
 | 
						|
                        ch.onopen = async event => {
 | 
						|
                            console.log(client.name, `打开 ${channel} 信道, 传输音乐数据`)
 | 
						|
                            // 将音乐数据分成多个小块,并逐个发送
 | 
						|
                            async function sendChunk(dataChannel, data, index = 0, buffer = new ArrayBuffer(0)) {
 | 
						|
                                while (index < data.byteLength) {
 | 
						|
                                    if (dataChannel.bufferedAmount <= THRESHOLD) {
 | 
						|
                                        const chunk = data.slice(index, index + CHUNK_SIZE)
 | 
						|
                                        dataChannel.send(chunk)
 | 
						|
                                        index += CHUNK_SIZE
 | 
						|
                                        buffer = appendBuffer(buffer, chunk)
 | 
						|
                                    }
 | 
						|
                                    await new Promise((resolve) => setTimeout(resolve, DELAY))
 | 
						|
                                }
 | 
						|
                                return buffer
 | 
						|
                            }
 | 
						|
                            await sendChunk(ch, item.arrayBuffer)
 | 
						|
                            console.log(`发送 ${channel} 信道数据结束`)
 | 
						|
                            ch.close() // 关闭信道
 | 
						|
                        }
 | 
						|
                    })
 | 
						|
                    return
 | 
						|
                }
 | 
						|
                console.log('未知类型:', type)
 | 
						|
            }
 | 
						|
        })
 | 
						|
 | 
						|
        // 延迟1500ms
 | 
						|
        //await new Promise((resolve) => setTimeout(resolve, 100))
 | 
						|
 | 
						|
        // 设置自己的主机名
 | 
						|
        const nameInput = document.createElement('input')
 | 
						|
        nameInput.type = 'text'
 | 
						|
        nameInput.placeholder = '请设置你的名字'
 | 
						|
        nameInput.value = name
 | 
						|
        nameInput.onchange = event => {
 | 
						|
            localStorage.setItem('username', event.target.value)
 | 
						|
            window.location.reload() // 简单刷新页面
 | 
						|
        }
 | 
						|
        document.body.appendChild(nameInput)
 | 
						|
    </script>
 | 
						|
</body>
 | 
						|
 | 
						|
</html> |