init

liusuyi

2024-10-10 38f29e38fcc668171dc05c53d40a36b895c86102

 ard-work/src/main/resources/templates/test.html

@@ -916,32 +916,10 @@

    let webrtcClient;
    //whep操作方法
    const retryPause = 2000;

    const video = document.getElementById('video');
    const message = document.getElementById('message');

    let nonAdvertisedCodecs = [];
    let pc = null;
    let restartTimeout = null;
    let sessionUrl = '';
    let offerData = '';
    let queuedCandidates = [];
    let defaultControls = false;

    const setMessage = (str) => {
        if (str !== '') {
            video.controls = false;
        } else {
            video.controls = defaultControls;
        }
        message.innerText = str;
    };

    const restartPause = 2000;
    const unquoteCredential = (v) => (
        JSON.parse(`"${v}"`)
    );

    const linkToIceServers = (links) => (
        (links !== null) ? links.split(', ').map((link) => {
            const m = link.match(/^<(.+?)>; rel="ice-server"(; username="(.*?)"; credential="(.*?)"; credential-type="password")?/i);
@@ -952,21 +930,20 @@
            if (m[3] !== undefined) {
                ret.username = unquoteCredential(m[3]);
                ret.credential = unquoteCredential(m[4]);
                ret.credentialType = 'password';
                ret.credentialType = "password";
            }

            return ret;
        }) : []
    );

    const parseOffer = (sdp) => {
    const parseOffer = (offer) => {
        const ret = {
            iceUfrag: '',
            icePwd: '',
            medias: [],
        };

        for (const line of sdp.split('\r\n')) {
        for (const line of offer.split('\r\n')) {
            if (line.startsWith('m=')) {
                ret.medias.push(line.slice('m='.length));
            } else if (ret.iceUfrag === '' && line.startsWith('a=ice-ufrag:')) {
@@ -978,131 +955,7 @@

        return ret;
    };

    const enableStereoPcmau = (section) => {
        let lines = section.split('\r\n');

        lines[0] += ' 118';
        lines.splice(lines.length - 1, 0, 'a=rtpmap:118 PCMU/8000/2');
        lines.splice(lines.length - 1, 0, 'a=rtcp-fb:118 transport-cc');

        lines[0] += ' 119';
        lines.splice(lines.length - 1, 0, 'a=rtpmap:119 PCMA/8000/2');
        lines.splice(lines.length - 1, 0, 'a=rtcp-fb:119 transport-cc');

        return lines.join('\r\n');
    };

    const enableMultichannelOpus = (section) => {
        let lines = section.split('\r\n');

        lines[0] += " 112";
        lines.splice(lines.length - 1, 0, "a=rtpmap:112 multiopus/48000/3");
        lines.splice(lines.length - 1, 0, "a=fmtp:112 channel_mapping=0,2,1;num_streams=2;coupled_streams=1");
        lines.splice(lines.length - 1, 0, "a=rtcp-fb:112 transport-cc");

        lines[0] += " 113";
        lines.splice(lines.length - 1, 0, "a=rtpmap:113 multiopus/48000/4");
        lines.splice(lines.length - 1, 0, "a=fmtp:113 channel_mapping=0,1,2,3;num_streams=2;coupled_streams=2");
        lines.splice(lines.length - 1, 0, "a=rtcp-fb:113 transport-cc");

        lines[0] += " 114";
        lines.splice(lines.length - 1, 0, "a=rtpmap:114 multiopus/48000/5");
        lines.splice(lines.length - 1, 0, "a=fmtp:114 channel_mapping=0,4,1,2,3;num_streams=3;coupled_streams=2");
        lines.splice(lines.length - 1, 0, "a=rtcp-fb:114 transport-cc");

        lines[0] += " 115";
        lines.splice(lines.length - 1, 0, "a=rtpmap:115 multiopus/48000/6");
        lines.splice(lines.length - 1, 0, "a=fmtp:115 channel_mapping=0,4,1,2,3,5;num_streams=4;coupled_streams=2");
        lines.splice(lines.length - 1, 0, "a=rtcp-fb:115 transport-cc");

        lines[0] += " 116";
        lines.splice(lines.length - 1, 0, "a=rtpmap:116 multiopus/48000/7");
        lines.splice(lines.length - 1, 0, "a=fmtp:116 channel_mapping=0,4,1,2,3,5,6;num_streams=4;coupled_streams=4");
        lines.splice(lines.length - 1, 0, "a=rtcp-fb:116 transport-cc");

        lines[0] += " 117";
        lines.splice(lines.length - 1, 0, "a=rtpmap:117 multiopus/48000/8");
        lines.splice(lines.length - 1, 0, "a=fmtp:117 channel_mapping=0,6,1,4,5,2,3,7;num_streams=5;coupled_streams=4");
        lines.splice(lines.length - 1, 0, "a=rtcp-fb:117 transport-cc");

        return lines.join('\r\n');
    };

    const enableL16 = (section) => {
        let lines = section.split('\r\n');

        lines[0] += " 120";
        lines.splice(lines.length - 1, 0, "a=rtpmap:120 L16/8000/2");
        lines.splice(lines.length - 1, 0, "a=rtcp-fb:120 transport-cc");

        lines[0] += " 121";
        lines.splice(lines.length - 1, 0, "a=rtpmap:121 L16/16000/2");
        lines.splice(lines.length - 1, 0, "a=rtcp-fb:121 transport-cc");

        lines[0] += " 122";
        lines.splice(lines.length - 1, 0, "a=rtpmap:122 L16/48000/2");
        lines.splice(lines.length - 1, 0, "a=rtcp-fb:122 transport-cc");

        return lines.join('\r\n');
    };

    const enableStereoOpus = (section) => {
        let opusPayloadFormat = '';
        let lines = section.split('\r\n');

        for (let i = 0; i < lines.length; i++) {
            if (lines[i].startsWith('a=rtpmap:') && lines[i].toLowerCase().includes('opus/')) {
                opusPayloadFormat = lines[i].slice('a=rtpmap:'.length).split(' ')[0];
                break;
            }
        }

        if (opusPayloadFormat === '') {
            return section;
        }

        for (let i = 0; i < lines.length; i++) {
            if (lines[i].startsWith('a=fmtp:' + opusPayloadFormat + ' ')) {
                if (!lines[i].includes('stereo')) {
                    lines[i] += ';stereo=1';
                }
                if (!lines[i].includes('sprop-stereo')) {
                    lines[i] += ';sprop-stereo=1';
                }
            }
        }

        return lines.join('\r\n');
    };

    const editOffer = (sdp) => {
        const sections = sdp.split('m=');

        for (let i = 0; i < sections.length; i++) {
            if (sections[i].startsWith('audio')) {
                sections[i] = enableStereoOpus(sections[i]);

                if (nonAdvertisedCodecs.includes('pcma/8000/2')) {
                    sections[i] = enableStereoPcmau(sections[i]);
                }

                if (nonAdvertisedCodecs.includes('multiopus/48000/6')) {
                    sections[i] = enableMultichannelOpus(sections[i]);
                }

                if (nonAdvertisedCodecs.includes('L16/48000/2')) {
                    sections[i] = enableL16(sections[i]);
                }

                break;
            }
        }

        return sections.join('m=');
    };

    const generateSdpFragment = (od, candidates) => {
    const generateSdpFragment = (offerData, candidates) => {
        const candidatesByMedia = {};
        for (const candidate of candidates) {
            const mid = candidate.sdpMLineIndex;
@@ -1112,12 +965,12 @@
            candidatesByMedia[mid].push(candidate);
        }

        let frag = 'a=ice-ufrag:' + od.iceUfrag + '\r\n'
            + 'a=ice-pwd:' + od.icePwd + '\r\n';
        let frag = 'a=ice-ufrag:' + offerData.iceUfrag + '\r\n'
            + 'a=ice-pwd:' + offerData.icePwd + '\r\n';

        let mid = 0;

        for (const media of od.medias) {
        for (const media of offerData.medias) {
            if (candidatesByMedia[mid] !== undefined) {
                frag += 'm=' + media + '\r\n'
                    + 'a=mid:' + mid + '\r\n';
@@ -1130,272 +983,180 @@
        }

        return frag;
    };
    }

    const loadStream = () => {
        requestICEServers();
    };
    class WHEPClient {
        constructor(whepUrl, videoId) {
            this.video = videoId;
            this.wurl = new URL('whep', whepUrl);
            this.pc = null;
            this.restartTimeout = null;
            this.eTag = '';
            this.queuedCandidates = [];
            this.start();
        }

    const supportsNonAdvertisedCodec = (codec, fmtp) => (
        new Promise((resolve, reject) => {
            const pc = new RTCPeerConnection({ iceServers: [] });
            pc.addTransceiver('audio', { direction: 'recvonly' });
            pc.createOffer()
                .then((offer) => {
                    if (offer.sdp.includes(' ' + codec)) { // codec is advertised, there's no need to add it manually
                        resolve(false);
                        return;
        start() {
            console.log("requesting ICE servers");
            fetch(this.wurl, {
                method: 'OPTIONS',
            })
                .then((res) => this.onIceServers(res))
                .catch((err) => {
                    console.log('error: ' + err);
                    this.scheduleRestart();
                });
        }

        onIceServers(res) {
            this.pc = new RTCPeerConnection({
                iceServers: linkToIceServers(res.headers.get('Link')),
            });

            const direction = "sendrecv";
            this.pc.addTransceiver("video", {direction});
            this.pc.addTransceiver("audio", {direction});

            this.pc.onicecandidate = (evt) => this.onLocalCandidate(evt);
            this.pc.oniceconnectionstatechange = () => this.onConnectionState();

            this.pc.ontrack = (evt) => {
                console.log("new track:", evt.track.kind);
                document.getElementById(this.video).srcObject = evt.streams[0];
            };

            this.pc.createOffer()
                .then((offer) => this.onLocalOffer(offer));
        }

        onLocalOffer(offer) {
            this.offerData = parseOffer(offer.sdp);
            this.pc.setLocalDescription(offer);

            console.log("sending offer");
            console.log(this.wurl);
            fetch(this.wurl, {
                method: 'POST',
                headers: {
                    'Content-Type': 'application/sdp',
                },
                body: offer.sdp,
            })
                .then((res) => {
                    if (res.status !== 201) {
                        throw new Error('bad status code');
                    }
                    const sections = offer.sdp.split('m=audio');
                    const lines = sections[1].split('\r\n');
                    lines[0] += ' 118';
                    lines.splice(lines.length - 1, 0, 'a=rtpmap:118 ' + codec);
                    if (fmtp !== undefined) {
                        lines.splice(lines.length - 1, 0, 'a=fmtp:118 ' + fmtp);
                    // this.eTag = res.headers.get('ETag');
                    this.eTag = res.headers.get("ETag") || res.headers.get('E-Tag');
                    this.wurl = new URL(res.headers.get('location'),  this.wurl.origin).toString();
                    return res.text();
                }
                )
                .then((sdp) => this.onRemoteAnswer(new RTCSessionDescription({
                    type: 'answer',
                    sdp,
                })))
                .catch((err) => {
                    console.log('error: ' + err);
                    this.scheduleRestart();
                });
        }

        onConnectionState() {
            if (this.restartTimeout !== null) {
                return;
            }

            console.log("peer connection state:", this.pc.iceConnectionState);

            switch (this.pc.iceConnectionState) {
                case "disconnected":
                    this.scheduleRestart();
            }
        }

        onRemoteAnswer(answer) {
            if (this.restartTimeout !== null) {
                return;
            }

            this.pc.setRemoteDescription(new RTCSessionDescription(answer));

            if (this.queuedCandidates.length !== 0) {
                this.sendLocalCandidates(this.queuedCandidates);
                this.queuedCandidates = [];
            }
        }

        onLocalCandidate(evt) {
            if (this.restartTimeout !== null) {
                return;
            }

            if (evt.candidate !== null) {
                if (this.eTag === '') {
                    console.log("222222222222222222222")
                    this.queuedCandidates.push(evt.candidate);
                } else {
                    console.log("333333333333333333333")
                    this.sendLocalCandidates([evt.candidate])
                }
            }
        }

        sendLocalCandidates(candidates) {
            fetch(this.wurl, {
                method: 'PATCH',
                headers: {
                    'Content-Type': 'application/trickle-ice-sdpfrag',
                    'If-Match': this.eTag,
                },

                body: generateSdpFragment(this.offerData, candidates),
            })
                .then((res) => {
                    if (res.status !== 204) {
                        throw new Error('bad status code');
                    }
                    sections[1] = lines.join('\r\n');
                    offer.sdp = sections.join('m=audio');
                    return pc.setLocalDescription(offer);
                })
                .then(() => {
                    return pc.setRemoteDescription(new RTCSessionDescription({
                        type: 'answer',
                        sdp: 'v=0\r\n'
                            + 'o=- 6539324223450680508 0 IN IP4 0.0.0.0\r\n'
                            + 's=-\r\n'
                            + 't=0 0\r\n'
                            + 'a=fingerprint:sha-256 0D:9F:78:15:42:B5:4B:E6:E2:94:3E:5B:37:78:E1:4B:54:59:A3:36:3A:E5:05:EB:27:EE:8F:D2:2D:41:29:25\r\n'
                            + 'm=audio 9 UDP/TLS/RTP/SAVPF 118\r\n'
                            + 'c=IN IP4 0.0.0.0\r\n'
                            + 'a=ice-pwd:7c3bf4770007e7432ee4ea4d697db675\r\n'
                            + 'a=ice-ufrag:29e036dc\r\n'
                            + 'a=sendonly\r\n'
                            + 'a=rtcp-mux\r\n'
                            + 'a=rtpmap:118 ' + codec + '\r\n'
                            + ((fmtp !== undefined) ? 'a=fmtp:118 ' + fmtp + '\r\n' : ''),
                    }));
                })
                .then(() => {
                    resolve(true);
                })
                .catch((err) => {
                    resolve(false);
                })
                .finally(() => {
                    pc.close();
                    console.log('error: ' + err);
                    this.scheduleRestart();
                });
        })
    );
        }

    const getNonAdvertisedCodecs = () => {
        Promise.all([
            ['pcma/8000/2'],
            ['multiopus/48000/6', 'channel_mapping=0,4,1,2,3,5;num_streams=4;coupled_streams=2'],
            ['L16/48000/2']
        ].map((c) => supportsNonAdvertisedCodec(c[0], c[1]).then((r) => (r) ? c[0] : false)))
            .then((c) => c.filter((e) => e !== false))
            .then((codecs) => {
                nonAdvertisedCodecs = codecs;
                loadStream();
            });
    };

    const onError = (err) => {
        if (restartTimeout === null) {
            setMessage(err + ', retrying in some seconds');

            if (pc !== null) {
                pc.close();
                pc = null;
        scheduleRestart() {
            if (this.restartTimeout !== null) {
                return;
            }

            restartTimeout = window.setTimeout(() => {
                restartTimeout = null;
                loadStream();
            }, retryPause);

            if (sessionUrl) {
                fetch(sessionUrl, {
                    method: 'DELETE',
                });
            if (this.pc !== null) {
                this.pc.close();
                this.pc = null;
            }
            sessionUrl = '';

            queuedCandidates = [];
            this.restartTimeout = window.setTimeout(() => {
                this.restartTimeout = null;
                this.start();
            }, restartPause);

            this.eTag = '';
            this.queuedCandidates = [];
        }
    };

    const sendLocalCandidates = (candidates) => {
        fetch(sessionUrl + window.location.search, {
            method: 'PATCH',
            headers: {
                'Content-Type': 'application/trickle-ice-sdpfrag',
                'If-Match': '*',
            },
            body: generateSdpFragment(offerData, candidates),
        })
            .then((res) => {
                switch (res.status) {
                    case 204:
                        break;
                    case 404:
                        throw new Error('stream not found');
                    default:
                        throw new Error(`bad status code ${res.status}`);
        stop() {
            if (this.pc) {
                try {
                    this.pc.close();
                } catch (e) {
                    console.log("Failure close peer connection:" + e);
                }
            })
            .catch((err) => {
                onError(err.toString());
            });
    };

    const onLocalCandidate = (evt) => {
        if (restartTimeout !== null) {
            return;
        }

        if (evt.candidate !== null) {
            if (sessionUrl === '') {
                queuedCandidates.push(evt.candidate);
            } else {
                sendLocalCandidates([evt.candidate])
                this.pc = null;
            }
        }
    };

    const onRemoteAnswer = (sdp) => {
        if (restartTimeout !== null) {
            return;
        }

        pc.setRemoteDescription(new RTCSessionDescription({
            type: 'answer',
            sdp,
        }))
            .then(() => {
                if (queuedCandidates.length !== 0) {
                    sendLocalCandidates(queuedCandidates);
                    queuedCandidates = [];
                }
            })
            .catch((err) => {
                onError(err.toString());
            });
    };

    const sendOffer = (offer) => {
        fetch(new URL('whep', window.location.href) + window.location.search, {
            method: 'POST',
            headers: {
                'Content-Type': 'application/sdp',
            },
            body: offer.sdp,
        })
            .then((res) => {
                switch (res.status) {
                    case 201:
                        break;
                    case 404:
                        throw new Error('stream not found');
                    case 400:
                        return res.json().then((e) => { throw new Error(e.error); });
                    default:
                        throw new Error(`bad status code ${res.status}`);
                }

                sessionUrl = new URL(res.headers.get('location'), window.location.href).toString();

                return res.text()
                    .then((sdp) => onRemoteAnswer(sdp));
            })
            .catch((err) => {
                onError(err.toString());
            });
    };

    const createOffer = () => {
        pc.createOffer()
            .then((offer) => {
                offer.sdp = editOffer(offer.sdp);
                offerData = parseOffer(offer.sdp);
                pc.setLocalDescription(offer)
                    .then(() => {
                        sendOffer(offer);
                    })
                    .catch((err) => {
                        onError(err.toString());
                    });
            })
            .catch((err) => {
                onError(err.toString());
            });
    };

    const onConnectionState = () => {
        if (restartTimeout !== null) {
            return;
        }

        if (pc.iceConnectionState === 'disconnected') {
            onError('peer connection closed');
        }
    };

    const onTrack = (evt) => {
        setMessage('');
        video.srcObject = evt.streams[0];
    };

    const requestICEServers = () => {
        fetch(new URL('whep', window.location.href) + window.location.search, {
            method: 'OPTIONS',
        })
            .then((res) => {
                pc = new RTCPeerConnection({
                    iceServers: linkToIceServers(res.headers.get('Link')),
                    // https://webrtc.org/getting-started/unified-plan-transition-guide
                    sdpSemantics: 'unified-plan',
                });

                const direction = 'sendrecv';
                pc.addTransceiver('video', { direction });
                pc.addTransceiver('audio', { direction });

                pc.onicecandidate = (evt) => onLocalCandidate(evt);
                pc.oniceconnectionstatechange = () => onConnectionState();
                pc.ontrack = (evt) => onTrack(evt);

                createOffer();
            })
            .catch((err) => {
                onError(err.toString());
            });
    };

    const parseBoolString = (str, defaultVal) => {
        str = (str || '');

        if (['1', 'yes', 'true'].includes(str.toLowerCase())) {
            return true;
        }
        if (['0', 'no', 'false'].includes(str.toLowerCase())) {
            return false;
        }
        return defaultVal;
    };

    const loadAttributesFromQuery = () => {
        const params = new URLSearchParams(window.location.search);
        video.controls = parseBoolString(params.get('controls'), true);
        video.muted = parseBoolString(params.get('muted'), true);
        video.autoplay = parseBoolString(params.get('autoplay'), true);
        video.playsInline = parseBoolString(params.get('playsinline'), true);
        defaultControls = video.controls;
    };

    }

    function realView(whepUrl, videoId) {
        console.log(whepUrl)
        webrtcClient = new WHEPClient(whepUrl, videoId);
    }
</script>