jest issue. about to try to fix https://github.com/nanostores/nanostores/issues/210

src/lib/prompt.tsx

@@ -0,0 +1,138 @@
+import { randomUUID } from "crypto";
+import { Op } from "./operator";
+import { atom, computed } from "nanostores";
+
+type Id = string;
+
+type IdAble = {
+    id: Id,
+}
+
+export enum Category {
+    subject = "subject",
+    style = "style",
+    vibes = "vibes",
+    medium = "medium",
+}
+
+const N_CATEGORIES = Object.keys(Category).length;
+
+export function catI(c: Category | string): number {
+    return Object.keys(Category).indexOf(c);
+}
+
+export type LibraryItem = {
+    id: Id,
+    name?: string,
+    prompt: string,
+    category: Category,
+}
+
+export type Library = Array<LibraryItem>;
+
+export type Nugget = IdAble & {
+    item: LibraryItem,
+    score: number,
+}
+
+export type Operation = IdAble & {
+    op: Op,
+    items: Array<Nugget>
+}
+
+export type PromptItem = Operation | Nugget
+
+export type Composition = Array<PromptItem>;
+
+export const $library = atom<Library>([])
+
+export function addItemToLibrary(item: LibraryItem) {
+    $library.set([
+        ...$library.get(), item,
+    ]);
+}
+
+export function removeItemFromLibrary(item: LibraryItem) {
+    $library.set($library.get().filter(i => i.id != item.id));
+}
+
+export const $composition = atom<Composition>([])
+
+export function insertIntoComposition(item: LibraryItem) {
+    $composition.set([
+        ...$composition.get(),
+        { id: randomUUID(), item, score: 0 } as Nugget,
+    ]);
+}
+
+export function removeFromComposition(item: PromptItem) {
+    $composition.set([
+        ...$composition.get().filter(i => i.id === item.id)
+    ]);
+}
+
+export function increaseNuggetScore(nuggetId: Id, amount: number = 1) {
+    $composition.set([
+        ...$composition.get().map(item => {
+            return (item.id == nuggetId && "score" in item) ? { ...item, score: item.score + amount } : item;
+        }
+        ),
+    ]);
+}
+export function decreaseNuggetScore(nuggetId: Id, amount: number = 1) {
+    $composition.set([
+        ...$composition.get().map(item => {
+            return (item.id == nuggetId && "score" in item) ? { ...item, score: item.score - amount } : item;
+        }
+        ),
+    ]);
+}
+
+export function changeOperationOp(operationId: Id, op: Op) {
+    $composition.set([
+        ...$composition.get().map(item => {
+            return (item.id == operationId) ? { ...item, op: op } : item;
+        }
+        ),
+    ]);
+}
+
+export function nuggetToText(nugget: Nugget) {
+    const sign = (nugget.score > 0 ? '+' : (nugget.score < 0 ? '-' : ''))
+    return "(" + nugget.item.prompt + ")" + (new Array(nugget.score)).map(i => sign).join("");
+}
+
+export function operationToText(operation: Operation): string {
+    return "(" + operation.items.map(nuggetToText).join(", ") + ")." + operation.op + "()";
+}
+
+export const textComposition = computed($composition, (composition) => {
+    const JOINER = ", ";
+    composition.map(item => {
+        return "op" in item ? operationToText(item) : nuggetToText(item);
+    }).join(JOINER);
+});
+
+export type SlottedComposition = PromptItem[][];
+
+/**
+ * This is necessary since a "composition" needs to have prompt items in
+ * different columns.
+ * 
+ * There are n columns (or slots) where n is the number of categories.
+ */
+export const $slottedComposition = computed($composition, (composition) => {
+    const slotted = new Array(N_CATEGORIES) as SlottedComposition;
+    composition.forEach(nugget => {
+        if ("op" in nugget) {
+            if (!nugget.items.length)
+                return null;
+            const cat = nugget.items[0].item.category;
+            slotted[catI(cat)].push(nugget);
+        } else {
+            const cat = nugget.item.category;
+            slotted[catI(cat)].push(nugget);
+        }
+    })
+    return slotted;
+});