chaturbate-dvr/handler/view/script.js

function data() {
    return {
        settings: {},
        channels: [],
        is_updating_channels: false,
        form_data: {
            username: "",
            resolution: "1080",
            resolution_fallback: "up",
            framerate: "30",
            filename_pattern: "{{.Username}}_{{.Year}}-{{.Month}}-{{.Day}}_{{.Hour}}-{{.Minute}}-{{.Second}}{{if .Sequence}}_{{.Sequence}}{{end}}",
            split_filesize: 0,
            split_duration: 0,
        },

        // openCreateDialog
        openCreateDialog() {
            document.getElementById("create-dialog").showModal()
        },

        // closeCreateDialog
        closeCreateDialog() {
            document.getElementById("create-dialog").close()
            this.resetCreateDialog()
        },

        // submitCreateDialog
        submitCreateDialog() {
            this.createChannel()
            this.closeCreateDialog()
        },

        // error
        error() {
            alert("Error occurred, please refresh the page if something is wrong.")
        },

        //
        async call(path, body) {
            try {
                var resp = await fetch(`http://localhost:8080/api/${path}`, {
                    body: JSON.stringify(body),
                    method: "POST",
                })
                if (resp.status !== 200) {
                    this.error()
                    return [null, true]
                }
                return [(await resp.json()), false]
            } catch {
                this.error()
                return [null, true]
            }
        },

        // getSettings
        async getSettings() {
            var [resp, err] = await this.call("get_settings", {})
            if (!err) {
                this.settings = resp
                this.resetCreateDialog()
            }
        },

        // init
        async init() {
            document.getElementById("create-dialog").addEventListener("close", () => this.resetCreateDialog())

            await this.getSettings()
            await this.listChannels()
            this.listenUpdate()
        },

        // resetCreateDialog
        resetCreateDialog() {
            document.getElementById("splitting-accordion").open = false

            this.form_data = {
                username: "",
                resolution: this.settings.resolution.toString(),
                resolution_fallback: this.settings.resolution_fallback,
                framerate: this.settings.framerate.toString(),
                filename_pattern: this.settings.filename_pattern,
                split_filesize: this.settings.split_filesize.toString(),
                split_duration: this.settings.split_duration.toString(),
            }
        },

        // createChannel
        async createChannel() {
            await this.call("create_channel", {
                username: this.form_data.username,
                resolution: parseInt(this.form_data.resolution),
                resolution_fallback: this.form_data.resolution_fallback,
                framerate: parseInt(this.form_data.framerate),
                filename_pattern: this.form_data.filename_pattern,
                split_filesize: parseInt(this.form_data.split_filesize),
                split_duration: parseInt(this.form_data.split_duration),
            })
        },

        // deleteChannel
        async deleteChannel(username) {
            var [_, err] = await this.call("delete_channel", { username })
            if (!err) {
                this.channels = this.channels.filter(ch => ch.username !== username)
            }
        },

        // pauseChannel
        async pauseChannel(username) {
            await this.call("pause_channel", { username })
        },

        // resumeChannel
        async resumeChannel(username) {
            await this.call("resume_channel", { username })
        },

        // listChannels
        async listChannels() {
            if (this.is_updating_channels) {
                return
            }
            var [resp, err] = await this.call("list_channels", {})
            if (!err) {
                this.channels = resp.channels
                this.channels.forEach(ch => {
                    this.scrollLogs(ch.username)
                })
            }
            this.is_updating_channels = false
        },

        // listenUpdate
        listenUpdate() {
            var source = new EventSource("http://localhost:8080/api/listen_update")

            source.onmessage = event => {
                var data = JSON.parse(event.data)

                // If the channel is not in the list or is stopped, refresh the list.
                if (!this.channels.some(ch => ch.username === data.username) || data.is_stopped) {
                    this.listChannels()
                    return
                }

                var index = this.channels.findIndex(ch => ch.username === data.username)

                if (index === -1) {
                    return
                }

                this.channels[index].segment_duration = data.segment_duration
                this.channels[index].segment_filesize = data.segment_filesize
                this.channels[index].filename = data.filename
                this.channels[index].last_streamed_at = data.last_streamed_at
                this.channels[index].is_online = data.is_online
                this.channels[index].is_paused = data.is_paused
                this.channels[index].logs = [...this.channels[index].logs, data.log]

                if (this.channels[index].logs.length > 100) {
                    this.channels[index].logs = this.channels[index].logs.slice(-100)
                }

                this.scrollLogs(data.username)
            }

            source.onerror = err => {
                source.close()
            }
        },

        downloadLogs(username) {
            var a = window.document.createElement('a');
            a.href = window.URL.createObjectURL(new Blob([this.channels[this.channels.findIndex(ch => ch.username === username)].logs.join('\n')], { type: 'text/plain', oneTimeOnly: true }));
            a.download = 'test.txt';

            // Append anchor to body.
            document.body.appendChild(a);
            a.click();

            // Remove anchor from body
            document.body.removeChild(a);
        },

        //
        scrollLogs(username) {
            // Wait for the DOM to update.
            setTimeout(() => {
                var logs_element = document.getElementById(`${username}-logs`)

                if (!logs_element) {
                    return
                }
                logs_element.scrollTop = logs_element.scrollHeight
            }, 1)
        },
    }
}