feat: initial impl of file proxy

This commit is contained in:
2025-10-21 23:45:04 +00:00
parent 6eded27121
commit 6234c5efd3
24 changed files with 420 additions and 125 deletions

View File

@@ -6,6 +6,7 @@ backend: convex
This project uses npm workspaces.
- `packages/convex` - convex functions and models
- `apps/drive-web` - frontend dashboard
- `apps/file-proxy` - proxies uploaded files via opaque share tokens
- `packages/path` - path utils
# General Guidelines

View File

@@ -2,3 +2,5 @@
VITE_CONVEX_URL=
# this is the convex url for invoking http actions
VITE_CONVEX_SITE_URL=
# this is the url to the file proxy
FILE_PROXY_URL=

View File

@@ -1,20 +1,20 @@
import type { Doc } from "@fileone/convex/dataModel"
import type { OpenedFile } from "@fileone/convex/filesystem"
import { ImagePreviewDialog } from "./image-preview-dialog"
export function FilePreviewDialog({
file,
openedFile,
onClose,
}: {
file: Doc<"files">
openedFile: OpenedFile
onClose: () => void
}) {
if (!file) return null
switch (file.mimeType) {
switch (openedFile.file.mimeType) {
case "image/jpeg":
case "image/png":
case "image/gif":
return <ImagePreviewDialog file={file} onClose={onClose} />
return (
<ImagePreviewDialog openedFile={openedFile} onClose={onClose} />
)
default:
return null
}

View File

@@ -0,0 +1,3 @@
export function fileShareUrl(shareToken: string) {
return `${import.meta.env.VITE_FILE_PROXY_URL}/files/${shareToken}`
}

View File

@@ -1,7 +1,5 @@
import { api } from "@fileone/convex/api"
import type { Doc } from "@fileone/convex/dataModel"
import type { OpenedFile } from "@fileone/convex/filesystem"
import { DialogTitle } from "@radix-ui/react-dialog"
import { useQuery as useConvexQuery } from "convex/react"
import { atom, useAtom, useAtomValue, useSetAtom } from "jotai"
import {
DownloadIcon,
@@ -18,9 +16,8 @@ import {
DialogClose,
DialogContent,
DialogHeader,
DialogOverlay,
} from "@/components/ui/dialog"
import { LoadingSpinner } from "@/components/ui/loading-spinner"
import { fileShareUrl } from "./file-share"
const zoomLevelAtom = atom(
1,
@@ -35,15 +32,12 @@ const zoomLevelAtom = atom(
)
export function ImagePreviewDialog({
file,
openedFile,
onClose,
}: {
file: Doc<"files">
openedFile: OpenedFile
onClose: () => void
}) {
const fileUrl = useConvexQuery(api.filesystem.fetchFileUrl, {
fileId: file._id,
})
const setZoomLevel = useSetAtom(zoomLevelAtom)
useEffect(
@@ -62,23 +56,12 @@ export function ImagePreviewDialog({
}
}}
>
<DialogOverlay className="flex items-center justify-center">
{!fileUrl ? (
<LoadingSpinner className="text-neutral-200 size-10" />
) : null}
</DialogOverlay>
{fileUrl ? <PreviewContent fileUrl={fileUrl} file={file} /> : null}
<PreviewContent openedFile={openedFile} />
</Dialog>
)
}
function PreviewContent({
fileUrl,
file,
}: {
fileUrl: string
file: Doc<"files">
}) {
function PreviewContent({ openedFile }: { openedFile: OpenedFile }) {
return (
<DialogContent
showCloseButton={false}
@@ -86,10 +69,10 @@ function PreviewContent({
>
<DialogHeader className="overflow-auto border-b border-b-border p-4 flex flex-row items-center justify-between">
<DialogTitle className="truncate flex-1">
{file.name}
{openedFile.file.name}
</DialogTitle>
<div className="flex flex-row items-center space-x-2">
<Toolbar fileUrl={fileUrl} file={file} />
<Toolbar openedFile={openedFile} />
<Button variant="ghost" size="icon" asChild>
<DialogClose>
<XIcon />
@@ -99,13 +82,13 @@ function PreviewContent({
</div>
</DialogHeader>
<div className="w-full h-full flex items-center justify-center max-h-[calc(100vh-10rem)] overflow-auto">
<ImagePreview fileUrl={fileUrl} file={file} />
<ImagePreview openedFile={openedFile} />
</div>
</DialogContent>
)
}
function Toolbar({ fileUrl, file }: { fileUrl: string; file: Doc<"files"> }) {
function Toolbar({ openedFile }: { openedFile: OpenedFile }) {
const setZoomLevel = useSetAtom(zoomLevelAtom)
const zoomInterval = useRef<ReturnType<typeof setInterval> | null>(null)
@@ -159,8 +142,8 @@ function Toolbar({ fileUrl, file }: { fileUrl: string; file: Doc<"files"> }) {
</Button>
<Button asChild>
<a
href={fileUrl}
download={file.name}
href={fileShareUrl(openedFile.shareToken)}
download={openedFile.file.name}
target="_blank"
className="flex flex-row items-center"
>
@@ -191,18 +174,12 @@ function ResetZoomButton() {
)
}
function ImagePreview({
fileUrl,
file,
}: {
fileUrl: string
file: Doc<"files">
}) {
function ImagePreview({ openedFile }: { openedFile: OpenedFile }) {
const zoomLevel = useAtomValue(zoomLevelAtom)
return (
<img
src={fileUrl}
alt={file.name}
src={fileShareUrl(openedFile.shareToken)}
alt={openedFile.file.name}
className="object-contain"
style={{ transform: `scale(${zoomLevel})` }}
/>

View File

@@ -3,12 +3,14 @@ import type { Doc, Id } from "@fileone/convex/dataModel"
import {
type FileSystemItem,
newFileSystemHandle,
type OpenedFile,
} from "@fileone/convex/filesystem"
import { useMutation } from "@tanstack/react-query"
import { createFileRoute } from "@tanstack/react-router"
import type { Row, Table } from "@tanstack/react-table"
import {
useMutation as useContextMutation,
useMutation as useConvexMutation,
useQuery as useConvexQuery,
} from "convex/react"
import { atom, useAtom, useAtomValue, useSetAtom, useStore } from "jotai"
@@ -77,22 +79,12 @@ const fileDragInfoAtom = atom<FileDragInfo | null>(null)
const optimisticDeletedItemsAtom = atom(
new Set<Id<"files"> | Id<"directories">>(),
)
const openedFileAtom = atom<Doc<"files"> | null>(null)
const openedFileAtom = atom<OpenedFile | null>(null)
const itemBeingRenamedAtom = atom<{
originalItem: FileSystemItem
name: string
} | null>(null)
const tableFilterAtom = atom((get) => {
const optimisticDeletedItems = get(optimisticDeletedItemsAtom)
console.log("optimisticDeletedItems", optimisticDeletedItems)
return (item: FileSystemItem) => {
const test = !optimisticDeletedItems.has(item.doc._id)
console.log("test", test)
return test
}
})
// MARK: page entry
function RouteComponent() {
const { directoryId } = Route.useParams()
@@ -100,7 +92,6 @@ function RouteComponent() {
const directory = useConvexQuery(api.files.fetchDirectory, {
directoryId,
})
const store = useStore()
const directoryContent = useConvexQuery(
api.filesystem.fetchDirectoryContent,
{
@@ -108,45 +99,12 @@ function RouteComponent() {
trashed: false,
},
)
const setOpenedFile = useSetAtom(openedFileAtom)
const setContextMenuTargetItems = useSetAtom(contextMenuTargetItemsAtom)
const tableFilter = useCallback(
(item: FileSystemItem) =>
store.get(optimisticDeletedItemsAtom).has(item.doc._id),
[store],
)
const openFile = useCallback(
(file: Doc<"files">) => {
setOpenedFile(file)
},
[setOpenedFile],
)
const directoryUrlFn = useCallback(
(directory: Doc<"directories">) => `/directories/${directory._id}`,
[],
)
const directoryUrlById = useCallback(
(directoryId: Id<"directories">) => `/directories/${directoryId}`,
[],
)
const handleContextMenuRequest = (
row: Row<FileSystemItem>,
table: Table<FileSystemItem>,
) => {
if (row.getIsSelected()) {
setContextMenuTargetItems(
table.getSelectedRowModel().rows.map((row) => row.original),
)
} else {
setContextMenuTargetItems([row.original])
}
}
if (!directory || !directoryContent || !rootDirectory) {
return <DirectoryPageSkeleton />
}
@@ -171,17 +129,7 @@ function RouteComponent() {
{/* DirectoryContentContextMenu must wrap div instead of DirectoryContentTable, otherwise radix will throw "event.preventDefault is not a function" error, idk why */}
<DirectoryContentContextMenu>
<div className="w-full">
<WithAtom atom={optimisticDeletedItemsAtom}>
{(optimisticDeletedItems) => (
<DirectoryContentTable
hiddenItems={optimisticDeletedItems}
directoryUrlFn={directoryUrlFn}
fileDragInfoAtom={fileDragInfoAtom}
onContextMenu={handleContextMenuRequest}
onOpenFile={openFile}
/>
)}
</WithAtom>
<_DirectoryContentTable />
</div>
</DirectoryContentContextMenu>
@@ -228,7 +176,7 @@ function RouteComponent() {
if (!openedFile) return null
return (
<FilePreviewDialog
file={openedFile}
openedFile={openedFile}
onClose={() => setOpenedFile(null)}
/>
)
@@ -238,6 +186,58 @@ function RouteComponent() {
)
}
// MARK: directory table
function _DirectoryContentTable() {
const optimisticDeletedItems = useAtomValue(optimisticDeletedItemsAtom)
const setOpenedFile = useSetAtom(openedFileAtom)
const setContextMenuTargetItems = useSetAtom(contextMenuTargetItemsAtom)
const { mutate: openFile } = useMutation({
mutationFn: useConvexMutation(api.filesystem.openFile),
onSuccess: (openedFile: OpenedFile) => {
console.log("openedFile", openedFile)
setOpenedFile(openedFile)
},
onError: (error) => {
console.error(error)
toast.error("Failed to open file")
},
})
const onTableOpenFile = (file: Doc<"files">) => {
openFile({ fileId: file._id })
}
const directoryUrlFn = useCallback(
(directory: Doc<"directories">) => `/directories/${directory._id}`,
[],
)
const handleContextMenuRequest = (
row: Row<FileSystemItem>,
table: Table<FileSystemItem>,
) => {
if (row.getIsSelected()) {
setContextMenuTargetItems(
table.getSelectedRowModel().rows.map((row) => row.original),
)
} else {
setContextMenuTargetItems([row.original])
}
}
return (
<DirectoryContentTable
hiddenItems={optimisticDeletedItems}
directoryUrlFn={directoryUrlFn}
fileDragInfoAtom={fileDragInfoAtom}
onContextMenu={handleContextMenuRequest}
onOpenFile={onTableOpenFile}
/>
)
}
// ==================================
// MARK: ctx menu

View File

@@ -0,0 +1,4 @@
CONVEX_URL=
# api key used to auth with the convex backend
# use the drexa cli to generate an api key, then add the api key to the api key table via the convex dashboard
API_KEY=

14
apps/file-proxy/auth.ts Normal file
View File

@@ -0,0 +1,14 @@
import { createMiddleware } from "hono/factory"
export type ApiKeyContextVariable = {
apiKey: string
}
const apiKeyMiddleware = createMiddleware<{ Variables: ApiKeyContextVariable }>(
async (c, next) => {
c.set("apiKey", process.env.API_KEY)
await next()
},
)
export { apiKeyMiddleware }

View File

@@ -0,0 +1,16 @@
import { ConvexHttpClient } from "convex/browser"
import { createMiddleware } from "hono/factory"
const _client = new ConvexHttpClient(process.env.CONVEX_URL)
export type ConvexContextVariables = {
convex: ConvexHttpClient
}
export const convexMiddleware = createMiddleware<{
Variables: ConvexContextVariables
}>(async (c, next) => {
c.var
c.set("convex", _client)
await next()
})

6
apps/file-proxy/env.d.ts vendored Normal file
View File

@@ -0,0 +1,6 @@
declare module "bun" {
interface Env {
CONVEX_URL: string
API_KEY: string
}
}

View File

@@ -1,12 +1,39 @@
import { Hono } from "hono"
import { api } from "@fileone/convex/api"
import { newRouter } from "./router"
const h = new Hono().basePath("/files")
const r = newRouter().basePath("/files")
h.get("/:fileId", async (c) => {
const fileId = c.req.param("fileId")
if (!fileId) {
return c.json({ error: "File ID is required" }, 400)
r.get(":shareToken", async (c) => {
const shareToken = c.req.param("shareToken")
if (!shareToken) {
return c.json({ error: "not found" }, 404)
}
const fileShare = await c.var.convex.query(api.fileshare.findFileShare, {
apiKey: c.var.apiKey,
shareToken,
})
if (!fileShare) {
return c.json({ error: "not found" }, 404)
}
const fileUrl = await c.var.convex.query(api.filesystem.getStorageUrl, {
apiKey: c.var.apiKey,
storageId: fileShare.storageId,
})
if (!fileUrl) {
return c.json({ error: "not found" }, 404)
}
const fileResponse = await fetch(fileUrl)
if (!fileResponse.ok) {
return c.json({ error: "not found" }, 404)
}
return new Response(fileResponse.body, {
status: fileResponse.status,
headers: fileResponse.headers,
})
})
export { h as files }
export { r as files }

View File

@@ -1,10 +1,16 @@
import { Hono } from "hono"
import { handleFileRequest } from "./files"
import { apiKeyMiddleware } from "./auth"
import { convexMiddleware } from "./convex"
import { files } from "./files"
Bun.serve({
routes: {
"/files/:fileId": {
GET: handleFileRequest,
},
},
})
const app = new Hono()
app.use(convexMiddleware)
app.use(apiKeyMiddleware)
app.route("/", files)
export default {
port: 8081,
fetch: app.fetch,
}

View File

@@ -3,6 +3,9 @@
"module": "index.ts",
"type": "module",
"private": true,
"scripts": {
"dev": "bun --hot run index.ts"
},
"devDependencies": {
"@types/bun": "latest"
},
@@ -11,6 +14,7 @@
},
"dependencies": {
"@fileone/convex": "workspace:*",
"arktype": "^2.1.23",
"convex": "^1.28.0",
"hono": "^4.10.1"
}

View File

@@ -1,7 +1,11 @@
import type { RouterTypes } from "bun"
import { Hono } from "hono"
import type { ApiKeyContextVariable } from "./auth"
import type { ConvexContextVariables } from "./convex"
function router<
R extends { [K in keyof R]: RouterTypes.RouteValue<Extract<K, string>> },
>(routes: R): R {
return routes
type ContextVariables = ConvexContextVariables & ApiKeyContextVariable
export function newRouter() {
return new Hono<{
Variables: ContextVariables
}>()
}

View File

@@ -83,6 +83,7 @@
"name": "@drexa/file-proxy",
"dependencies": {
"@fileone/convex": "workspace:*",
"arktype": "^2.1.23",
"convex": "^1.28.0",
"hono": "^4.10.1",
},
@@ -128,6 +129,12 @@
"convex": "1.28.0",
},
"packages": {
"@ark/regex": ["@ark/regex@0.0.0", "", { "dependencies": { "@ark/util": "0.50.0" } }, "sha512-p4vsWnd/LRGOdGQglbwOguIVhPmCAf5UzquvnDoxqhhPWTP84wWgi1INea8MgJ4SnI2gp37f13oA4Waz9vwNYg=="],
"@ark/schema": ["@ark/schema@0.50.0", "", { "dependencies": { "@ark/util": "0.50.0" } }, "sha512-hfmP82GltBZDadIOeR3argKNlYYyB2wyzHp0eeAqAOFBQguglMV/S7Ip2q007bRtKxIMLDqFY6tfPie1dtssaQ=="],
"@ark/util": ["@ark/util@0.50.0", "", {}, "sha512-tIkgIMVRpkfXRQIEf0G2CJryZVtHVrqcWHMDa5QKo0OEEBu0tHkRSIMm4Ln8cd8Bn9TPZtvc/kE2Gma8RESPSg=="],
"@babel/code-frame": ["@babel/code-frame@7.27.1", "", { "dependencies": { "@babel/helper-validator-identifier": "^7.27.1", "js-tokens": "^4.0.0", "picocolors": "^1.1.1" } }, "sha512-cjQ7ZlQ0Mv3b47hABuTevyTuYN4i+loJKGeV9flcCgIK37cCXRh+L1bd3iBHlynerhQ7BhCkn2BPbQUL+rGqFg=="],
"@babel/compat-data": ["@babel/compat-data@7.28.4", "", {}, "sha512-YsmSKC29MJwf0gF8Rjjrg5LQCmyh+j/nD8/eP7f+BeoQTKYqs9RoWbjGOdy0+1Ekr68RJZMUOPVQaQisnIo4Rw=="],
@@ -542,6 +549,8 @@
"aria-hidden": ["aria-hidden@1.2.6", "", { "dependencies": { "tslib": "^2.0.0" } }, "sha512-ik3ZgC9dY/lYVVM++OISsaYDeg1tb0VtP5uL3ouh1koGOaUMDPpbFIei4JkFimWUFPn90sbMNMXQAIVOlnYKJA=="],
"arktype": ["arktype@2.1.23", "", { "dependencies": { "@ark/regex": "0.0.0", "@ark/schema": "0.50.0", "@ark/util": "0.50.0" } }, "sha512-tyxNWX6xJVMb2EPJJ3OjgQS1G/vIeQRrZuY4DeBNQmh8n7geS+czgbauQWB6Pr+RXiOO8ChEey44XdmxsqGmfQ=="],
"asn1js": ["asn1js@3.0.6", "", { "dependencies": { "pvtsutils": "^1.3.6", "pvutils": "^1.1.3", "tslib": "^2.8.1" } }, "sha512-UOCGPYbl0tv8+006qks/dTgV9ajs97X2p0FAbyS2iyCRrmLSRolDaHdp+v/CLgnzHc3fVB+CwYiUmei7ndFcgA=="],
"ast-types": ["ast-types@0.16.1", "", { "dependencies": { "tslib": "^2.0.1" } }, "sha512-6t10qk83GOG8p0vKmaCr8eiilZwO171AvbROMtvvNiwrTly62t+7XkA8RdIIVbpMhCASAsxgAzdRSwh6nw/5Dg=="],

View File

@@ -17,12 +17,15 @@ import type * as betterauth_auth from "../betterauth/auth.js";
import type * as convex__generated_api from "../convex/_generated/api.js";
import type * as convex__generated_server from "../convex/_generated/server.js";
import type * as files from "../files.js";
import type * as fileshare from "../fileshare.js";
import type * as filesystem from "../filesystem.js";
import type * as functions from "../functions.js";
import type * as http from "../http.js";
import type * as model_apikey from "../model/apikey.js";
import type * as model_directories from "../model/directories.js";
import type * as model_filepreview from "../model/filepreview.js";
import type * as model_files from "../model/files.js";
import type * as model_fileshare from "../model/fileshare.js";
import type * as model_filesystem from "../model/filesystem.js";
import type * as model_user from "../model/user.js";
import type * as shared_error from "../shared/error.js";
@@ -54,12 +57,15 @@ declare const fullApi: ApiFromModules<{
"convex/_generated/api": typeof convex__generated_api;
"convex/_generated/server": typeof convex__generated_server;
files: typeof files;
fileshare: typeof fileshare;
filesystem: typeof filesystem;
functions: typeof functions;
http: typeof http;
"model/apikey": typeof model_apikey;
"model/directories": typeof model_directories;
"model/filepreview": typeof model_filepreview;
"model/files": typeof model_files;
"model/fileshare": typeof model_fileshare;
"model/filesystem": typeof model_filesystem;
"model/user": typeof model_user;
"shared/error": typeof shared_error;

View File

@@ -0,0 +1,12 @@
import { v } from "convex/values"
import { apiKeyAuthenticatedQuery } from "./functions"
import * as FileShare from "./model/fileshare"
export const findFileShare = apiKeyAuthenticatedQuery({
args: {
shareToken: v.string(),
},
handler: async (ctx, { shareToken }) => {
return await FileShare.find(ctx, { shareToken })
},
})

View File

@@ -1,11 +1,13 @@
import { v } from "convex/values"
import {
apiKeyAuthenticatedQuery,
authenticatedMutation,
authenticatedQuery,
authorizedGet,
} from "./functions"
import * as Directories from "./model/directories"
import * as Files from "./model/files"
import * as FileSystem from "./model/filesystem"
import {
deleteItemsPermanently,
emptyTrash as emptyTrashImpl,
@@ -160,6 +162,15 @@ export const restoreItems = authenticatedMutation({
},
})
export const getStorageUrl = apiKeyAuthenticatedQuery({
args: {
storageId: v.id("_storage"),
},
handler: async (ctx, { storageId }) => {
return await ctx.storage.getUrl(storageId)
},
})
export const fetchFileUrl = authenticatedQuery({
args: {
fileId: v.id("files"),
@@ -168,3 +179,12 @@ export const fetchFileUrl = authenticatedQuery({
return await fetchFileUrlImpl(ctx, { fileId })
},
})
export const openFile = authenticatedMutation({
args: {
fileId: v.id("files"),
},
handler: async (ctx, { fileId }) => {
return await FileSystem.openFile(ctx, { fileId })
},
})

View File

@@ -26,6 +26,10 @@ export type AuthenticatedMutationCtx = MutationCtx & {
identity: UserIdentity
}
export type ApiKeyAuthenticatedQueryCtx = QueryCtx & {
__branded: "ApiKeyAuthenticatedQueryCtx"
}
/**
* Custom query that automatically provides authenticated user context
* Throws an error if the user is not authenticated
@@ -52,6 +56,21 @@ export const authenticatedMutation = customMutation(
}),
)
/**
* Custom query that requires api key authentication for a query.
*/
export const apiKeyAuthenticatedQuery = customQuery(query, {
args: {
apiKey: v.string(),
},
input: async (ctx, args) => {
if (!(await ApiKey.verifyApiKey(ctx, args.apiKey))) {
throw Err.create(Err.Code.Unauthenticated, "Invalid API key")
}
return { ctx: ctx as ApiKeyAuthenticatedQueryCtx, args }
},
})
/**
* Custom mutation that requires api key authentication for a mutation.
*/

View File

@@ -0,0 +1,36 @@
import type { Doc, Id } from "../_generated/dataModel"
import type {
AuthenticatedMutationCtx,
AuthenticatedQueryCtx,
} from "../functions"
import * as FileShare from "./fileshare"
const PREVIEW_FILE_SHARE_VALID_FOR_MS = 1000 * 60 * 10 // 10 minutes
export async function find(
ctx: AuthenticatedMutationCtx | AuthenticatedQueryCtx,
{ storageId }: { storageId: Id<"_storage"> },
) {
return await FileShare.findByStorageId(ctx, { storageId })
}
export async function create(
ctx: AuthenticatedMutationCtx,
{ storageId }: { storageId: Id<"_storage"> },
) {
return await FileShare.create(ctx, {
shareToken: crypto.randomUUID(),
storageId,
expiresAt: new Date(Date.now() + PREVIEW_FILE_SHARE_VALID_FOR_MS),
})
}
export async function extend(
ctx: AuthenticatedMutationCtx,
{ doc }: { doc: Doc<"fileShares"> },
) {
return await FileShare.updateExpiry(ctx, {
doc,
expiresAt: new Date(Date.now() + PREVIEW_FILE_SHARE_VALID_FOR_MS),
})
}

View File

@@ -0,0 +1,83 @@
import type { Doc, Id } from "../_generated/dataModel"
import type { MutationCtx } from "../_generated/server"
import type {
ApiKeyAuthenticatedQueryCtx,
AuthenticatedMutationCtx,
AuthenticatedQueryCtx,
} from "../functions"
import * as Err from "../shared/error"
export async function create(
ctx: MutationCtx,
{
shareToken,
storageId,
expiresAt,
}: { shareToken: string; storageId: Id<"_storage">; expiresAt?: Date },
) {
const id = await ctx.db.insert("fileShares", {
shareToken,
storageId,
expiresAt: expiresAt?.getTime(),
})
const doc = await ctx.db.get(id)
if (!doc) {
throw Err.create(Err.Code.Internal, "Failed to create file share")
}
return doc
}
export async function remove(
ctx: AuthenticatedMutationCtx,
{ doc }: { doc: Doc<"fileShares"> },
) {
return await ctx.db.delete(doc._id)
}
export async function find(
ctx:
| AuthenticatedMutationCtx
| AuthenticatedQueryCtx
| ApiKeyAuthenticatedQueryCtx,
{ shareToken }: { shareToken: string },
) {
const doc = await ctx.db
.query("fileShares")
.withIndex("byShareToken", (q) => q.eq("shareToken", shareToken))
.first()
if (!doc) {
throw Err.create(Err.Code.NotFound, "File share not found")
}
if (hasExpired(doc)) {
throw Err.create(Err.Code.NotFound, "File share not found")
}
return doc
}
export async function findByStorageId(
ctx: AuthenticatedMutationCtx | AuthenticatedQueryCtx,
{ storageId }: { storageId: Id<"_storage"> },
) {
return await ctx.db
.query("fileShares")
.withIndex("byStorageId", (q) => q.eq("storageId", storageId))
.first()
}
export function hasExpired(doc: Doc<"fileShares">) {
if (!doc.expiresAt) {
return false
}
return doc.expiresAt < Date.now()
}
export async function updateExpiry(
ctx: AuthenticatedMutationCtx,
{ doc, expiresAt }: { doc: Doc<"fileShares">; expiresAt: Date },
) {
return await ctx.db.patch(doc._id, {
expiresAt: expiresAt.getTime(),
})
}

View File

@@ -17,7 +17,9 @@ import {
newFileHandle,
} from "../shared/filesystem"
import * as Directories from "./directories"
import * as FilePreview from "./filepreview"
import * as Files from "./files"
import * as FileShare from "./fileshare"
export const VDirectoryHandle = v.object({
kind: v.literal(FileType.Directory),
@@ -228,3 +230,33 @@ export async function fetchFileUrl(
return url
}
export async function openFile(
ctx: AuthenticatedMutationCtx,
{ fileId }: { fileId: Id<"files"> },
) {
const file = await authorizedGet(ctx, fileId)
if (!file) {
throw Err.create(Err.Code.NotFound, "file not found")
}
const fileShare = await FilePreview.find(ctx, {
storageId: file.storageId,
})
if (fileShare && !FileShare.hasExpired(fileShare)) {
await FilePreview.extend(ctx, { doc: fileShare })
return {
file,
shareToken: fileShare.shareToken,
}
}
const newFileShare = await FilePreview.create(ctx, {
storageId: file.storageId,
})
return {
file,
shareToken: newFileShare.shareToken,
}
}

View File

@@ -24,6 +24,7 @@ const schema = defineSchema({
"name",
"deletedAt",
]),
directories: defineTable({
name: v.string(),
userId: v.string(), // BetterAuth user IDs are strings, not Convex Ids
@@ -40,13 +41,21 @@ const schema = defineSchema({
"name",
"deletedAt",
]),
apiKeys: defineTable({
publicId: v.string(),
hashedKey: v.string(),
createdAt: v.number(),
updatedAt: v.number(),
expiresAt: v.optional(v.number()),
}).index("byPublicId", ["publicId"]),
fileShares: defineTable({
shareToken: v.string(),
storageId: v.id("_storage"),
expiresAt: v.optional(v.number()),
})
.index("byShareToken", ["shareToken"])
.index("byExpiredAt", ["expiresAt"])
.index("byStorageId", ["storageId"]),
})
export default schema

View File

@@ -57,6 +57,11 @@ export type FileHandle = {
export type FileSystemHandle = DirectoryHandle | FileHandle
export type OpenedFile = {
file: Doc<"files">
shareToken: string
}
export type DeleteResult = {
deleted: {
files: number