feat: saves twitter cookies to envs

electron/main.js

@@ -311,7 +311,8 @@ const createMainWindow = async () => {
 
     try {
       await scraper.login(username, password, email);
-      return { success: true };
+      const cookies = await scraper.getCookies();
+      return { success: true, cookies };
     } catch (error) {
       console.error('Twitter login error:', error);
       return { success: false, error: error.message };

frontend/components/SetupPage/SetupYourAgent/SetupYourAgent.tsx

@@ -118,18 +118,20 @@ const SetupYourAgentForm = ({ serviceTemplate }: SetupYourAgentFormProps) => {
 
         // validate the twitter credentials
         setSubmitButtonText('Validating Twitter credentials...');
-        const isTwitterCredentialsValid = electronApi?.validateTwitterLogin
-          ? await validateTwitterCredentials(
-              values.xEmail,
-              values.xUsername,
-              values.xPassword,
-              electronApi.validateTwitterLogin,
-            )
-          : false;
+        const { isValid: isTwitterCredentialsValid, cookies } =
+          electronApi?.validateTwitterLogin
+            ? await validateTwitterCredentials(
+                values.xEmail,
+                values.xUsername,
+                values.xPassword,
+                electronApi.validateTwitterLogin,
+              )
+            : { isValid: false };
         setTwitterCredentialsValidationStatus(
           isTwitterCredentialsValid ? 'valid' : 'invalid',
         );
         if (!isTwitterCredentialsValid) return;
+        if (!cookies) return;
 
         // wait for agent setup to complete
         setSubmitButtonText('Setting up agent...');
@@ -151,6 +153,10 @@ const SetupYourAgentForm = ({ serviceTemplate }: SetupYourAgentFormProps) => {
               ...serviceTemplate.env_variables.TWIKIT_PASSWORD,
               value: values.xPassword,
             },
+            TWIKIT_COOKIES: {
+              ...serviceTemplate.env_variables.TWIKIT_COOKIES,
+              value: cookies,
+            },
             GENAI_API_KEY: {
               ...serviceTemplate.env_variables.GENAI_API_KEY,
               value: values.geminiApiKey,

frontend/components/SetupPage/SetupYourAgent/validation.ts

@@ -1,6 +1,7 @@
 import { ServiceTemplate } from '@/client';
 import { StakingProgramId } from '@/enums/StakingProgram';
 import { ServicesService } from '@/service/Services';
+import { XCookie } from '@/types/Cookies';
 
 /**
  * Validate the Google Gemini API key
@@ -21,6 +22,14 @@ export const validateGeminiApiKey = async (apiKey: string) => {
   }
 };
 
+const formatXCookies = (cookiesArray: XCookie[]) => {
+  const cookiesObject: Record<string, string> = {};
+  cookiesArray.forEach((cookie) => {
+    cookiesObject[cookie.key] = cookie.value;
+  });
+  return JSON.stringify(cookiesObject);
+};
+
 /**
  * Validate the Twitter credentials
  */
@@ -36,25 +45,26 @@ export const validateTwitterCredentials = async (
     email: string;
     username: string;
     password: string;
-  }) => Promise<{ success: boolean }>,
-) => {
-  if (!email || !username || !password) return false;
+  }) => Promise<{ success: boolean; cookies?: XCookie[] }>,
+): Promise<{ isValid: boolean; cookies?: string }> => {
+  if (!email || !username || !password) return { isValid: false };
 
   try {
-    const isValidated = await validateTwitterLogin({
+    const result = await validateTwitterLogin({
       username,
       password,
       email,
     });
-    if (isValidated.success) {
-      return true;
+
+    if (result.success && result.cookies) {
+      return { isValid: true, cookies: formatXCookies(result.cookies) };
     }
 
-    console.error('Error validating Twitter credentials:', isValidated);
-    return false;
+    console.error('Error validating Twitter credentials:', result);
+    return { isValid: false };
   } catch (error) {
     console.error('Unexpected error validating Twitter credentials:', error);
-    return false;
+    return { isValid: false };
   }
 };
 

frontend/constants/serviceTemplates.ts

@@ -189,7 +189,7 @@ export const SERVICE_TEMPLATES: ServiceTemplate[] = [
         provision_type: EnvProvisionType.COMPUTED,
       },
       CELO_LEDGER_RPC: {
-        name: 'Base ledger RPC',
+        name: 'Celo ledger RPC',
         description: '',
         value: '',
         provision_type: EnvProvisionType.COMPUTED,
@@ -212,6 +212,12 @@ export const SERVICE_TEMPLATES: ServiceTemplate[] = [
         value: '',
         provision_type: EnvProvisionType.USER,
       },
+      TWIKIT_COOKIES: {
+        name: 'Twitter cookies',
+        description: '',
+        value: '',
+        provision_type: EnvProvisionType.USER,
+      },
       GENAI_API_KEY: {
         name: 'Gemini api key',
         description: '',

frontend/context/ElectronApiProvider.tsx

@@ -1,6 +1,7 @@
 import { get } from 'lodash';
 import { createContext, PropsWithChildren } from 'react';
 
+import { XCookie } from '@/types/Cookies';
 import { ElectronStore, ElectronTrayIconStatus } from '@/types/ElectronApi';
 
 type ElectronApiContextProps = {
@@ -40,7 +41,7 @@ type ElectronApiContextProps = {
     username: string;
     password: string;
     email: string;
-  }) => Promise<{ success: boolean }>;
+  }) => Promise<{ success: boolean; cookies?: XCookie[] }>;
 };
 
 export const ElectronApiContext = createContext<ElectronApiContextProps>({

frontend/types/Cookies.ts

@@ -0,0 +1,14 @@
+/**
+ * Cookies returned by agent-twitter-client
+ * when log in
+ */
+export type XCookie = {
+  key: string;
+  value: string;
+  domain?: string;
+  path?: string;
+  secure?: boolean;
+  httpOnly?: boolean;
+  sameSite?: string;
+  expires?: string;
+};