fix: cn.overleaf.com

Author: Junyi-99

webapp/_webapp/src/background.ts

@@ -26,12 +26,11 @@ export type Handler<A, T> = {
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 export type HandlerAny = Handler<any, any>;
 
-export const getCookiesHandler: Handler<void, { session?: string; gclb?: string }> = {
+export const getCookiesHandler: Handler<string, { session?: string; gclb?: string }> = {
   name: HANDLER_NAMES.GET_COOKIES,
-  handler: async (_, sendResponse) => {
-    const currentDomain = window.location.hostname;
-    const cookies = await getAllCookies("overleaf_session2", "https://" + currentDomain);
-    const gclb = await getAllCookies("GCLB", "https://" + currentDomain);
+  handler: async (domain: string, sendResponse) => {
+    const cookies = await getAllCookies("overleaf_session2", "https://" + domain);
+    const gclb = await getAllCookies("GCLB", "https://" + domain);
     sendResponse({ session: cookies[0]?.value, gclb: gclb[0]?.value });
   },
 };

webapp/_webapp/src/components/message-entry-container/tools/paper-score-comment/index.tsx

@@ -28,7 +28,7 @@ export const PaperScoreCommentCard = ({ messageId, message, preparing, animated
   const [showFilters, setShowFilters] = useState(false);
 
   useEffect(() => {
-    getCookies().then((cookies) => {
+    getCookies(window.location.hostname).then((cookies) => {
       setOverleafSession(cookies.session);
       setGclb(cookies.gclb);
     });

webapp/_webapp/src/devtool/app.tsx

@@ -11,7 +11,7 @@ const App = () => {
   const [gclb, setGclb] = useState(localStorage.getItem("pd.auth.gclb") ?? "");
 
   useEffect(() => {
-    getCookies().then((cookies) => {
+    getCookies(window.location.hostname).then((cookies) => {
       setOverleafSession(cookies.session ?? localStorage.getItem("pd.auth.overleafSession") ?? "");
       setGclb(cookies.gclb ?? localStorage.getItem("pd.auth.gclb") ?? "");
     });

webapp/_webapp/src/hooks/useSendMessageStream.ts

@@ -97,7 +97,7 @@ export function useSendMessageStream() {
       }));
 
       if (import.meta.env.DEV && alwaysSyncProject) {
-        const { session, gclb } = await getCookies();
+        const { session, gclb } = await getCookies(window.location.hostname);
         await sync(
           user?.id || "",
           getProjectId(),
@@ -157,7 +157,7 @@ export function useSendMessageStream() {
         {
           sync: async () => {
             try {
-              const { session, gclb } = await getCookies();
+              const { session, gclb } = await getCookies(window.location.hostname);
               await sync(
                 user?.id || "",
                 getProjectId(),

webapp/_webapp/src/intermediate.ts

@@ -88,16 +88,16 @@ function makeFunction<A, T>(handlerName: string, opts?: MakeFunctionOpts): (args
   return fn;
 }
 
-let getCookies: () => Promise<{ session: string; gclb: string }>;
+let getCookies: (domain: string) => Promise<{ session: string; gclb: string }>;
 if (import.meta.env.DEV) {
-  getCookies = async () => {
+  getCookies = async (_: string) => {
     return {
       session: localStorage.getItem("pd.auth.overleafSession") ?? "",
       gclb: localStorage.getItem("pd.auth.gclb") ?? "",
     };
   };
 } else {
-  getCookies = makeFunction<void, { session: string; gclb: string }>(HANDLER_NAMES.GET_COOKIES);
+  getCookies = makeFunction<string, { session: string; gclb: string }>(HANDLER_NAMES.GET_COOKIES);
 }
 
 export { getCookies };

webapp/_webapp/src/stores/conversation/handlers/handleStreamError.ts

@@ -18,7 +18,7 @@ export async function handleStreamError(
   sendMessageStream: (message: string, selectedText: string) => Promise<void>,
 ) {
   try {
-    const { session, gclb } = await getCookies();
+    const { session, gclb } = await getCookies(window.location.hostname);
     if (streamError.errorMessage.includes("project is out of date")) {
       await sync(
         userId,

webapp/_webapp/src/views/login/login-with-overleaf.tsx

@@ -29,7 +29,7 @@ export default function LoginWithOverleaf({
       setLoginLoadingMessage("Please wait while we log you in...");
       setIsLoginLoading(true);
 
-      const { session } = await getCookies();
+      const { session } = await getCookies(window.location.hostname);
       const resp = await loginByOverleaf({ overleafToken: session });
 
       setToken(resp.token);

webapp/_webapp/src/views/settings/sections/account-settings.tsx

@@ -13,7 +13,7 @@ export const AccountSettings = () => {
   const [gclb, setGclb] = useState("");
 
   useEffect(() => {
-    getCookies().then((cookies) => {
+    getCookies(window.location.hostname).then((cookies) => {
       setOverleafSession(cookies.session);
       setGclb(cookies.gclb);
     });