@serifpersia/dango 3.6.5 → 3.6.8

package/server/dist/discord-gateway.js

@@ -1,385 +1 @@
-import WebSocket from 'ws';
-import logger from './logger.js';
-import { CONFIG } from './config.js';
-import dotenv from 'dotenv';
-const log = logger.child({ module: 'DiscordGateway' });
-class DiscordGatewayService {
-    gateway = null;
-    heartbeat = null;
-    token = null;
-    reconnectTimeout = null;
-    lastActivity = null;
-    isEnabled = false;
-    RECONNECT_DELAY = 15000;
-    externalAssetCache = new Map();
-    setEnabled(enabled) {
-        this.isEnabled = enabled;
-        if (enabled) {
-            this.loadToken();
-            this.connect();
-        }
-        else {
-            this.disconnect();
-        }
-    }
-    get serviceEnabled() {
-        return this.isEnabled && !!this.token && this.token.length > 30;
-    }
-    hasToken() {
-        return !!this.token && this.token.length > 30;
-    }
-    loadToken() {
-        this.token = process.env.DISCORD_GATEWAY_TOKEN || null;
-    }
-    reloadToken() {
-        dotenv.config({ path: CONFIG.ENV_PATH, override: true });
-        this.loadToken();
-        if (this.serviceEnabled) {
-            this.disconnect();
-            this.connect();
-        }
-    }
-    async getExternalAsset(url) {
-        if (!CONFIG.DISCORD_CLIENT_ID || !this.token)
-            return null;
-        const cached = this.externalAssetCache.get(url);
-        if (cached)
-            return cached;
-        try {
-            const res = await fetch(`https://discord.com/api/v9/applications/${CONFIG.DISCORD_CLIENT_ID}/external-assets`, {
-                method: 'POST',
-                headers: {
-                    Authorization: this.token,
-                    'Content-Type': 'application/json',
-                },
-                body: JSON.stringify({ urls: [url] }),
-            });
-            const text = await res.text();
-            if (!res.ok) {
-                log.warn(`external-assets proxy failed ${res.status}: ${text.slice(0, 200)}`);
-                return null;
-            }
-            const data = JSON.parse(text);
-            if (!data[0]?.external_asset_path)
-                return null;
-            const mp = `mp:${data[0].external_asset_path}`;
-            this.externalAssetCache.set(url, mp);
-            return mp;
-        }
-        catch (err) {
-            log.warn({ err }, 'external-assets proxy error');
-            return null;
-        }
-    }
-    resolveImageUrl(data) {
-        const isSafeUrl = (url) => {
-            if (!url.startsWith('https://'))
-                return false;
-            if (url.includes('localhost') || url.includes('127.0.0.1'))
-                return false;
-            if (url.includes('s4.anilist.co') || url.includes('anilistcdn'))
-                return true;
-            const blocked = [
-                'youtube-anime.com',
-                'animepahe',
-                'animeya.cc',
-                'gogocdn.net',
-                'weeabo0.xyz',
-                'japaneseasmr.com',
-            ];
-            return !blocked.some((d) => url.includes(d));
-        };
-        let thumb = data.thumbnail;
-        if (thumb && thumb.includes('/api/image-proxy')) {
-            const m = thumb.match(/url=([^&]+)/);
-            if (m) {
-                try {
-                    thumb = decodeURIComponent(m[1]);
-                }
-                catch (_e) {
-                    void _e;
-                }
-            }
-        }
-        if (thumb && data.providerName !== 'AnimePahe' && isSafeUrl(thumb))
-            return thumb;
-        if (data.thumbnails) {
-            const safe = data.thumbnails.find((t) => t && (t.includes('s4.anilist.co') || t.includes('anilistcdn')));
-            if (safe)
-                return safe;
-        }
-        if (thumb && isSafeUrl(thumb))
-            return thumb;
-        return null;
-    }
-    async updatePresence(data) {
-        this.lastActivity = data;
-        if (!this.isEnabled || !this.gateway || this.gateway.readyState !== WebSocket.OPEN)
-            return;
-        const name = 'dango';
-        const details = data.title;
-        const state = data.stateLine
-            ? data.stateLine
-            : `Episode ${data.episode}${data.totalEpisodes ? `/${data.totalEpisodes}` : ''}`;
-        const type = data.providerName === 'ASMR' || data.providerName === 'Radio' || data.providerName === 'Music'
-            ? 2
-            : 3;
-        const imageUrl = this.resolveImageUrl(data);
-        let assets;
-        if (imageUrl) {
-            if (CONFIG.DISCORD_CLIENT_ID) {
-                const proxied = await this.getExternalAsset(imageUrl);
-                if (proxied) {
-                    assets = {
-                        large_image: proxied,
-                        large_text: data.title.slice(0, 128),
-                        small_image: 'small',
-                        small_text: 'dango',
-                    };
-                }
-                else {
-                    assets = { large_image: 'logo', large_text: data.title.slice(0, 128) };
-                }
-            }
-        }
-        else if (CONFIG.DISCORD_CLIENT_ID) {
-            assets = { large_image: 'logo', large_text: data.title.slice(0, 128) };
-        }
-        const presence = {
-            since: null,
-            activities: [
-                {
-                    name,
-                    type,
-                    state: data.isPlaying ? state : `${state} (Paused)`,
-                    details,
-                    ...(CONFIG.DISCORD_CLIENT_ID ? { application_id: CONFIG.DISCORD_CLIENT_ID } : {}),
-                    ...(assets ? { assets } : {}),
-                    buttons: ['Learn More'],
-                    metadata: { button_urls: ['https://github.com/serifpersia/dango'] },
-                    status_display_type: 2,
-                },
-            ],
-            status: 'online',
-            afk: false,
-        };
-        const nowMs = Date.now();
-        if (data.isPlaying) {
-            if (data.currentTime && data.currentTime > 0) {
-                presence.since = nowMs - Math.round(data.currentTime * 1000);
-            }
-            if (data.duration && data.duration > data.currentTime) {
-                presence.activities[0].timestamps = {
-                    start: nowMs - Math.round(data.currentTime * 1000),
-                    end: nowMs + Math.round((data.duration - data.currentTime) * 1000),
-                };
-            }
-        }
-        try {
-            this.gateway.send(JSON.stringify({ op: 3, d: presence }));
-        }
-        catch (err) {
-            log.error({ err }, 'Failed to send gateway presence update');
-        }
-    }
-    setIdleStatus(page) {
-        if (!this.isEnabled || !this.gateway || this.gateway.readyState !== WebSocket.OPEN)
-            return;
-        const pageLabels = {
-            home: { details: 'Home', state: 'Browsing anime' },
-            search: { details: 'Search', state: 'Exploring titles...' },
-            watchlist: { details: 'Watchlist', state: 'Reviewing the watchlist' },
-            anime: { details: 'Anime Info', state: 'Reading show details' },
-            insights: { details: 'Insights', state: 'Reviewing stats' },
-            settings: { details: 'Settings', state: 'Tweaking preferences' },
-            trackers: { details: 'Trackers', state: 'Syncing watchlists' },
-            map: { details: 'Map', state: 'Exploring the global user map' },
-            asmr: { details: 'ASMR', state: 'Browsing ASMR works' },
-            radio: { details: 'Radio', state: 'Browsing radio stations' },
-            music: { details: 'Music', state: 'Browsing music' },
-            tv: { details: 'TV', state: 'Browsing movies & shows' },
-            manga: { details: 'Manga', state: 'Browsing manga' },
-        };
-        const label = pageLabels[page] ?? { details: 'dango', state: 'Idle' };
-        try {
-            this.gateway.send(JSON.stringify({
-                op: 3,
-                d: {
-                    since: null,
-                    activities: [
-                        {
-                            name: 'dango',
-                            type: page === 'asmr' || page === 'radio' || page === 'music' ? 2 : 3,
-                            state: label.state,
-                            details: label.details,
-                            ...(CONFIG.DISCORD_CLIENT_ID ? { application_id: CONFIG.DISCORD_CLIENT_ID } : {}),
-                            ...(CONFIG.DISCORD_CLIENT_ID
-                                ? { assets: { large_image: 'logo', large_text: 'dango' } }
-                                : {}),
-                            buttons: ['Learn More'],
-                            metadata: { button_urls: ['https://github.com/serifpersia/dango'] },
-                            status_display_type: 2,
-                        },
-                    ],
-                    status: 'online',
-                    afk: false,
-                },
-            }));
-        }
-        catch (err) {
-            log.error({ err }, 'Failed to send gateway idle status');
-        }
-    }
-    clearPresence() {
-        if (!this.isEnabled)
-            return;
-        if (this.gateway && this.gateway.readyState === WebSocket.OPEN) {
-            try {
-                this.gateway.send(JSON.stringify({
-                    op: 3,
-                    d: { since: null, activities: [], status: 'online', afk: false },
-                }));
-            }
-            catch {
-                // ignore
-            }
-        }
-        this.lastActivity = null;
-    }
-    shutdown() {
-        this.isEnabled = false;
-        this.token = null;
-        this.disconnect();
-    }
-    disconnect() {
-        if (this.reconnectTimeout) {
-            clearTimeout(this.reconnectTimeout);
-            this.reconnectTimeout = null;
-        }
-        if (this.gateway) {
-            try {
-                if (this.gateway.readyState === WebSocket.OPEN) {
-                    this.gateway.send(JSON.stringify({
-                        op: 3,
-                        d: { since: null, activities: [], status: 'online', afk: false },
-                    }));
-                }
-                this.gateway.close();
-            }
-            catch {
-                // ignore
-            }
-            this.gateway = null;
-        }
-        if (this.heartbeat) {
-            clearInterval(this.heartbeat);
-            this.heartbeat = null;
-        }
-    }
-    connect() {
-        if (!this.isEnabled || !this.token || this.gateway)
-            return;
-        const wsUrl = 'wss://gateway.discord.gg/?encoding=json&v=10';
-        log.debug('Connecting Discord gateway');
-        this.gateway = new WebSocket(wsUrl);
-        let heartbeatInterval = 41250;
-        this.gateway.on('message', (data) => {
-            try {
-                const msg = JSON.parse(data.toString());
-                if (msg.op === 10) {
-                    heartbeatInterval = msg.d.heartbeat_interval;
-                    this.heartbeat = setInterval(() => {
-                        try {
-                            if (this.gateway && this.gateway.readyState === WebSocket.OPEN) {
-                                this.gateway.send(JSON.stringify({ op: 1, d: null }));
-                            }
-                        }
-                        catch {
-                            // ignore
-                        }
-                    }, heartbeatInterval);
-                    const identify = {
-                        op: 2,
-                        d: {
-                            token: this.token,
-                            capabilities: 1021,
-                            client_state: {
-                                guild_hashes: {},
-                                highest_last_message_id: '0',
-                                private_channels_version: '0',
-                                read_state_version: 0,
-                                user_guild_settings_version: -1,
-                                user_settings_version: -1,
-                            },
-                            compress: false,
-                            presence: {
-                                since: null,
-                                activities: [],
-                                status: 'online',
-                                afk: false,
-                            },
-                            properties: {
-                                os: 'Windows',
-                                os_version: '10.0.19044',
-                                os_arch: 'x64',
-                                browser: 'Discord Client',
-                                client_version: '0.0.20',
-                                client_build_number: 152131,
-                                client_event_source: null,
-                                release_channel: 'stable',
-                                system_locale: 'en-US',
-                            },
-                        },
-                    };
-                    if (this.gateway) {
-                        this.gateway.send(JSON.stringify(identify));
-                    }
-                }
-                else if (msg.op === 11) {
-                    // ignore
-                }
-                else if (msg.op === 0) {
-                    if (msg.t === 'READY') {
-                        log.info(`Gateway READY as ${msg.d.user?.username || 'unknown'}`);
-                        if (this.lastActivity) {
-                            void this.updatePresence(this.lastActivity).catch((err) => log.warn({ err }, 'failed to restore presence after READY'));
-                        }
-                    }
-                }
-                else if (msg.op === 7) {
-                    log.warn('Gateway reconnect requested by Discord');
-                    this.handleDisconnect();
-                }
-                else if (msg.op === 9) {
-                    log.warn('Gateway invalid session, reconnecting...');
-                    this.handleDisconnect();
-                }
-            }
-            catch (err) {
-                log.debug({ err }, 'Gateway message parse error');
-            }
-        });
-        this.gateway.on('close', (code, reason) => {
-            log.warn(`Gateway closed ${code} ${reason.toString().slice(0, 100)}`);
-            this.handleDisconnect();
-        });
-        this.gateway.on('error', (err) => {
-            log.error(err, 'Gateway WebSocket error');
-            this.handleDisconnect();
-        });
-    }
-    handleDisconnect() {
-        if (this.heartbeat) {
-            clearInterval(this.heartbeat);
-            this.heartbeat = null;
-        }
-        this.gateway = null;
-        if (this.isEnabled && this.token) {
-            this.reconnectTimeout = setTimeout(() => {
-                this.reconnectTimeout = null;
-                this.connect();
-            }, this.RECONNECT_DELAY);
-        }
-    }
-}
-export const discordGatewayService = new DiscordGatewayService();
+import o from"ws";import d from"./logger.js";import{CONFIG as r}from"./config.js";import m from"dotenv";const a=d.child({module:"DiscordGateway"});class p{gateway=null;heartbeat=null;token=null;reconnectTimeout=null;lastActivity=null;isEnabled=!1;RECONNECT_DELAY=15e3;externalAssetCache=new Map;setEnabled(e){this.isEnabled=e,e?(this.loadToken(),this.connect()):this.disconnect()}get serviceEnabled(){return this.isEnabled&&!!this.token&&this.token.length>30}hasToken(){return!!this.token&&this.token.length>30}loadToken(){this.token=process.env.DISCORD_GATEWAY_TOKEN||null}reloadToken(){m.config({path:r.ENV_PATH,override:!0}),this.loadToken(),this.serviceEnabled&&(this.disconnect(),this.connect())}async getExternalAsset(e){if(!r.DISCORD_CLIENT_ID||!this.token)return null;const n=this.externalAssetCache.get(e);if(n)return n;try{const s=await fetch(`https://discord.com/api/v9/applications/${r.DISCORD_CLIENT_ID}/external-assets`,{method:"POST",headers:{Authorization:this.token,"Content-Type":"application/json"},body:JSON.stringify({urls:[e]})}),t=await s.text();if(!s.ok)return a.warn(`external-assets proxy failed ${s.status}: ${t.slice(0,200)}`),null;const i=JSON.parse(t);if(!i[0]?.external_asset_path)return null;const l=`mp:${i[0].external_asset_path}`;return this.externalAssetCache.set(e,l),l}catch(s){return a.warn({err:s},"external-assets proxy error"),null}}resolveImageUrl(e){const n=t=>!t.startsWith("https://")||t.includes("localhost")||t.includes("127.0.0.1")?!1:t.includes("s4.anilist.co")||t.includes("anilistcdn")?!0:!["youtube-anime.com","animepahe","animeya.cc","gogocdn.net","weeabo0.xyz","japaneseasmr.com"].some(l=>t.includes(l));let s=e.thumbnail;if(s&&s.includes("/api/image-proxy")){const t=s.match(/url=([^&]+)/);if(t)try{s=decodeURIComponent(t[1])}catch(i){}}if(s&&e.providerName!=="AnimePahe"&&n(s))return s;if(e.thumbnails){const t=e.thumbnails.find(i=>i&&(i.includes("s4.anilist.co")||i.includes("anilistcdn")));if(t)return t}return s&&n(s)?s:null}async updatePresence(e){if(this.lastActivity=e,!this.isEnabled||!this.gateway||this.gateway.readyState!==o.OPEN)return;const n="dango",s=e.title,t=e.stateLine?e.stateLine:`Episode ${e.episode}${e.totalEpisodes?`/${e.totalEpisodes}`:""}`,i=e.providerName==="ASMR"||e.providerName==="Radio"||e.providerName==="Music"?2:3,l=this.resolveImageUrl(e);let c;if(l){if(r.DISCORD_CLIENT_ID){const g=await this.getExternalAsset(l);g?c={large_image:g,large_text:e.title.slice(0,128),small_image:"small",small_text:"dango"}:c={large_image:"logo",large_text:e.title.slice(0,128)}}}else r.DISCORD_CLIENT_ID&&(c={large_image:"logo",large_text:e.title.slice(0,128)});const h={since:null,activities:[{name:n,type:i,state:e.isPlaying?t:`${t} (Paused)`,details:s,...r.DISCORD_CLIENT_ID?{application_id:r.DISCORD_CLIENT_ID}:{},...c?{assets:c}:{},buttons:["Learn More"],metadata:{button_urls:["https://github.com/serifpersia/dango"]},status_display_type:2}],status:"online",afk:!1},u=Date.now();e.isPlaying&&(e.currentTime&&e.currentTime>0&&(h.since=u-Math.round(e.currentTime*1e3)),e.duration&&e.duration>e.currentTime&&(h.activities[0].timestamps={start:u-Math.round(e.currentTime*1e3),end:u+Math.round((e.duration-e.currentTime)*1e3)}));try{this.gateway.send(JSON.stringify({op:3,d:h}))}catch(g){a.error({err:g},"Failed to send gateway presence update")}}setIdleStatus(e){if(!this.isEnabled||!this.gateway||this.gateway.readyState!==o.OPEN)return;const s={home:{details:"Home",state:"Browsing anime"},search:{details:"Search",state:"Exploring titles..."},watchlist:{details:"Watchlist",state:"Reviewing the watchlist"},anime:{details:"Anime Info",state:"Reading show details"},insights:{details:"Insights",state:"Reviewing stats"},settings:{details:"Settings",state:"Tweaking preferences"},trackers:{details:"Trackers",state:"Syncing watchlists"},map:{details:"Map",state:"Exploring the global user map"},asmr:{details:"ASMR",state:"Browsing ASMR works"},radio:{details:"Radio",state:"Browsing radio stations"},music:{details:"Music",state:"Browsing music"},tv:{details:"TV",state:"Browsing movies & shows"},manga:{details:"Manga",state:"Browsing manga"}}[e]??{details:"dango",state:"Idle"};try{this.gateway.send(JSON.stringify({op:3,d:{since:null,activities:[{name:"dango",type:e==="asmr"||e==="radio"||e==="music"?2:3,state:s.state,details:s.details,...r.DISCORD_CLIENT_ID?{application_id:r.DISCORD_CLIENT_ID}:{},...r.DISCORD_CLIENT_ID?{assets:{large_image:"logo",large_text:"dango"}}:{},buttons:["Learn More"],metadata:{button_urls:["https://github.com/serifpersia/dango"]},status_display_type:2}],status:"online",afk:!1}}))}catch(t){a.error({err:t},"Failed to send gateway idle status")}}clearPresence(){if(this.isEnabled){if(this.gateway&&this.gateway.readyState===o.OPEN)try{this.gateway.send(JSON.stringify({op:3,d:{since:null,activities:[],status:"online",afk:!1}}))}catch{}this.lastActivity=null}}shutdown(){this.isEnabled=!1,this.token=null,this.disconnect()}disconnect(){if(this.reconnectTimeout&&(clearTimeout(this.reconnectTimeout),this.reconnectTimeout=null),this.gateway){try{this.gateway.readyState===o.OPEN&&this.gateway.send(JSON.stringify({op:3,d:{since:null,activities:[],status:"online",afk:!1}})),this.gateway.close()}catch{}this.gateway=null}this.heartbeat&&(clearInterval(this.heartbeat),this.heartbeat=null)}connect(){if(!this.isEnabled||!this.token||this.gateway)return;const e="wss://gateway.discord.gg/?encoding=json&v=10";a.debug("Connecting Discord gateway"),this.gateway=new o(e);let n=41250;this.gateway.on("message",s=>{try{const t=JSON.parse(s.toString());if(t.op===10){n=t.d.heartbeat_interval,this.heartbeat=setInterval(()=>{try{this.gateway&&this.gateway.readyState===o.OPEN&&this.gateway.send(JSON.stringify({op:1,d:null}))}catch{}},n);const i={op:2,d:{token:this.token,capabilities:1021,client_state:{guild_hashes:{},highest_last_message_id:"0",private_channels_version:"0",read_state_version:0,user_guild_settings_version:-1,user_settings_version:-1},compress:!1,presence:{since:null,activities:[],status:"online",afk:!1},properties:{os:"Windows",os_version:"10.0.19044",os_arch:"x64",browser:"Discord Client",client_version:"0.0.20",client_build_number:152131,client_event_source:null,release_channel:"stable",system_locale:"en-US"}}};this.gateway&&this.gateway.send(JSON.stringify(i))}else t.op===11||(t.op===0?t.t==="READY"&&(a.info(`Gateway READY as ${t.d.user?.username||"unknown"}`),this.lastActivity&&this.updatePresence(this.lastActivity).catch(i=>a.warn({err:i},"failed to restore presence after READY"))):t.op===7?(a.warn("Gateway reconnect requested by Discord"),this.handleDisconnect()):t.op===9&&(a.warn("Gateway invalid session, reconnecting..."),this.handleDisconnect()))}catch(t){a.debug({err:t},"Gateway message parse error")}}),this.gateway.on("close",(s,t)=>{a.warn(`Gateway closed ${s} ${t.toString().slice(0,100)}`),this.handleDisconnect()}),this.gateway.on("error",s=>{a.error(s,"Gateway WebSocket error"),this.handleDisconnect()})}handleDisconnect(){this.heartbeat&&(clearInterval(this.heartbeat),this.heartbeat=null),this.gateway=null,this.isEnabled&&this.token&&(this.reconnectTimeout=setTimeout(()=>{this.reconnectTimeout=null,this.connect()},this.RECONNECT_DELAY))}}const v=new p;export{v as discordGatewayService};