feat: add support for import id

Author: capJavert

bin/importProfileFromJSON.ts

@@ -7,6 +7,7 @@ import { QueryFailedError, type DataSource } from 'typeorm';
 import { readFile, stat, readdir } from 'node:fs/promises';
 import { importUserExperienceFromJSON } from '../src/common/profile/import';
 import path from 'node:path';
+import { randomUUID } from 'node:crypto';
 
 /**
  * Import profile from JSON to user by id
@@ -28,16 +29,22 @@ const main = async () => {
           type: 'string',
           short: 'l',
         },
+        uid: {
+          type: 'string',
+        },
       },
     });
 
     const paramsSchema = z.object({
       path: z.string().nonempty(),
       limit: z.coerce.number().int().positive().default(10),
+      uid: z.string().nonempty().default(randomUUID()),
     });
 
     const params = paramsSchema.parse(values);
 
+    console.log('Starting import with ID:', params.uid);
+
     con = await createOrGetConnection();
 
     const pathStat = await stat(params.path);
@@ -79,6 +86,7 @@ const main = async () => {
           con: con.manager,
           dataJson: dataJSON,
           userId: 'testuser',
+          importId: params.uid,
         });
       } catch (error) {
         failedImports += 1;

src/common/profile/import.ts

@@ -141,10 +141,12 @@ export const importUserExperienceWork = async ({
     skills,
     ended_at: endedAt,
     location,
+    flags,
   } = userExperience;
 
   const insertResult = await con.getRepository(UserExperienceWork).insert(
     con.getRepository(UserExperienceWork).create({
+      flags,
       userId: userId,
       ...(await resolveUserCompanyPart({
         name: company,
@@ -202,10 +204,12 @@ export const importUserExperienceEducation = async ({
     ended_at: endedAt,
     location,
     subtitle,
+    flags,
   } = userExperience;
 
   const insertResult = await con.getRepository(UserExperienceEducation).insert(
     con.getRepository(UserExperienceEducation).create({
+      flags,
       userId: userId,
       ...(await resolveUserCompanyPart({
         name: company,
@@ -250,12 +254,14 @@ export const importUserExperienceCertification = async ({
     title,
     started_at: startedAt,
     ended_at: endedAt,
+    flags,
   } = userExperience;
 
   const insertResult = await con
     .getRepository(UserExperienceCertification)
     .insert(
       con.getRepository(UserExperienceCertification).create({
+        flags,
         userId: userId,
         ...(await resolveUserCompanyPart({
           name: company,
@@ -291,10 +297,12 @@ export const importUserExperienceProject = async ({
     started_at: startedAt,
     ended_at: endedAt,
     skills,
+    flags,
   } = userExperience;
 
   const insertResult = await con.getRepository(UserExperienceProject).insert(
     con.getRepository(UserExperienceProject).create({
+      flags,
       userId: userId,
       title,
       description,
@@ -318,10 +326,12 @@ export const importUserExperienceFromJSON = async ({
   con,
   dataJson,
   userId,
+  importId,
 }: {
   con: EntityManager;
   dataJson: unknown;
   userId: string;
+  importId?: string;
 }) => {
   if (!userId) {
     throw new Error('userId is required');
@@ -352,26 +362,31 @@ export const importUserExperienceFromJSON = async ({
 
   await con.transaction(async (entityManager) => {
     for (const item of data) {
-      switch (item.type) {
+      const importData = {
+        ...item,
+        flags: importId ? { import: importId } : undefined,
+      };
+
+      switch (importData.type) {
         case UserExperienceType.Work:
           await importUserExperienceWork({
-            data: item,
+            data: importData,
             con: entityManager,
             userId,
           });
 
           break;
         case UserExperienceType.Education:
           await importUserExperienceEducation({
-            data: item,
+            data: importData,
             con: entityManager,
             userId,
           });
 
           break;
         case UserExperienceType.Certification:
           await importUserExperienceCertification({
-            data: item,
+            data: importData,
             con: entityManager,
             userId,
           });
@@ -381,14 +396,14 @@ export const importUserExperienceFromJSON = async ({
         case UserExperienceType.OpenSource:
         case UserExperienceType.Volunteering:
           await importUserExperienceProject({
-            data: item,
+            data: importData,
             con: entityManager,
             userId,
           });
 
           break;
         default:
-          throw new Error(`Unsupported experience type: ${item.type}`);
+          throw new Error(`Unsupported experience type: ${importData.type}`);
       }
     }
   });

src/common/schema/profile.ts

@@ -121,6 +121,7 @@ export const userExperienceWorkImportSchema = z.object({
       country: z.string().nullish(),
     })
     .nullish(),
+  flags: z.object({ import: z.string() }).partial().optional(),
 });
 
 export const userExperienceEducationImportSchema = z.object({
@@ -145,6 +146,7 @@ export const userExperienceEducationImportSchema = z.object({
     .nullish()
     .transform((n) => (n === null ? undefined : n)),
   subtitle: z.string().nullish(),
+  flags: z.object({ import: z.string() }).partial().optional(),
 });
 
 export const userExperienceCertificationImportSchema = z.object({
@@ -157,6 +159,7 @@ export const userExperienceCertificationImportSchema = z.object({
     .default('Certification'),
   started_at: z.coerce.date().default(() => new Date()),
   ended_at: z.coerce.date().nullish().default(null),
+  flags: z.object({ import: z.string() }).partial().optional(),
 });
 
 export const userExperienceProjectImportSchema = z.object({
@@ -173,4 +176,5 @@ export const userExperienceProjectImportSchema = z.object({
     .array(z.string())
     .nullish()
     .transform((n) => (n === null ? undefined : n)),
+  flags: z.object({ import: z.string() }).partial().optional(),
 });