@mzedstudio/uploadthingtrack 0.1.0

package/src/component/schema.ts

@@ -0,0 +1,44 @@
+import { defineSchema, defineTable } from "convex/server";
+import { v } from "convex/values";
+import { accessRuleValidator } from "./types";
+
+export default defineSchema({
+  files: defineTable({
+    key: v.string(),
+    url: v.string(),
+    name: v.string(),
+    size: v.number(),
+    mimeType: v.string(),
+    uploadedAt: v.number(),
+    userId: v.string(),
+    tags: v.optional(v.array(v.string())),
+    folder: v.optional(v.string()),
+    access: v.optional(accessRuleValidator),
+    metadata: v.optional(v.any()),
+    expiresAt: v.optional(v.number()),
+    replacedAt: v.optional(v.number()),
+    customId: v.optional(v.string()),
+    fileType: v.optional(v.string()),
+  })
+    .index("by_key", ["key"])
+    .index("by_user", ["userId"])
+    .index("by_user_uploadedAt", ["userId", "uploadedAt"])
+    .index("by_folder", ["folder"])
+    .index("by_expiresAt", ["expiresAt"]),
+
+  folderRules: defineTable({
+    folder: v.string(),
+    access: accessRuleValidator,
+    updatedAt: v.number(),
+  }).index("by_folder", ["folder"]),
+
+  globals: defineTable({
+    singleton: v.literal("globals"),
+    uploadthingApiKey: v.optional(v.string()),
+    defaultTtlMs: v.optional(v.number()),
+    ttlByMimeType: v.optional(v.record(v.string(), v.number())),
+    ttlByFileType: v.optional(v.record(v.string(), v.number())),
+    deleteRemoteOnExpire: v.optional(v.boolean()),
+    deleteBatchSize: v.optional(v.number()),
+  }).index("by_singleton", ["singleton"]),
+});

package/src/component/stats.ts

@@ -0,0 +1,23 @@
+import { query } from "./_generated/server";
+import { v } from "convex/values";
+
+export const getUsageStats = query({
+  args: {
+    userId: v.string(),
+  },
+  handler: async (ctx, args) => {
+    let totalFiles = 0;
+    let totalBytes = 0;
+
+    const query = ctx.db
+      .query("files")
+      .withIndex("by_user", (q: any) => q.eq("userId", args.userId));
+
+    for await (const file of query) {
+      totalFiles += 1;
+      totalBytes += file.size;
+    }
+
+    return { totalFiles, totalBytes };
+  },
+});

package/src/component/types.ts

@@ -0,0 +1,49 @@
+import { v, type Infer } from "convex/values";
+
+export const accessRuleValidator = v.object({
+  visibility: v.union(
+    v.literal("public"),
+    v.literal("private"),
+    v.literal("restricted"),
+  ),
+  allowUserIds: v.optional(v.array(v.string())),
+  denyUserIds: v.optional(v.array(v.string())),
+});
+
+export type AccessRule = Infer<typeof accessRuleValidator>;
+
+export const fileInfoValidator = v.object({
+  key: v.string(),
+  url: v.string(),
+  name: v.string(),
+  size: v.number(),
+  mimeType: v.string(),
+  uploadedAt: v.optional(v.number()),
+  fileType: v.optional(v.string()),
+  customId: v.optional(v.string()),
+});
+
+export type FileInfo = Infer<typeof fileInfoValidator>;
+
+export const fileUpsertOptionsValidator = v.object({
+  tags: v.optional(v.array(v.string())),
+  folder: v.optional(v.string()),
+  access: v.optional(accessRuleValidator),
+  metadata: v.optional(v.any()),
+  expiresAt: v.optional(v.number()),
+  ttlMs: v.optional(v.number()),
+  fileType: v.optional(v.string()),
+});
+
+export type FileUpsertOptions = Infer<typeof fileUpsertOptionsValidator>;
+
+export const configUpdateValidator = v.object({
+  uploadthingApiKey: v.optional(v.string()),
+  defaultTtlMs: v.optional(v.number()),
+  ttlByMimeType: v.optional(v.record(v.string(), v.number())),
+  ttlByFileType: v.optional(v.record(v.string(), v.number())),
+  deleteRemoteOnExpire: v.optional(v.boolean()),
+  deleteBatchSize: v.optional(v.number()),
+});
+
+export type ConfigUpdate = Infer<typeof configUpdateValidator>;

package/src/test.ts

@@ -0,0 +1,21 @@
+/// <reference types="vite/client" />
+import type { TestConvex } from "convex-test";
+import type { GenericSchema, SchemaDefinition } from "convex/server";
+import schema from "./component/schema.js";
+const modules = import.meta.glob("./component/**/*.ts");
+
+/**
+ * Register the UploadThing File Tracker component with a test Convex instance.
+ *
+ * @param t - The test convex instance from `convexTest()`.
+ * @param name - The component name as registered in convex.config.ts.
+ *               Defaults to "uploadthingFileTracker".
+ */
+export function register(
+  t: TestConvex<SchemaDefinition<GenericSchema, boolean>>,
+  name: string = "uploadthingFileTracker",
+) {
+  t.registerComponent(name, schema, modules);
+}
+
+export default { register, schema, modules };