changes to work with new noteSettings format

codex-team · Dec 5, 2023 · f7347ef · f7347ef
1 parent 1bb2048
commit f7347ef
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/src/domain/service/noteSettings.ts b/src/domain/service/noteSettings.ts
@@ -4,6 +4,7 @@ import type NoteSettingsRepository from '@repository/noteSettings.repository.js'
 import type TeamRepository from '@repository/team.repository.js';
 import type { MemberRole, Team, TeamMember, TeamMemberCreationAttributes } from '@domain/entities/team.js';
 import type User from '@domain/entities/user.js';
+import { createInvitationHash } from '@infrastructure/utils/invitationHash.js';
 
 /**
  * Service responsible for Note Settings
@@ -47,6 +48,7 @@ export default class NoteSettingsService {
     return await this.noteSettingsRepository.addNoteSettings({
       noteId: noteId,
       isPublic: isPublic,
+      invitationHash: createInvitationHash(),
     });
   }
 

diff --git a/src/repository/storage/postgres/orm/sequelize/noteSettings.ts b/src/repository/storage/postgres/orm/sequelize/noteSettings.ts
@@ -28,6 +28,11 @@ export class NoteSettingsModel extends Model<InferAttributes<NoteSettingsModel>,
    * Is note public
    */
   public declare isPublic: CreationOptional<NoteSettings['isPublic']>;
+
+  /**
+   * Invitation hash
+   */
+  public declare invitationHash: NoteSettings['invitationHash'];
 }
 
 /**
@@ -88,6 +93,10 @@ export default class NoteSettingsSequelizeStorage {
         allowNull: false,
         defaultValue: true,
       },
+      invitationHash: {
+        type: DataTypes.STRING,
+        allowNull: false,
+      },
     }, {
       tableName: this.tableName,
       sequelize: this.database,
@@ -169,12 +178,14 @@ export default class NoteSettingsSequelizeStorage {
     noteId,
     customHostname,
     isPublic,
+    invitationHash,
   }: NoteSettingsCreationAttributes
   ): Promise<NoteSettings> {
     const settings = await this.model.create({
       noteId,
       customHostname,
       isPublic,
+      invitationHash,
     });
 
     return settings;

diff --git a/src/tests/utils/insert-data.ts b/src/tests/utils/insert-data.ts
@@ -44,7 +44,7 @@ async function insertNotes(db: SequelizeOrm): Promise<void> {
  */
 async function insertNoteSettings(db: SequelizeOrm): Promise<void> {
   for (const noteSetting of noteSettings) {
-    await db.connection.query(`INSERT INTO public.note_settings (id, "note_id", "custom_hostname", "is_public") VALUES (${noteSetting.id}, '${noteSetting.note_id}', '${noteSetting.custom_hostname}', ${noteSetting.is_public})`);
+    await db.connection.query(`INSERT INTO public.note_settings (id, "note_id", "custom_hostname", "is_public", "invitation_hash") VALUES (${noteSetting.id}, '${noteSetting.note_id}', '${noteSetting.custom_hostname}', ${noteSetting.is_public}, '${noteSetting.invitation_hash}')`);
   }
 }