rivet-design 0.23.1 → 0.25.0

package/dist/storage/profiles/ProfileRecords.js

@@ -61,6 +61,16 @@ class ProfileRecords {
         this.db = db;
         this.plans = new ProfilePlans_1.ProfilePlans(db);
     }
+    static openReadOnly({ root, accountId, }) {
+        if (!accountId.trim())
+            throw new Error('A signed-in account is required for profiles');
+        const directory = ProfileRecords.directoryFor({ root, accountId });
+        const database = (0, database_1.openReadOnlyDatabase)({
+            databasePath: path_1.default.join(directory, 'profile.sqlite'),
+            schemaVersion: migrations_1.PROFILE_SCHEMA_VERSION,
+        });
+        return database ? new ProfileRecords(directory, database) : null;
+    }
     static async open({ root, accountId, }) {
         if (!accountId.trim())
             throw new Error('A signed-in account is required for profiles');
@@ -74,7 +84,9 @@ class ProfileRecords {
             .values({ singleton: 1, id: (0, crypto_1.randomUUID)(), isEnabled: false, revision: 1 })
             .onConflictDoNothing()
             .run();
-        return new ProfileRecords(directory, db);
+        const records = new ProfileRecords(directory, db);
+        records.plans.initializeOnboarding();
+        return records;
     }
     static directoryFor({ root, accountId, }) {
         if (!accountId.trim())
@@ -245,6 +257,34 @@ class ProfileRecords {
             this.bump();
         });
     }
+    removeReference(referenceId) {
+        (0, database_1.transaction)(this.db, () => {
+            const reference = this.libraryReferences([referenceId]).find((item) => item.id === referenceId);
+            if (!reference)
+                throw new Error('Unknown saved reference');
+            const activeImport = this.db
+                .select({ id: schema_1.importJobs.id })
+                .from(schema_1.importJobs)
+                .where((0, drizzle_orm_1.and)((0, drizzle_orm_1.inArray)(schema_1.importJobs.sourceId, reference.sourceIds), (0, drizzle_orm_1.inArray)(schema_1.importJobs.state, ['pending', 'paused', 'failed'])))
+                .get();
+            if (activeImport)
+                throw new Error('This reference belongs to an unfinished import. Finish the import before removing the reference.');
+            this.db
+                .delete(schema_1.memberships)
+                .where((0, drizzle_orm_1.eq)(schema_1.memberships.referenceId, referenceId))
+                .run();
+            this.queue(reference, 'delete');
+            this.bump();
+        });
+    }
+    saveLocalReferences(input) {
+        return (0, database_1.transaction)(this.db, () => {
+            const source = this.addSource(input.source);
+            const references = input.references.map((reference) => this.writeReference(source.id, reference));
+            this.bump();
+            return { source, references };
+        });
+    }
     saveReference({ sourceId, reference, }) {
         return (0, database_1.transaction)(this.db, () => this.writeReference(sourceId, reference));
     }
