vtsuru.live/src/client/data/initialize.ts

import { isLoggedIn, useAccount } from "@/api/account";
import { attachConsole, info, warn } from "@tauri-apps/plugin-log";
import { useSettings } from "../store/useSettings";
import { useWebFetcher } from "@/store/useWebFetcher";
import { useBiliCookie } from "../store/useBiliCookie";
import { getBuvid, getRoomKey } from "./utils";
import { initInfo } from "./info";
import { TrayIcon, TrayIconOptions } from '@tauri-apps/api/tray';
import { Menu } from "@tauri-apps/api/menu";
import { getCurrentWindow } from "@tauri-apps/api/window";
import {
  isPermissionGranted,
  onAction,
  requestPermission,
  sendNotification,
} from '@tauri-apps/plugin-notification';
import { openUrl } from "@tauri-apps/plugin-opener";
import { CN_HOST } from "@/data/constants";
import { invoke } from "@tauri-apps/api/core";
import { check } from '@tauri-apps/plugin-updater';
import { relaunch } from '@tauri-apps/plugin-process';

const accountInfo = useAccount();

export const clientInited = ref(false);
let tray: TrayIcon;
export async function initAll() {
  if (clientInited.value) {
    return;
  }
  let permissionGranted = await isPermissionGranted();

  // If not we need to request it
  if (!permissionGranted) {
    const permission = await requestPermission();
    permissionGranted = permission === 'granted';
    if (permissionGranted) {
      info('Notification permission granted');
    }

  }
  initNotificationHandler();
  const detach = await attachConsole();
  const settings = useSettings();
  const biliCookie = useBiliCookie();
  await settings.init();
  info('[init] 已加载账户信息');
  biliCookie.init();
  info('[init] 已加载bilibili cookie');
  initInfo();
  info('[init] 开始更新数据');

  if (isLoggedIn && accountInfo.value.isBiliVerified) {
    const danmakuInitNoticeRef = window.$notification.info({
      title: '正在初始化弹幕客户端...',
      closable: false
    });
    const result = await initDanmakuClient();
    danmakuInitNoticeRef.destroy();
    if (result.success) {
      window.$notification.success({
        title: '弹幕客户端初始化完成',
        duration: 3000
      });
    } else {
      window.$notification.error({
        title: '弹幕客户端初始化失败: ' + result.message,
      });
    }
  }
  info('[init] 已加载弹幕客户端');
  // 初始化系统托盘图标和菜单
  const menu = await Menu.new({
    items: [
      {
        id: 'quit',
        text: '退出',
        action: () => {
          invoke('quit_app');
        },
      },
    ],
  });
  const iconData = await (await fetch('https://oss.suki.club/vtsuru/icon.ico')).arrayBuffer();
  const appWindow = getCurrentWindow();
  const options: TrayIconOptions = {
    // here you can add a tray menu, title, tooltip, event handler, etc
    menu: menu,
    title: 'VTsuru.Client',
    tooltip: 'VTsuru 事件收集器',
    icon: iconData,
    action: (event) => {

      switch (event.type) {
        case 'DoubleClick':
          appWindow.show();
          break;
          case 'Click':
          appWindow.show();
          break;
        }
    }
  };


  tray = await TrayIcon.new(options);

  clientInited.value = true;
}
export function OnClientUnmounted() {
  if (clientInited.value) {
    clientInited.value = false;
  }

  tray.close();
}

async function checkUpdate() {
  const update = await check();
  if (update) {
    console.log(
      `found update ${update.version} from ${update.date} with notes ${update.body}`
    );
    let downloaded = 0;
    let contentLength = 0;
    // alternatively we could also call update.download() and update.install() separately
    await update.downloadAndInstall((event) => {
      switch (event.event) {
        case 'Started':
          contentLength = event.data.contentLength || 0;
          console.log(`started downloading ${event.data.contentLength} bytes`);
          break;
        case 'Progress':
          downloaded += event.data.chunkLength;
          console.log(`downloaded ${downloaded} from ${contentLength}`);
          break;
        case 'Finished':
          console.log('download finished');
          break;
      }
    });

    console.log('update installed');
    await relaunch();
  }
}

export const isInitedDanmakuClient = ref(false);
export const isInitingDanmakuClient = ref(false);
export async function initDanmakuClient() {
  const biliCookie = useBiliCookie();
  const settings = useSettings();
  if (isInitedDanmakuClient.value || isInitingDanmakuClient.value) {
    return { success: true, message: '' };
  }
  isInitingDanmakuClient.value = true;
  let result = { success: false, message: '' };
  try {
    if (isLoggedIn) {
      if (settings.settings.useDanmakuClientType === 'openlive') {
        result = await initOpenLive();
      } else {
        const cookie = await biliCookie.getBiliCookie();
        if (!cookie) {
          if (settings.settings.fallbackToOpenLive) {
            settings.settings.useDanmakuClientType = 'openlive';
            settings.save();
            info('未设置bilibili cookie, 根据设置切换为openlive');
            result = await initOpenLive();
          } else {
            info('未设置bilibili cookie, 跳过弹幕客户端初始化');
            window.$notification.warning({
              title: '未设置bilibili cookie, 跳过弹幕客户端初始化',
              duration: 5,
            });
            result = { success: false, message: '未设置bilibili cookie' };
          }
        } else {
          const resp = await callStartDanmakuClient();
          if (!resp?.success) {
            warn('加载弹幕客户端失败: ' + resp?.message);
            result = { success: false, message: resp?.message };
          } else {
            info('已加载弹幕客户端');
            result = { success: true, message: '' };
          }
        }
      }
    }
    return result;
  } catch (err) {
    warn('加载弹幕客户端失败: ' + err);
    return { success: false, message: '加载弹幕客户端失败' };
  } finally {
    if (result) {
      isInitedDanmakuClient.value = true;
    }
    isInitingDanmakuClient.value = false;
  }
}
export async function initOpenLive() {
  const reuslt = await callStartDanmakuClient();
  if (reuslt?.success == true) {
    info('已加载弹幕客户端 [openlive]');
  } else {
    warn('加载弹幕客户端失败 [openlive]: ' + reuslt?.message);
  }
  return reuslt;
}
function initNotificationHandler(){
  onAction((event) => {
    if (event.extra?.type === 'question-box') {
      openUrl(CN_HOST + '/manage/question-box');
    }
  });
}

export async function callStartDanmakuClient() {
  const biliCookie = useBiliCookie();
  const settings = useSettings();
  const webFetcher = useWebFetcher();
  if (settings.settings.useDanmakuClientType === 'direct') {
    const key = await getRoomKey(
      accountInfo.value.biliRoomId!, await biliCookie.getBiliCookie() || '');
    if (!key) {
      warn('获取房间密钥失败, 无法连接弹幕客户端');
      return { success: false, message: '无法获取房间密钥' };
    }
    const buvid = await getBuvid();
    if (!buvid) {
      warn('获取buvid失败, 无法连接弹幕客户端');
      return { success: false, message: '无法获取buvid' };
    }
    return await webFetcher.Start('direct', {
      roomId: accountInfo.value.biliRoomId!,
      buvid: buvid.data,
      token: key,
      tokenUserId: biliCookie.uId!,
    }, true);
  } else {
    return await webFetcher.Start('openlive', undefined, true);
  }
}