1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
import { z } from "zod"
import { fn } from "./util/fn"
import { Actor } from "./actor"
import { Database } from "./drizzle"
import { Identifier } from "./identifier"
import { UserTable } from "./schema/user.sql"
import { BillingTable } from "./schema/billing.sql"
import { WorkspaceTable } from "./schema/workspace.sql"
import { Key } from "./key"
import { eq, sql } from "drizzle-orm"
export namespace Workspace {
export const create = fn(
z.object({
name: z.string().min(1),
}),
async ({ name }) => {
const account = Actor.assert("account")
const workspaceID = Identifier.create("workspace")
const userID = Identifier.create("user")
await Database.transaction(async (tx) => {
await tx.insert(WorkspaceTable).values({
id: workspaceID,
name,
})
await tx.insert(UserTable).values({
workspaceID,
id: userID,
accountID: account.properties.accountID,
name: "",
role: "admin",
})
await tx.insert(BillingTable).values({
workspaceID,
id: Identifier.create("billing"),
balance: 0,
})
})
await Actor.provide(
"system",
{
workspaceID,
},
() => Key.create({ userID, name: "Default API Key" }),
)
return workspaceID
},
)
export const update = fn(
z.object({
name: z.string().min(1).max(255),
}),
async ({ name }) => {
Actor.assertAdmin()
const workspaceID = Actor.workspace()
return await Database.use((tx) =>
tx
.update(WorkspaceTable)
.set({
name,
})
.where(eq(WorkspaceTable.id, workspaceID)),
)
},
)
export const remove = fn(z.void(), async () => {
await Database.use((tx) =>
tx
.update(WorkspaceTable)
.set({ timeDeleted: sql`now()` })
.where(eq(WorkspaceTable.id, Actor.workspace())),
)
})
}
|