@@ -372,7 +412,22 @@ class ProfileRecords {
             .all();
     }
     coverage(sourceId) {
-        const items = this.references(sourceId);
+        return this.referenceCoverage(this.references(sourceId), sourceId !== undefined);
+    }
+    importCoverage(jobId) {
+        const ids = this.db
+            .select({ id: schema_1.memberships.referenceId })
+            .from(schema_1.memberships)
+            .where((0, drizzle_orm_1.eq)(schema_1.memberships.seenJobId, jobId));
+        const items = this.db
+            .select()
+            .from(schema_1.references)
+            .where((0, drizzle_orm_1.inArray)(schema_1.references.id, ids))
+            .all()
+            .map(({ id, revision, body }) => ({ ...body, id, revision }));
+        return this.referenceCoverage(items, true, jobId);
+    }
+    referenceCoverage(items, isScoped, jobId) {
         const receipts = new Map(this.db
             .select()
             .from(schema_1.indexWork)
@@ -383,7 +438,7 @@ class ProfileRecords {
             downloaded: items.filter((item) => item.media.length > 0 &&
                 item.media.length >= (item.remoteImages?.length ?? 0)).length,
             unsupported: items.filter((item) => item.unsupportedReason).length,
-            failed: new Set(this.failures()
+            failed: new Set(this.failures(jobId)
                 .filter((failure) => items.some((item) => item.id === failure.referenceId))
                 .map((failure) => failure.referenceId)).size,
             analyzed: items.filter((item) => this.observation(item.id)).length,
@@ -394,7 +449,7 @@ class ProfileRecords {
                     work.revision === item.revision &&
                     this.observation(item.id));
             }).length,
-            pendingIndex: sourceId
+            pendingIndex: isScoped
                 ? items.filter((item) => receipts.get(item.id)?.state === 'pending')
                     .length
                 : [...receipts.values()].filter((item) => item.state === 'pending')

package/dist/storage/profiles/planSchema.d.ts

@@ -26,9 +26,9 @@ export declare const referenceDiscoveries: import("drizzle-orm/sqlite-core").SQL
                 id: string;
                 total: number;
                 collections: {
+                    description: string | null;
                     url: string | null;
                     title: string;
-                    description: string | null;
                     id: string;
                     children: string[];
                     total: number | null;
@@ -95,9 +95,9 @@ export declare const referencePlans: import("drizzle-orm/sqlite-core").SQLiteTab
                 discoveryId: string;
                 sources: {
                     references: {
-                        provider: "arena" | "pinterest" | "x" | "cosmos";
                         title: string;
                         text: string;
+                        provider: "pinterest" | "arena" | "cosmos" | "x";
                         remoteId: string;
                         media: {
                             hash: string;
@@ -106,22 +106,22 @@ export declare const referencePlans: import("drizzle-orm/sqlite-core").SQLiteTab
                             representation: "variant" | "image" | "video-cover" | "page";
                             page?: number | undefined;
                         }[];
-                        url?: string | undefined;
                         variant?: {
                             projectId: string;
                             sessionId: string;
                             variantId: string;
                             treeSha?: string | undefined;
                         } | undefined;
+                        url?: string | undefined;
                         note?: string | undefined;
                         remoteImages?: string[] | undefined;
                         remoteImageRepresentations?: ("image" | "video-cover")[] | undefined;
                         unsupportedReason?: string | undefined;
                     }[];
                     collection: {
+                        description: string | null;
                         url: string | null;
                         title: string;
-                        description: string | null;
                         id: string;
                         children: string[];
                         total: number | null;
@@ -281,7 +281,7 @@ export declare const connectorOnboarding: import("drizzle-orm/sqlite-core").SQLi
             name: string;
             tableName: "connector_onboarding";
             dataType: "string enum";
-            data: "arena" | "pinterest" | "x" | "cosmos";
+            data: "pinterest" | "arena" | "cosmos" | "x";
             driverParam: string;
             notNull: true;
             hasDefault: false;

package/dist/storage/profiles/schema.d.ts

@@ -127,11 +127,11 @@ export declare const guidance: import("drizzle-orm/sqlite-core").SQLiteTableWith
             dataType: "object json";
             data: {
                 path: string;
+                source: string;
                 hash: string;
                 mimeType: "image/png" | "image/jpeg" | "image/webp" | "image/gif";
                 bytes: number;
                 representation: "variant" | "image" | "video-cover" | "page";
-                source: string;
                 alt: string;
                 title?: string | undefined;
                 page?: number | undefined;
@@ -159,14 +159,14 @@ export declare const guidance: import("drizzle-orm/sqlite-core").SQLiteTableWith
                 url: string;
                 kind: "url" | "repository";
                 intent: string;
+                digest?: string | undefined;
+                revision?: string | undefined;
                 files?: {
                     name: string;
                     content: string;
                     path?: string | undefined;
                     sha256?: string | undefined;
                 }[] | undefined;
-                digest?: string | undefined;
-                revision?: string | undefined;
             })[];
             driverParam: string;
             notNull: false;
@@ -288,7 +288,7 @@ export declare const providerCooldowns: import("drizzle-orm/sqlite-core").SQLite
             name: string;
             tableName: "provider_cooldowns";
             dataType: "string";
-            data: "variant" | "text" | "arena" | "pinterest" | "image" | "x" | "cosmos" | "pdf";
+            data: "variant" | "text" | "pinterest" | "arena" | "cosmos" | "x" | "image" | "pdf";
             driverParam: string;
             notNull: true;
             hasDefault: false;
@@ -359,7 +359,7 @@ export declare const sources: import("drizzle-orm/sqlite-core").SQLiteTableWithC
             name: string;
             tableName: "sources";
             dataType: "string";
-            data: "variant" | "text" | "arena" | "pinterest" | "image" | "x" | "cosmos" | "pdf";
+            data: "variant" | "text" | "pinterest" | "arena" | "cosmos" | "x" | "image" | "pdf";
             driverParam: string;
             notNull: true;
             hasDefault: false;
@@ -527,9 +527,9 @@ export declare const references: import("drizzle-orm/sqlite-core").SQLiteTableWi
             tableName: "profile_references";
             dataType: "object json";
             data: {
-                provider: "variant" | "text" | "arena" | "pinterest" | "image" | "x" | "cosmos" | "pdf";
                 title: string;
                 text: string;
+                provider: "variant" | "text" | "pinterest" | "arena" | "cosmos" | "x" | "image" | "pdf";
                 remoteId: string;
                 media: {
                     hash: string;
@@ -538,13 +538,13 @@ export declare const references: import("drizzle-orm/sqlite-core").SQLiteTableWi
                     representation: "variant" | "image" | "video-cover" | "page";
                     page?: number | undefined;
                 }[];
-                url?: string | undefined;
                 variant?: {
                     projectId: string;
                     sessionId: string;
                     variantId: string;
                     treeSha?: string | undefined;
                 } | undefined;
+                url?: string | undefined;
                 note?: string | undefined;
                 remoteImages?: string[] | undefined;
                 remoteImageRepresentations?: ("image" | "video-cover")[] | undefined;
@@ -916,7 +916,7 @@ export declare const importJobs: import("drizzle-orm/sqlite-core").SQLiteTableWi
             name: string;
             tableName: "import_jobs";
             dataType: "string enum";
-            data: "pending" | "failed" | "complete" | "superseded" | "paused";
+            data: "pending" | "failed" | "superseded" | "complete" | "paused";
             driverParam: string;
             notNull: true;
             hasDefault: false;

package/dist/types/cosmos-profile.d.ts

@@ -24,17 +24,17 @@ export declare const CosmosProfileCollectionSchema: z.ZodObject<{
     parentId: z.ZodNullable<z.ZodEffects<z.ZodString, string, string>>;
     children: z.ZodArray<z.ZodEffects<z.ZodString, string, string>, "many">;
 }, "strip", z.ZodTypeAny, {
+    description: string | null;
     url: string;
     title: string;
-    description: string | null;
     id: string;
     children: string[];
     total: number;
     parentId: string | null;
 }, {
+    description: string | null;
     url: string;
     title: string;
-    description: string | null;
     id: string;
     children: string[];
     total: number;
@@ -66,17 +66,17 @@ export declare const CosmosProfilePageSchema: z.ZodObject<{
         parentId: z.ZodNullable<z.ZodEffects<z.ZodString, string, string>>;
         children: z.ZodArray<z.ZodEffects<z.ZodString, string, string>, "many">;
     }, "strip", z.ZodTypeAny, {
+        description: string | null;
         url: string;
         title: string;
-        description: string | null;
         id: string;
         children: string[];
         total: number;
         parentId: string | null;
     }, {
+        description: string | null;
         url: string;
         title: string;
-        description: string | null;
         id: string;
         children: string[];
         total: number;
@@ -88,9 +88,9 @@ export declare const CosmosProfilePageSchema: z.ZodObject<{
     total: number;
     nextCursor: string | null;
     collections: {
+        description: string | null;
         url: string;
         title: string;
-        description: string | null;
         id: string;
         children: string[];
         total: number;
@@ -106,9 +106,9 @@ export declare const CosmosProfilePageSchema: z.ZodObject<{
     total: number;
     nextCursor: string | null;
     collections: {
+        description: string | null;
         url: string;
         title: string;
-        description: string | null;
         id: string;
         children: string[];
         total: number;

package/dist/types/cosmos.d.ts

@@ -49,7 +49,6 @@ export declare const CosmosElementSchema: z.ZodObject<{
     }>, "many">;
     unsupportedReason: z.ZodOptional<z.ZodString>;
 }, "strip", z.ZodTypeAny, {
-    owner: string | null;
     url: string;
     title: string;
     text: string;
@@ -59,10 +58,10 @@ export declare const CosmosElementSchema: z.ZodObject<{
         representation: "image" | "video-cover";
     }[];
     author: string | null;
+    owner: string | null;
     sourceUrl: string | null;
     unsupportedReason?: string | undefined;
 }, {
-    owner: string | null;
     url: string;
     title: string;
     text: string;
@@ -72,6 +71,7 @@ export declare const CosmosElementSchema: z.ZodObject<{
         representation: "image" | "video-cover";
     }[];
     author: string | null;
+    owner: string | null;
     sourceUrl: string | null;
     unsupportedReason?: string | undefined;
 }>;
@@ -112,7 +112,6 @@ export declare const CosmosPageSchema: z.ZodObject<{
         }>, "many">;
         unsupportedReason: z.ZodOptional<z.ZodString>;
     }, "strip", z.ZodTypeAny, {
-        owner: string | null;
         url: string;
         title: string;
         text: string;
@@ -122,10 +121,10 @@ export declare const CosmosPageSchema: z.ZodObject<{
             representation: "image" | "video-cover";
         }[];
         author: string | null;
+        owner: string | null;
         sourceUrl: string | null;
         unsupportedReason?: string | undefined;
     }, {
-        owner: string | null;
         url: string;
         title: string;
         text: string;
@@ -135,6 +134,7 @@ export declare const CosmosPageSchema: z.ZodObject<{
             representation: "image" | "video-cover";
         }[];
         author: string | null;
+        owner: string | null;
         sourceUrl: string | null;
         unsupportedReason?: string | undefined;
     }>, "many">;
@@ -148,7 +148,6 @@ export declare const CosmosPageSchema: z.ZodObject<{
         children: string[];
     };
     elements: {
-        owner: string | null;
         url: string;
         title: string;
         text: string;
@@ -158,6 +157,7 @@ export declare const CosmosPageSchema: z.ZodObject<{
             representation: "image" | "video-cover";
         }[];
         author: string | null;
+        owner: string | null;
         sourceUrl: string | null;
         unsupportedReason?: string | undefined;
     }[];
@@ -171,7 +171,6 @@ export declare const CosmosPageSchema: z.ZodObject<{
         children: string[];
     };
     elements: {
-        owner: string | null;
         url: string;
         title: string;
         text: string;
@@ -181,6 +180,7 @@ export declare const CosmosPageSchema: z.ZodObject<{
             representation: "image" | "video-cover";
         }[];
         author: string | null;
+        owner: string | null;
         sourceUrl: string | null;
         unsupportedReason?: string | undefined;
     }[];
@@ -224,7 +224,6 @@ export declare const CosmosResultSchema: z.ZodDiscriminatedUnion<"status", [z.Zo
         }>, "many">;
         unsupportedReason: z.ZodOptional<z.ZodString>;
     }, "strip", z.ZodTypeAny, {
-        owner: string | null;
         url: string;
         title: string;
         text: string;
@@ -234,10 +233,10 @@ export declare const CosmosResultSchema: z.ZodDiscriminatedUnion<"status", [z.Zo
             representation: "image" | "video-cover";
         }[];
         author: string | null;
+        owner: string | null;
         sourceUrl: string | null;
         unsupportedReason?: string | undefined;
     }, {
-        owner: string | null;
         url: string;
         title: string;
         text: string;
@@ -247,6 +246,7 @@ export declare const CosmosResultSchema: z.ZodDiscriminatedUnion<"status", [z.Zo
             representation: "image" | "video-cover";
         }[];
         author: string | null;
+        owner: string | null;
         sourceUrl: string | null;
         unsupportedReason?: string | undefined;
     }>, "many">;
@@ -263,7 +263,6 @@ export declare const CosmosResultSchema: z.ZodDiscriminatedUnion<"status", [z.Zo
         children: string[];
     };
     elements: {
-        owner: string | null;
         url: string;
         title: string;
         text: string;
@@ -273,6 +272,7 @@ export declare const CosmosResultSchema: z.ZodDiscriminatedUnion<"status", [z.Zo
             representation: "image" | "video-cover";
         }[];
         author: string | null;
+        owner: string | null;
         sourceUrl: string | null;
         unsupportedReason?: string | undefined;
     }[];
@@ -287,7 +287,6 @@ export declare const CosmosResultSchema: z.ZodDiscriminatedUnion<"status", [z.Zo
         children: string[];
     };
     elements: {
-        owner: string | null;
         url: string;
         title: string;
         text: string;
@@ -297,6 +296,7 @@ export declare const CosmosResultSchema: z.ZodDiscriminatedUnion<"status", [z.Zo
             representation: "image" | "video-cover";
         }[];
         author: string | null;
+        owner: string | null;
         sourceUrl: string | null;
         unsupportedReason?: string | undefined;
     }[];