diff --git a/src/becca/entities/rows.ts b/src/becca/entities/rows.ts index 4428b6dde..119d392ec 100644 --- a/src/becca/entities/rows.ts +++ b/src/becca/entities/rows.ts @@ -91,7 +91,8 @@ export interface BranchRow { * end user. Those types should be used only for checking against, they are * not for direct use. */ -export type NoteType = ("file" | "image" | "search" | "noteMap" | "launcher" | "doc" | "contentWidget" | "text" | "relationMap" | "render" | "canvas" | "mermaid" | "book" | "webView" | "code"); +export const ALLOWED_NOTE_TYPES = [ "file", "image", "search", "noteMap", "launcher", "doc", "contentWidget", "text", "relationMap", "render", "canvas", "mermaid", "book", "webView", "code" ] as const; +export type NoteType = typeof ALLOWED_NOTE_TYPES[number]; export interface NoteRow { noteId: string; @@ -106,5 +107,5 @@ export interface NoteRow { dateModified: string; utcDateCreated: string; utcDateModified: string; - content?: string; + content?: string | Buffer; } diff --git a/src/services/import/zip.ts b/src/services/import/zip.ts index 93ccc2da9..0c4008ddf 100644 --- a/src/services/import/zip.ts +++ b/src/services/import/zip.ts @@ -20,7 +20,7 @@ import BNote = require('../../becca/entities/bnote'); import NoteMeta = require('../meta/note_meta'); import AttributeMeta = require('../meta/attribute_meta'); import { Stream } from 'stream'; -import { NoteType } from '../../becca/entities/rows'; +import { ALLOWED_NOTE_TYPES, NoteType } from '../../becca/entities/rows'; interface MetaFile { files: NoteMeta[] @@ -499,10 +499,6 @@ async function importZip(taskContext: TaskContext, fileBuffer: Buffer, importRoo } } else { - if (typeof content !== "string") { - throw new Error("Incorrect content type."); - } - ({note} = noteService.createNewNote({ parentNoteId: parentNoteId, title: noteTitle || "", @@ -653,7 +649,11 @@ function resolveNoteType(type: string | undefined): NoteType { return 'webView'; } - return "text"; + if (type && (ALLOWED_NOTE_TYPES as readonly string[]).includes(type)) { + return type as NoteType; + } else { + return "text"; + } } export = { diff --git a/src/services/note-interface.ts b/src/services/note-interface.ts index 0b99c31ef..2cfa0b86a 100644 --- a/src/services/note-interface.ts +++ b/src/services/note-interface.ts @@ -7,7 +7,7 @@ export interface NoteParams { parentNoteId: string; templateNoteId?: string; title: string; - content: string; + content: string | Buffer; /** text, code, file, image, search, book, relationMap, canvas, webView */ type: NoteType; /** default value is derived from default mimes for type */ diff --git a/src/services/search/expressions/note_content_fulltext.ts b/src/services/search/expressions/note_content_fulltext.ts index ada9705a2..02626bf16 100644 --- a/src/services/search/expressions/note_content_fulltext.ts +++ b/src/services/search/expressions/note_content_fulltext.ts @@ -74,7 +74,7 @@ class NoteContentFulltextExp extends Expression { } if (isProtected) { - if (!protectedSessionService.isProtectedSessionAvailable() || !content) { + if (!protectedSessionService.isProtectedSessionAvailable() || !content || typeof content !== "string") { return; } @@ -86,7 +86,7 @@ class NoteContentFulltextExp extends Expression { } } - if (!content) { + if (!content || typeof content !== "string") { return; }