✨ Feat: Add user authentication for seed command (#541)

* ✨ Feat: Add user authentication for seed command

* ✨ Feat: Add test event for seed command

* ✨ Feat: Move auth logic to auth service

* ✨ Feat: Simplify event creation in seed command

* 🐛Fix: Use UserId instead of email for creating user session

* ✨ Feat: Call auth service for session creation

* 🐛Fix: localhost env variable

* 🧹 Chore: Correct capitalization in seed command warning

* 🔧 Fix: Validate user input as ObjectId in seed command

* 🔧 Fix: Update user option syntax in seed command for clarity

---------

Co-authored-by: Tyler Dane <tyler@switchback.tech>

Author: claudiahdz

packages/backend/.env.example

@@ -60,6 +60,7 @@ SUPERTOKENS_KEY=UNIQUE_KEY_FROM_YOUR_SUPERTOKENS_ACCOUNT
 ####################################################
 # Set these values to save time while using the CLI
 
+LOCAL_DOMAIN=localhost:3000
 STAGING_DOMAIN=staging.yourdomain.com
 PROD_DOMAIN=app.yourdomain.com
 

packages/backend/src/auth/controllers/auth.controller.ts

@@ -49,20 +49,12 @@ class AuthController {
     }
 
     if (cUserId) {
-      const sUserId = supertokens.convertToRecipeUserId(cUserId);
-      await Session.createNewSession(req, res, "public", sUserId);
-    }
-
-    const user = await findCompassUserBy("_id", cUserId);
-
-    if (!user) {
+      await compassAuthService.createSessionForUser(cUserId);
+    } else {
       res.promise({ error: "User doesn't exist" });
       return;
     }
 
-    const sUserId = supertokens.convertToRecipeUserId(user._id.toString());
-    await Session.createNewSession(req, res, "public", sUserId);
-
     res.promise({
       message: `User session created for ${cUserId}`,
     });

packages/backend/src/auth/services/compass.auth.service.ts

@@ -1,10 +1,14 @@
+import supertokens from "supertokens-node";
 import Session from "supertokens-node/recipe/session";
+import { Logger } from "@core/logger/winston.logger";
 import { error } from "@backend/common/errors/handlers/error.handler";
 import { SyncError } from "@backend/common/errors/sync/sync.errors";
 import { getSync } from "@backend/sync/util/sync.queries";
 import { canDoIncrementalSync } from "@backend/sync/util/sync.util";
 import { findCompassUserBy } from "@backend/user/queries/user.queries";
 
+const logger = Logger("app:auth.service");
+
 class CompassAuthService {
   determineAuthMethod = async (gUserId: string) => {
     const user = await findCompassUserBy("google.googleId", gUserId);
@@ -28,6 +32,29 @@ class CompassAuthService {
     return { authMethod, user };
   };
 
+  createSessionForUser = async (cUserId: string) => {
+    const userId = cUserId;
+    const sUserId = supertokens.convertToRecipeUserId(cUserId);
+
+    try {
+      const session = await Session.createNewSessionWithoutRequestResponse(
+        "public",
+        sUserId,
+      );
+      const accessToken = session.getAccessToken();
+      logger.info(`user session created for ${userId}`);
+      return {
+        sessionHandle: session.getHandle(),
+        userId: sUserId,
+        compassUserId: userId,
+        accessToken,
+      };
+    } catch (err) {
+      logger.error("Error creating session:", err);
+      throw new Error("Failed to create session");
+    }
+  };
+
   revokeSessionsByUser = async (userId: string) => {
     const sessionsRevoked = await Session.revokeAllSessionsForUser(userId);
     return { sessionsRevoked: sessionsRevoked.length };

packages/scripts/src/cli.ts

@@ -1,7 +1,7 @@
 // sort-imports-ignore
 import { Command } from "commander";
 import "./init";
-// eslint-disable-next-line prettier/prettier
+
 import { CliValidator } from "./cli.validator";
 import { runBuild } from "./commands/build";
 import { startDeleteFlow } from "./commands/delete";
@@ -86,10 +86,7 @@ class CompassCli {
     program
       .command("seed")
       .description("seed the database with events")
-      .option(
-        "-u, --user [id | email]",
-        "specify which user to seed events for",
-      );
+      .option("-u, --user <id>", "specify which user to seed events for");
     return program;
   }
 }

packages/scripts/src/commands/seed.ts

@@ -1,32 +1,73 @@
+import axios from "axios";
+import dayjs from "dayjs";
 import pkg from "inquirer";
-import { log } from "@scripts/common/cli.utils";
+import { ObjectId } from "mongodb";
+import { getApiBaseUrl, log } from "@scripts/common/cli.utils";
+import { Schema_Event } from "@core/types/event.types";
+import { createMockStandaloneEvent } from "@core/util/test/ccal.event.factory";
+import compassAuthService from "@backend/auth/services/compass.auth.service";
 import mongoService from "@backend/common/services/mongo.service";
 import { findCompassUserBy } from "@backend/user/queries/user.queries";
 
 const { prompt } = pkg;
 
+async function createEvent(
+  events: Schema_Event[],
+  baseUrl: string,
+  accessToken: string,
+) {
+  await axios.post(`${baseUrl}/event`, events, {
+    headers: { Cookie: `sAccessToken=${accessToken}` },
+  });
+}
+
 async function seedEvents(userInput: string) {
   try {
+    // Validate userInput as ObjectId
+    if (!ObjectId.isValid(userInput)) {
+      log.error(
+        `Provided user id is not a valid ObjectId: ${userInput}.
+        Please provide your Compass user's _id value.`,
+      );
+      process.exit(1);
+    }
+
     // Connect to MongoDB
     await mongoService.waitUntilConnected();
 
-    // Determine if input is email or ID and get the user ID
-    const isEmail = userInput.includes("@");
-    const user = await findCompassUserBy(isEmail ? "email" : "_id", userInput);
+    const user = await findCompassUserBy("_id", userInput);
 
     if (!user) {
-      log.error(
-        `User not found with ${isEmail ? "email" : "ID"}: ${userInput}`,
-      );
+      log.error(`User not found with Compass ID: ${userInput}`);
       process.exit(1);
     }
 
     const userId = user._id.toString();
-    log.info(`Running seed command for user: ${userId}...`);
+    log.info(`Running seed command for user: ${userId} (${user.email})...`);
+
+    // Creates user session
+    const { accessToken } =
+      await compassAuthService.createSessionForUser(userId);
+    const baseUrl = await getApiBaseUrl("local");
+
+    // Test Event
+    const eventOverrides = {
+      user: userId,
+      isAllDay: false,
+      isSomeday: false,
+      startDate: dayjs().hour(10).minute(0).second(0).toISOString(),
+      endDate: dayjs().hour(11).minute(0).second(0).toISOString(),
+    };
+    const event = createMockStandaloneEvent(eventOverrides);
+    const events: Schema_Event[] = [event];
+
+    await createEvent(events, baseUrl, accessToken);
 
-    // TODO: Add logic to create events
+    // TODO: Create a variety of events for seeding
 
-    log.success(`Successfully created events for user: ${userInput}`);
+    log.success(
+      `Successfully created events for user: ${user.email} with id: ${userId}`,
+    );
   } catch (error) {
     log.error("Failed to seed events:");
     console.error(error);
@@ -48,8 +89,8 @@ export async function runSeed(userInput: string, force = false) {
     "⚠️  WARNING ⚠️",
     "",
     "This command will:",
-    "• Create multiple events in your Compass calendar database",
-    "• Create multiple events in your primary Google calendar",
+    "• Create multiple events in your Compass Calendar database",
+    "• Create multiple events in your primary Google Calendar",
     "",
     "🔔 RECOMMENDATION:",
     "It's strongly recommended to use this command with a test account",

packages/scripts/src/common/cli.constants.ts

@@ -15,6 +15,7 @@ export const CATEGORY_VM = {
 };
 
 export const CLI_ENV = {
+  LOCAL_DOMAIN: process.env["LOCAL_DOMAIN"] || `localhost:3000`,
   STAGING_DOMAIN: process.env["STAGING_DOMAIN"],
   PROD_DOMAIN: process.env["PROD_DOMAIN"],
 };

packages/scripts/src/common/cli.types.ts

@@ -21,4 +21,4 @@ export type Options_Cli = Options_Cli_Root &
   Options_Cli_Build &
   Options_Cli_Delete;
 
-export type Environment_Cli = "staging" | "production";
+export type Environment_Cli = "local" | "staging" | "production";

packages/scripts/src/common/cli.utils.ts

@@ -12,9 +12,9 @@ export const fileExists = (file: string) => {
 
 export const getApiBaseUrl = async (environment: Environment_Cli) => {
   const category = environment ? environment : await getEnvironmentAnswer();
-  const isStaging = category === "staging";
-  const domain = await getDomainAnswer(isStaging);
-  const baseUrl = `https://${domain}/api`;
+  const domain = await getDomainAnswer(category);
+  const baseUrl =
+    environment === "local" ? `http://${domain}/api` : `https://${domain}/api`;
 
   return baseUrl;
 };
@@ -41,7 +41,14 @@ export const getClientId = async (environment: Environment_Cli) => {
   throw Error("Invalid destination");
 };
 
-const getDomainAnswer = async (isStaging: boolean) => {
+const getDomainAnswer = async (env: string) => {
+  const isLocal = env === "local";
+  const isStaging = env === "staging";
+
+  if (isLocal && CLI_ENV.LOCAL_DOMAIN !== undefined) {
+    return CLI_ENV.LOCAL_DOMAIN;
+  }
+
   if (isStaging && CLI_ENV.STAGING_DOMAIN !== undefined) {
     return CLI_ENV.STAGING_DOMAIN;
   }