fix: ytm login

src/components/login/google/YTM.tsx

@@ -4,24 +4,25 @@ import { Button, Avatar, Text, ActivityIndicator } from 'react-native-paper';
 import { WebView } from 'react-native-webview';
 import { useFocusEffect } from '@react-navigation/native';
 import CookieManager from '@react-native-cookies/cookies';
-import { get_option, get_current_user } from 'libmuse';
+import { get_current_user } from 'libmuse';
 import { useTranslation } from 'react-i18next';
 
 import { saveSecure as saveItem } from '@utils/ChromeStorageAPI';
 import { StorageKeys } from '@enums/Storage';
 import { User, UseYTMLogin } from './useYTMLogin';
-import { museStore } from '@utils/muse';
 import useCollapsible from '../useCollapsible';
+import { initMuse } from '@utils/muse';
 
 const jsCode = 'window.ReactNativeWebView.postMessage(document.cookie)';
-const auth = get_option('auth');
 
 const clearCookies = () => {
-  museStore.reset();
   saveItem(StorageKeys.YTMCOOKIES, null);
 };
 
-const checkYTM = async () => console.log(await get_current_user());
+const checkYTM = async () => {
+  await initMuse();
+  get_current_user().then(console.log).catch(console.log);
+};
 
 const Login = () => {
   const { t } = useTranslation();
@@ -35,7 +36,7 @@ const Login = () => {
 
   const onMessage = (event: any) => {
     const { data } = event.nativeEvent;
-    setCookies(data?.split(';'));
+    setCookies(data?.split('; '));
   };
 
   useFocusEffect(
@@ -125,9 +126,7 @@ const Explore = ({ ytmLogin }: Props) => {
     <LoggedInPage
       user={user}
       logout={() => {
-        saveItem(StorageKeys.YTMCOOKIES, null);
-        museStore.set('token', null);
-        auth.token = null;
+        saveItem(StorageKeys.YTMCOOKIES, null).then(initMuse);
         clear();
       }}
     />

src/stores/initializeStores.ts

@@ -7,7 +7,7 @@ import { initialize as initializeAppStore } from './appStore';
 import { initialize as initializeRegexStore } from './regexStore';
 import { initializeR128Gain } from '../utils/ffmpeg/r128Store';
 import { dataSaverPlaylist, initCache } from '../utils/Cache';
-import { initialize as initializeMuse } from '@utils/muse';
+import { initMuse } from '../utils/muse';
 import { useAPM } from './usePersistStore';
 
 const { NoxModule } = NativeModules;
@@ -42,7 +42,7 @@ export const initializeStores = async ({
     default:
       break;
   }
-  await initializeMuse();
+  await initMuse();
   await initializeAppStore();
   await initializeRegexStore();
   await initializeR128Gain();

src/utils/muse.ts

@@ -1,34 +1,33 @@
-import { setup, MemoryStore } from 'libmuse';
+import * as crypto from 'expo-crypto';
+import { get_option } from 'libmuse';
 
-import {
-  getSecure as getItem,
-  saveSecure as setItem,
-} from '@utils/ChromeStorageAPI';
+import { getSecure as getItem } from '@utils/ChromeStorageAPI';
+import { StorageKeys } from '@enums/Storage';
 
-const MUSE_KEY = 'museStore';
-class MyStore extends MemoryStore {
-  set(key: string, value: unknown) {
-    super.set(key, value);
-    setItem(MUSE_KEY, JSON.stringify(Object.fromEntries(this.map)));
-  }
-  async init() {
-    const v = await getItem(MUSE_KEY);
-    try {
-      this.map = new Map(Object.entries(JSON.parse(v ?? '{}')));
-    } catch {
-      this.map = new Map();
-    }
-  }
-
-  reset() {
-    this.map = new Map();
-    setItem(MUSE_KEY, null);
-  }
-}
-
-export const museStore = new MyStore();
+const findCookie = (cookies: string, match: string) => {
+  const cookieArr = cookies.split('; ').map(v => v.split('='));
+  return cookieArr.find(cookie => cookie[0] === match)?.[1];
+};
 
-export const initialize = async () => {
-  await museStore.init();
-  setup({ store: museStore });
+export const initMuse = async (
+  cookies: Promise<string> = getItem(StorageKeys.YTMCOOKIES),
+) => {
+  const sapisid = findCookie(await cookies, 'SAPISID') ?? '';
+  // https://github.com/sigma67/ytmusicapi/blob/9ce284a7eae9c4cdc04bb098f7549cc5f1c80e22/ytmusicapi/helpers.py#L52
+  const get_headers = async () => {
+    const now = new Date();
+    const timems = now.getTime() + now.getTimezoneOffset() * 60 * 1000;
+    const timesec = Math.round(timems / 1000);
+    const SAPISIDHASH = await crypto.digestStringAsync(
+      crypto.CryptoDigestAlgorithm.SHA1,
+      `${timesec} ${sapisid} https://music.youtube.com`,
+    );
+    return {
+      Authorization: `SAPISIDHASH ${timesec}_${SAPISIDHASH}`,
+      Cookie: await cookies,
+    };
+  };
+  const auth = get_option('auth');
+  auth.get_headers = get_headers;
+  auth.requires_login = async () => false;
 };