Files
ERitors-Scribe-Desktop/electron/database/models/Upload.ts
natreex e45a15225b Add enable/disable management for book tools (characters, worlds, and locations)
- Introduced toggling functionality for managing `characters`, `worlds`, and `locations` tool availability per book.
- Updated `CharacterComponent`, `WorldSetting`, and `LocationComponent` with toggle switches for tool enablement.
- Added `book_tools` database table and related schema migration for storing tool settings.
- Extended API calls, models, and IPC handlers to support tool enablement states.
- Localized new strings for English with supporting descriptions and messages.
- Adjusted conditional rendering logic across components to respect tool enablement.
2026-01-14 17:42:59 -05:00

264 lines
16 KiB
TypeScript

import { getUserEncryptionKey } from "../keyManager.js";
import System from "../System.js";
import { CompleteBook } from "./Book.js";
import BookRepo, { EritBooksTable, BookToolsTable } from "../repositories/book.repository.js";
import ActRepository, { BookActSummariesTable } from "../repositories/act.repository.js";
import GuidelineRepo, { BookAIGuideLineTable, BookGuideLineTable } from "../repositories/guideline.repository.js";
import ChapterRepo, {
BookChapterInfosTable,
BookChaptersTable
} from "../repositories/chapter.repository.js";
import CharacterRepo, {
BookCharactersAttributesTable,
BookCharactersTable
} from "../repositories/character.repository.js";
import IncidentRepository, { BookIncidentsTable } from "../repositories/incident.repository.js";
import IssueRepository, { BookIssuesTable } from "../repositories/issue.repository.js";
import LocationRepo, {
BookLocationTable,
LocationElementTable,
LocationSubElementTable
} from "../repositories/location.repository.js";
import PlotPointRepository, { BookPlotPointsTable } from "../repositories/plotpoint.repository.js";
import WorldRepository, {
BookWorldElementsTable,
BookWorldTable
} from "../repositories/world.repository.js";
import ChapterContentRepository, { BookChapterContentTable } from "../repositories/chaptercontent.repository.js";
export default class Upload {
/**
* Prepares a complete book with all related data for synchronization upload.
* Fetches all book-related tables from the database, decrypts encrypted fields
* using the user's encryption key, and returns a complete book object ready for sync.
*
* @param userId - The unique identifier of the user who owns the book
* @param bookId - The unique identifier of the book to upload
* @param lang - The language code for localization ("fr" or "en")
* @returns A promise that resolves to a CompleteBook object containing all decrypted book data
*/
static async uploadBookForSync(userId: string, bookId: string, lang: "fr" | "en"): Promise<CompleteBook> {
const userEncryptionKey: string = getUserEncryptionKey(userId);
const [
encryptedBooks,
encryptedActSummaries,
encryptedAIGuidelines,
encryptedChapters,
encryptedCharacters,
encryptedGuidelines,
encryptedIncidents,
encryptedIssues,
encryptedLocations,
encryptedPlotPoints,
encryptedWorlds,
bookToolsData
]: [
EritBooksTable[],
BookActSummariesTable[],
BookAIGuideLineTable[],
BookChaptersTable[],
BookCharactersTable[],
BookGuideLineTable[],
BookIncidentsTable[],
BookIssuesTable[],
BookLocationTable[],
BookPlotPointsTable[],
BookWorldTable[],
BookToolsTable | null
] = await Promise.all([
BookRepo.fetchEritBooksTable(userId, bookId, lang),
ActRepository.fetchBookActSummaries(userId, bookId, lang),
GuidelineRepo.fetchBookAIGuideLine(userId, bookId, lang),
ChapterRepo.fetchBookChapters(userId, bookId, lang),
CharacterRepo.fetchBookCharacters(userId, bookId, lang),
GuidelineRepo.fetchBookGuideLineTable(userId, bookId, lang),
IncidentRepository.fetchBookIncidents(userId, bookId, lang),
IssueRepository.fetchBookIssues(userId, bookId, lang),
LocationRepo.fetchBookLocations(userId, bookId, lang),
PlotPointRepository.fetchBookPlotPoints(userId, bookId, lang),
WorldRepository.fetchBookWorlds(userId, bookId, lang),
BookRepo.fetchBookTools(userId, bookId, lang)
]);
const [
nestedChapterContents,
nestedChapterInfos,
nestedCharacterAttributes,
nestedWorldElements,
nestedLocationElements
]: [
BookChapterContentTable[][],
BookChapterInfosTable[][],
BookCharactersAttributesTable[][],
BookWorldElementsTable[][],
LocationElementTable[][]
] = await Promise.all([
Promise.all(encryptedChapters.map((chapter: BookChaptersTable): Promise<BookChapterContentTable[]> =>
ChapterContentRepository.fetchBookChapterContents(userId, chapter.chapter_id, lang))),
Promise.all(encryptedChapters.map((chapter: BookChaptersTable): Promise<BookChapterInfosTable[]> =>
ChapterRepo.fetchBookChapterInfos(userId, chapter.chapter_id, lang))),
Promise.all(encryptedCharacters.map((character: BookCharactersTable): Promise<BookCharactersAttributesTable[]> =>
CharacterRepo.fetchBookCharactersAttributes(userId, character.character_id, lang))),
Promise.all(encryptedWorlds.map((world: BookWorldTable): Promise<BookWorldElementsTable[]> =>
WorldRepository.fetchBookWorldElements(userId, world.world_id, lang))),
Promise.all(encryptedLocations.map((location: BookLocationTable): Promise<LocationElementTable[]> =>
LocationRepo.fetchLocationElements(userId, location.loc_id, lang)))
]);
const encryptedChapterContents: BookChapterContentTable[] = nestedChapterContents.flat();
const encryptedChapterInfos: BookChapterInfosTable[] = nestedChapterInfos.flat();
const encryptedCharacterAttributes: BookCharactersAttributesTable[] = nestedCharacterAttributes.flat();
const encryptedWorldElements: BookWorldElementsTable[] = nestedWorldElements.flat();
const encryptedLocationElements: LocationElementTable[] = nestedLocationElements.flat();
const nestedLocationSubElements: LocationSubElementTable[][] = await Promise.all(
encryptedLocationElements.map((element: LocationElementTable): Promise<LocationSubElementTable[]> =>
LocationRepo.fetchLocationSubElements(userId, element.element_id, lang))
);
const encryptedLocationSubElements: LocationSubElementTable[] = nestedLocationSubElements.flat();
const eritBooks: EritBooksTable[] = encryptedBooks.map((book: EritBooksTable): EritBooksTable => ({
...book,
title: System.decryptDataWithUserKey(book.title, userEncryptionKey),
sub_title: book.sub_title ? System.decryptDataWithUserKey(book.sub_title, userEncryptionKey) : null,
summary: book.summary ? System.decryptDataWithUserKey(book.summary, userEncryptionKey) : null,
cover_image: book.cover_image ? System.decryptDataWithUserKey(book.cover_image, userEncryptionKey) : null
}));
const actSummaries: BookActSummariesTable[] = encryptedActSummaries.map((actSummary: BookActSummariesTable): BookActSummariesTable => ({
...actSummary,
summary: actSummary.summary ? System.decryptDataWithUserKey(actSummary.summary, userEncryptionKey) : null
}));
const aiGuideLine: BookAIGuideLineTable[] = encryptedAIGuidelines.map((guideLine: BookAIGuideLineTable): BookAIGuideLineTable => ({
...guideLine,
global_resume: guideLine.global_resume ? System.decryptDataWithUserKey(guideLine.global_resume, userEncryptionKey) : null,
themes: guideLine.themes ? System.decryptDataWithUserKey(guideLine.themes, userEncryptionKey) : null,
tone: guideLine.tone ? System.decryptDataWithUserKey(guideLine.tone, userEncryptionKey) : null,
atmosphere: guideLine.atmosphere ? System.decryptDataWithUserKey(guideLine.atmosphere, userEncryptionKey) : null,
current_resume: guideLine.current_resume ? System.decryptDataWithUserKey(guideLine.current_resume, userEncryptionKey) : null
}));
const chapters: BookChaptersTable[] = encryptedChapters.map((chapter: BookChaptersTable): BookChaptersTable => ({
...chapter,
title: System.decryptDataWithUserKey(chapter.title, userEncryptionKey)
}));
const chapterContents: BookChapterContentTable[] = encryptedChapterContents.map((chapterContent: BookChapterContentTable): BookChapterContentTable => ({
...chapterContent,
content: chapterContent.content ? JSON.parse(System.decryptDataWithUserKey(chapterContent.content, userEncryptionKey)) : null
}));
const chapterInfos: BookChapterInfosTable[] = encryptedChapterInfos.map((chapterInfo: BookChapterInfosTable): BookChapterInfosTable => ({
...chapterInfo,
summary: chapterInfo.summary ? System.decryptDataWithUserKey(chapterInfo.summary, userEncryptionKey) : null,
goal: chapterInfo.goal ? System.decryptDataWithUserKey(chapterInfo.goal, userEncryptionKey) : null
}));
const characters: BookCharactersTable[] = encryptedCharacters.map((character: BookCharactersTable): BookCharactersTable => ({
...character,
first_name: System.decryptDataWithUserKey(character.first_name, userEncryptionKey),
last_name: character.last_name ? System.decryptDataWithUserKey(character.last_name, userEncryptionKey) : null,
category: System.decryptDataWithUserKey(character.category, userEncryptionKey),
title: character.title ? System.decryptDataWithUserKey(character.title, userEncryptionKey) : null,
role: character.role ? System.decryptDataWithUserKey(character.role, userEncryptionKey) : null,
biography: character.biography ? System.decryptDataWithUserKey(character.biography, userEncryptionKey) : null,
history: character.history ? System.decryptDataWithUserKey(character.history, userEncryptionKey) : null
}));
const characterAttributes: BookCharactersAttributesTable[] = encryptedCharacterAttributes.map((attribute: BookCharactersAttributesTable): BookCharactersAttributesTable => ({
...attribute,
attribute_name: System.decryptDataWithUserKey(attribute.attribute_name, userEncryptionKey),
attribute_value: System.decryptDataWithUserKey(attribute.attribute_value, userEncryptionKey)
}));
const guideLine: BookGuideLineTable[] = encryptedGuidelines.map((guide: BookGuideLineTable): BookGuideLineTable => ({
...guide,
tone: guide.tone ? System.decryptDataWithUserKey(guide.tone, userEncryptionKey) : null,
atmosphere: guide.atmosphere ? System.decryptDataWithUserKey(guide.atmosphere, userEncryptionKey) : null,
writing_style: guide.writing_style ? System.decryptDataWithUserKey(guide.writing_style, userEncryptionKey) : null,
themes: guide.themes ? System.decryptDataWithUserKey(guide.themes, userEncryptionKey) : null,
symbolism: guide.symbolism ? System.decryptDataWithUserKey(guide.symbolism, userEncryptionKey) : null,
motifs: guide.motifs ? System.decryptDataWithUserKey(guide.motifs, userEncryptionKey) : null,
narrative_voice: guide.narrative_voice ? System.decryptDataWithUserKey(guide.narrative_voice, userEncryptionKey) : null,
pacing: guide.pacing ? System.decryptDataWithUserKey(guide.pacing, userEncryptionKey) : null,
intended_audience: guide.intended_audience ? System.decryptDataWithUserKey(guide.intended_audience, userEncryptionKey) : null,
key_messages: guide.key_messages ? System.decryptDataWithUserKey(guide.key_messages, userEncryptionKey) : null
}));
const incidents: BookIncidentsTable[] = encryptedIncidents.map((incident: BookIncidentsTable): BookIncidentsTable => ({
...incident,
title: System.decryptDataWithUserKey(incident.title, userEncryptionKey),
summary: incident.summary ? System.decryptDataWithUserKey(incident.summary, userEncryptionKey) : null
}));
const issues: BookIssuesTable[] = encryptedIssues.map((issue: BookIssuesTable): BookIssuesTable => ({
...issue,
name: System.decryptDataWithUserKey(issue.name, userEncryptionKey)
}));
const locations: BookLocationTable[] = encryptedLocations.map((location: BookLocationTable): BookLocationTable => ({
...location,
loc_name: System.decryptDataWithUserKey(location.loc_name, userEncryptionKey)
}));
const plotPoints: BookPlotPointsTable[] = encryptedPlotPoints.map((plotPoint: BookPlotPointsTable): BookPlotPointsTable => ({
...plotPoint,
title: System.decryptDataWithUserKey(plotPoint.title, userEncryptionKey),
summary: plotPoint.summary ? System.decryptDataWithUserKey(plotPoint.summary, userEncryptionKey) : null
}));
const worlds: BookWorldTable[] = encryptedWorlds.map((world: BookWorldTable): BookWorldTable => ({
...world,
name: System.decryptDataWithUserKey(world.name, userEncryptionKey),
history: world.history ? System.decryptDataWithUserKey(world.history, userEncryptionKey) : null,
politics: world.politics ? System.decryptDataWithUserKey(world.politics, userEncryptionKey) : null,
economy: world.economy ? System.decryptDataWithUserKey(world.economy, userEncryptionKey) : null,
religion: world.religion ? System.decryptDataWithUserKey(world.religion, userEncryptionKey) : null,
languages: world.languages ? System.decryptDataWithUserKey(world.languages, userEncryptionKey) : null
}));
const worldElements: BookWorldElementsTable[] = encryptedWorldElements.map((worldElement: BookWorldElementsTable): BookWorldElementsTable => ({
...worldElement,
name: System.decryptDataWithUserKey(worldElement.name, userEncryptionKey),
description: worldElement.description ? System.decryptDataWithUserKey(worldElement.description, userEncryptionKey) : null
}));
const locationElements: LocationElementTable[] = encryptedLocationElements.map((locationElement: LocationElementTable): LocationElementTable => ({
...locationElement,
element_name: System.decryptDataWithUserKey(locationElement.element_name, userEncryptionKey),
element_description: locationElement.element_description ? System.decryptDataWithUserKey(locationElement.element_description, userEncryptionKey) : null
}));
const locationSubElements: LocationSubElementTable[] = encryptedLocationSubElements.map((locationSubElement: LocationSubElementTable): LocationSubElementTable => ({
...locationSubElement,
sub_elem_name: System.decryptDataWithUserKey(locationSubElement.sub_elem_name, userEncryptionKey),
sub_elem_description: locationSubElement.sub_elem_description ? System.decryptDataWithUserKey(locationSubElement.sub_elem_description, userEncryptionKey) : null
}));
const bookTools: BookToolsTable[] = bookToolsData ? [bookToolsData] : [];
return {
eritBooks,
actSummaries,
aiGuideLine,
chapters,
chapterContents,
chapterInfos,
characters,
characterAttributes,
guideLine,
incidents,
issues,
locations,
plotPoints,
worlds,
worldElements,
locationElements,
locationSubElements,
bookTools
};
}
}