appflare 0.0.28 → 0.1.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (151) hide show
  1. package/cli/commands/index.ts +140 -0
  2. package/cli/generate.ts +149 -0
  3. package/cli/index.ts +56 -447
  4. package/cli/load-config.ts +182 -0
  5. package/cli/schema-compiler.ts +657 -0
  6. package/cli/templates/auth/README.md +156 -0
  7. package/cli/templates/auth/config.ts +61 -0
  8. package/cli/templates/auth/route-config.ts +18 -0
  9. package/cli/templates/auth/route-handler.ts +18 -0
  10. package/cli/templates/auth/route-request-utils.ts +55 -0
  11. package/cli/templates/auth/route.ts +14 -0
  12. package/cli/templates/core/README.md +266 -0
  13. package/cli/templates/core/app-creation.ts +19 -0
  14. package/cli/templates/core/client/appflare.ts +37 -0
  15. package/cli/templates/core/client/index.ts +6 -0
  16. package/cli/templates/core/client/storage.ts +100 -0
  17. package/cli/templates/core/client/types.ts +54 -0
  18. package/cli/templates/core/client-modules/appflare.ts +112 -0
  19. package/cli/templates/core/client-modules/handlers/index.ts +747 -0
  20. package/cli/templates/core/client-modules/handlers.ts +1 -0
  21. package/cli/templates/core/client-modules/index.ts +7 -0
  22. package/cli/templates/core/client-modules/storage.ts +180 -0
  23. package/cli/templates/core/client-modules/types.ts +145 -0
  24. package/cli/templates/core/client.ts +39 -0
  25. package/cli/templates/core/drizzle.ts +15 -0
  26. package/cli/templates/core/export.ts +14 -0
  27. package/cli/templates/core/handlers-route.ts +23 -0
  28. package/cli/templates/core/handlers.ts +1 -0
  29. package/cli/templates/core/imports.ts +8 -0
  30. package/cli/templates/core/server.ts +38 -0
  31. package/cli/templates/core/types.ts +6 -0
  32. package/cli/templates/core/wrangler.ts +109 -0
  33. package/cli/templates/handlers/README.md +353 -0
  34. package/cli/templates/handlers/auth.ts +37 -0
  35. package/cli/templates/handlers/execution.ts +39 -0
  36. package/cli/templates/handlers/generators/context/context-creation.ts +80 -0
  37. package/cli/templates/handlers/generators/context/error-helpers.ts +11 -0
  38. package/cli/templates/handlers/generators/context/scheduler.ts +24 -0
  39. package/cli/templates/handlers/generators/context/storage-api.ts +112 -0
  40. package/cli/templates/handlers/generators/context/storage-helpers.ts +59 -0
  41. package/cli/templates/handlers/generators/context/types.ts +18 -0
  42. package/cli/templates/handlers/generators/context.ts +43 -0
  43. package/cli/templates/handlers/generators/execution.ts +15 -0
  44. package/cli/templates/handlers/generators/handlers.ts +13 -0
  45. package/cli/templates/handlers/generators/registration/modules/cron.ts +30 -0
  46. package/cli/templates/handlers/generators/registration/modules/realtime.ts +631 -0
  47. package/cli/templates/handlers/generators/registration/modules/scheduler.ts +56 -0
  48. package/cli/templates/handlers/generators/registration/modules/storage.ts +194 -0
  49. package/cli/templates/handlers/generators/registration/sections.ts +210 -0
  50. package/cli/templates/handlers/generators/types/context.ts +64 -0
  51. package/cli/templates/handlers/generators/types/core.ts +101 -0
  52. package/cli/templates/handlers/generators/types/operations.ts +135 -0
  53. package/cli/templates/handlers/generators/types/query-definitions.ts +1029 -0
  54. package/cli/templates/handlers/generators/types/query-runtime.ts +417 -0
  55. package/cli/templates/handlers/index.ts +43 -0
  56. package/cli/templates/handlers/operations.ts +116 -0
  57. package/cli/templates/handlers/registration.ts +83 -0
  58. package/cli/templates/handlers/types.ts +15 -0
  59. package/cli/templates/handlers/utils.ts +48 -0
  60. package/cli/types.ts +108 -0
  61. package/cli/utils/handler-discovery.ts +366 -0
  62. package/cli/utils/json-utils.ts +24 -0
  63. package/cli/utils/path-utils.ts +19 -0
  64. package/cli/utils/schema-discovery.ts +390 -0
  65. package/index.ts +27 -4
  66. package/package.json +23 -20
  67. package/react/index.ts +5 -3
  68. package/react/use-infinite-query.ts +190 -0
  69. package/react/use-mutation.ts +54 -0
  70. package/react/use-query.ts +158 -0
  71. package/schema.ts +262 -0
  72. package/tsconfig.json +2 -4
  73. package/cli/README.md +0 -108
  74. package/cli/core/build.ts +0 -187
  75. package/cli/core/config.ts +0 -92
  76. package/cli/core/discover-handlers.ts +0 -143
  77. package/cli/core/handlers.ts +0 -7
  78. package/cli/core/index.ts +0 -205
  79. package/cli/generators/generate-api-client/client.ts +0 -163
  80. package/cli/generators/generate-api-client/extract-configuration.ts +0 -121
  81. package/cli/generators/generate-api-client/index.ts +0 -973
  82. package/cli/generators/generate-api-client/types.ts +0 -164
  83. package/cli/generators/generate-api-client/utils.ts +0 -22
  84. package/cli/generators/generate-api-client.ts +0 -1
  85. package/cli/generators/generate-cloudflare-worker/helpers.ts +0 -24
  86. package/cli/generators/generate-cloudflare-worker/index.ts +0 -2
  87. package/cli/generators/generate-cloudflare-worker/worker.ts +0 -148
  88. package/cli/generators/generate-cloudflare-worker/wrangler.ts +0 -108
  89. package/cli/generators/generate-cloudflare-worker.ts +0 -4
  90. package/cli/generators/generate-cron-handlers/cron-handlers-block.ts +0 -2
  91. package/cli/generators/generate-cron-handlers/handler-entries.ts +0 -29
  92. package/cli/generators/generate-cron-handlers/index.ts +0 -61
  93. package/cli/generators/generate-cron-handlers/runtime-block.ts +0 -49
  94. package/cli/generators/generate-cron-handlers/type-helpers-block.ts +0 -60
  95. package/cli/generators/generate-db-handlers/index.ts +0 -33
  96. package/cli/generators/generate-db-handlers/prepare.ts +0 -24
  97. package/cli/generators/generate-db-handlers/templates.ts +0 -189
  98. package/cli/generators/generate-db-handlers.ts +0 -1
  99. package/cli/generators/generate-hono-server/auth.ts +0 -97
  100. package/cli/generators/generate-hono-server/imports.ts +0 -55
  101. package/cli/generators/generate-hono-server/index.ts +0 -52
  102. package/cli/generators/generate-hono-server/routes.ts +0 -115
  103. package/cli/generators/generate-hono-server/template.ts +0 -371
  104. package/cli/generators/generate-hono-server.ts +0 -1
  105. package/cli/generators/generate-scheduler-handlers/constants.ts +0 -8
  106. package/cli/generators/generate-scheduler-handlers/handler-entries.ts +0 -22
  107. package/cli/generators/generate-scheduler-handlers/index.ts +0 -51
  108. package/cli/generators/generate-scheduler-handlers/runtime-block.ts +0 -68
  109. package/cli/generators/generate-scheduler-handlers/scheduler-handlers-block.ts +0 -2
  110. package/cli/generators/generate-scheduler-handlers/type-helpers-block.ts +0 -68
  111. package/cli/generators/generate-scheduler-handlers.ts +0 -1
  112. package/cli/generators/generate-websocket-durable-object/auth.ts +0 -30
  113. package/cli/generators/generate-websocket-durable-object/imports.ts +0 -55
  114. package/cli/generators/generate-websocket-durable-object/index.ts +0 -41
  115. package/cli/generators/generate-websocket-durable-object/query-handlers.ts +0 -18
  116. package/cli/generators/generate-websocket-durable-object/template.ts +0 -714
  117. package/cli/generators/generate-websocket-durable-object.ts +0 -1
  118. package/cli/schema/schema-static-types.ts +0 -702
  119. package/cli/schema/schema.ts +0 -151
  120. package/cli/utils/tsc.ts +0 -54
  121. package/cli/utils/utils.ts +0 -190
  122. package/cli/utils/zod-utils.ts +0 -121
  123. package/lib/README.md +0 -50
  124. package/lib/db.ts +0 -19
  125. package/lib/location.ts +0 -110
  126. package/lib/values.ts +0 -27
  127. package/react/README.md +0 -67
  128. package/react/hooks/useMutation.ts +0 -89
  129. package/react/hooks/usePaginatedQuery.ts +0 -213
  130. package/react/hooks/useQuery.ts +0 -106
  131. package/react/shared/queryShared.ts +0 -174
  132. package/server/README.md +0 -218
  133. package/server/auth.ts +0 -107
  134. package/server/database/builders.ts +0 -83
  135. package/server/database/context.ts +0 -327
  136. package/server/database/populate.ts +0 -234
  137. package/server/database/query-builder.ts +0 -161
  138. package/server/database/query-utils.ts +0 -25
  139. package/server/db.ts +0 -2
  140. package/server/storage/auth.ts +0 -16
  141. package/server/storage/bucket.ts +0 -22
  142. package/server/storage/context.ts +0 -34
  143. package/server/storage/index.ts +0 -38
  144. package/server/storage/operations.ts +0 -149
  145. package/server/storage/route-handler.ts +0 -60
  146. package/server/storage/types.ts +0 -55
  147. package/server/storage/utils.ts +0 -47
  148. package/server/storage.ts +0 -6
  149. package/server/types/schema-refs.ts +0 -66
  150. package/server/types/types.ts +0 -633
  151. package/server/utils/id-utils.ts +0 -230
@@ -1,164 +0,0 @@
1
- import { DiscoveredHandler } from "../../utils/utils";
2
- import { handlerTypePrefix, renderObjectKey } from "./utils";
3
-
4
- type PathTree<T> = {
5
- leaf?: { path: string; items: T[] };
6
- children: Map<string, PathTree<T>>;
7
- };
8
-
9
- const buildPathTree = <T>(byPath: Map<string, T[]>): PathTree<T> => {
10
- const root: PathTree<T> = { children: new Map() };
11
- for (const [path, items] of Array.from(byPath.entries())) {
12
- const segments = path.split("/").filter(Boolean);
13
- let node = root;
14
- for (const segment of segments) {
15
- if (!node.children.has(segment)) {
16
- node.children.set(segment, { children: new Map() });
17
- }
18
- node = node.children.get(segment)!;
19
- }
20
- node.leaf = { path, items };
21
- }
22
- return root;
23
- };
24
-
25
- const indent = (depth: number): string => "\t".repeat(depth);
26
-
27
- const renderPathTreeLines = <T>(
28
- node: PathTree<T>,
29
- depth: number,
30
- renderLeaf: (leaf: { path: string; items: T[] }, depth: number) => string[]
31
- ): string[] => {
32
- const lines: string[] = [];
33
- if (node.leaf) {
34
- lines.push(...renderLeaf(node.leaf, depth));
35
- }
36
- const children = Array.from(node.children.entries()).sort((a, b) =>
37
- a[0].localeCompare(b[0])
38
- );
39
- for (const [segment, child] of children) {
40
- lines.push(`${indent(depth + 1)}${renderObjectKey(segment)}: {`);
41
- lines.push(...renderPathTreeLines(child, depth + 1, renderLeaf));
42
- lines.push(`${indent(depth + 1)}}`);
43
- }
44
- return lines;
45
- };
46
-
47
- export function generateTypeBlocks(
48
- handlers: DiscoveredHandler[],
49
- importAliasBySource: Map<string, string>
50
- ): string[] {
51
- const typeBlocks: string[] = [];
52
- for (const h of handlers) {
53
- const importAlias = importAliasBySource.get(h.sourceFileAbs)!;
54
- const handlerAccessor = `${importAlias}[${JSON.stringify(h.name)}]`;
55
- const pascal = handlerTypePrefix(h);
56
- typeBlocks.push(
57
- `type ${pascal}Definition = typeof ${handlerAccessor};\n` +
58
- `type ${pascal}Args = HandlerArgs<${pascal}Definition>;\n` +
59
- `type ${pascal}Result = HandlerResult<${pascal}Definition>;\n` +
60
- `type ${pascal}Client = AppflareHandler<${pascal}Definition>;`
61
- );
62
- }
63
- return typeBlocks;
64
- }
65
-
66
- export function generateQueriesTypeLines(
67
- queriesByFile: Map<string, DiscoveredHandler[]>
68
- ): string {
69
- const tree = buildPathTree(queriesByFile);
70
- const renderLeaf = (
71
- leaf: { path: string; items: DiscoveredHandler[] },
72
- depth: number
73
- ): string[] => {
74
- const inner = leaf.items
75
- .slice()
76
- .sort((a, b) => a.name.localeCompare(b.name))
77
- .map((h) => {
78
- const pascal = handlerTypePrefix(h);
79
- return `${indent(depth + 1)}${h.name}: ${pascal}Client;`;
80
- })
81
- .join("\n");
82
-
83
- return inner ? [inner] : [`${indent(depth + 1)}// (none)`];
84
- };
85
-
86
- const lines: string[] = [];
87
- const children = Array.from(tree.children.entries()).sort((a, b) =>
88
- a[0].localeCompare(b[0])
89
- );
90
- for (const [segment, child] of children) {
91
- lines.push(`${indent(1)}${renderObjectKey(segment)}: {`);
92
- lines.push(...renderPathTreeLines(child, 1, renderLeaf));
93
- lines.push(`${indent(1)}}`);
94
- }
95
-
96
- return lines.join("\n");
97
- }
98
-
99
- export function generateMutationsTypeLines(
100
- mutationsByFile: Map<string, DiscoveredHandler[]>
101
- ): string {
102
- const tree = buildPathTree(mutationsByFile);
103
- const renderLeaf = (
104
- leaf: { path: string; items: DiscoveredHandler[] },
105
- depth: number
106
- ): string[] => {
107
- const inner = leaf.items
108
- .slice()
109
- .sort((a, b) => a.name.localeCompare(b.name))
110
- .map((h) => {
111
- const pascal = handlerTypePrefix(h);
112
- return `${indent(depth + 1)}${h.name}: ${pascal}Client;`;
113
- })
114
- .join("\n");
115
-
116
- return inner ? [inner] : [`${indent(depth + 1)}// (none)`];
117
- };
118
-
119
- const lines: string[] = [];
120
- const children = Array.from(tree.children.entries()).sort((a, b) =>
121
- a[0].localeCompare(b[0])
122
- );
123
- for (const [segment, child] of children) {
124
- lines.push(`${indent(1)}${renderObjectKey(segment)}: {`);
125
- lines.push(...renderPathTreeLines(child, 1, renderLeaf));
126
- lines.push(`${indent(1)}}`);
127
- }
128
-
129
- return lines.join("\n");
130
- }
131
-
132
- export function generateInternalTypeLines(
133
- internalByFile: Map<string, DiscoveredHandler[]>,
134
- importAliasBySource: Map<string, string>
135
- ): string {
136
- const tree = buildPathTree(internalByFile);
137
- const renderLeaf = (
138
- leaf: { path: string; items: DiscoveredHandler[] },
139
- depth: number
140
- ): string[] => {
141
- const inner = leaf.items
142
- .slice()
143
- .sort((a, b) => a.name.localeCompare(b.name))
144
- .map((h) => {
145
- const alias = importAliasBySource.get(h.sourceFileAbs)!;
146
- return `${indent(depth + 1)}${h.name}: typeof ${alias}[${JSON.stringify(h.name)}];`;
147
- })
148
- .join("\n");
149
-
150
- return inner ? [inner] : [`${indent(depth + 1)}// (none)`];
151
- };
152
-
153
- const lines: string[] = [];
154
- const children = Array.from(tree.children.entries()).sort((a, b) =>
155
- a[0].localeCompare(b[0])
156
- );
157
- for (const [segment, child] of children) {
158
- lines.push(`${indent(1)}${renderObjectKey(segment)}: {`);
159
- lines.push(...renderPathTreeLines(child, 1, renderLeaf));
160
- lines.push(`${indent(1)}}`);
161
- }
162
-
163
- return lines.join("\n");
164
- }
@@ -1,22 +0,0 @@
1
- import { isValidIdentifier } from "../../utils/utils";
2
-
3
- export const sortedEntries = <T>(map: Map<string, T[]>): Array<[string, T[]]> =>
4
- Array.from(map.entries()).sort(([a], [b]) => a.localeCompare(b));
5
-
6
- export const renderObjectKey = (key: string): string =>
7
- isValidIdentifier(key) ? key : JSON.stringify(key);
8
-
9
- export const handlerTypePrefix = (h: any): string => {
10
- const base = (h.routePath ?? h.fileName) as string;
11
- return (
12
- base
13
- .replace(/[^a-zA-Z0-9]/g, "")
14
- .replace(/^./, (c: string) => c.toUpperCase()) +
15
- (h.name as string)
16
- .replace(/[^a-zA-Z0-9]/g, "")
17
- .replace(/^./, (c: string) => c.toUpperCase())
18
- );
19
- };
20
-
21
- export const normalizeTableName = (fileName: string): string =>
22
- fileName.endsWith("s") ? fileName : `${fileName}s`;
@@ -1 +0,0 @@
1
- export { generateApiClient } from "./generate-api-client/index";
@@ -1,24 +0,0 @@
1
- import path from "node:path";
2
-
3
- export const DEFAULT_ALLOWED_ORIGINS = ["http://localhost:3000"];
4
-
5
- export const DEFAULT_SCHEDULER_QUEUE_BINDING = "APPFLARE_SCHEDULER_QUEUE";
6
-
7
- export const resolveAllowedOrigins = (origins?: string[]): string[] =>
8
- origins && origins.length > 0 ? origins : DEFAULT_ALLOWED_ORIGINS;
9
-
10
- export const sanitizeWorkerName = (configDirAbs: string): string => {
11
- const base = path.basename(configDirAbs);
12
- const slug = base.replace(/[^A-Za-z0-9_-]/g, "-").toLowerCase();
13
- return slug || "appflare-worker";
14
- };
15
-
16
- export const toBucketName = (binding: string): string =>
17
- binding.toLowerCase().replace(/_/g, "-") || "appflare-storage";
18
-
19
- export const toQueueName = (base: string): string =>
20
- base
21
- .toLowerCase()
22
- .replace(/[^a-z0-9-]/g, "-")
23
- .replace(/--+/g, "-")
24
- .replace(/^-+|-+$/g, "") || "appflare-scheduler";
@@ -1,2 +0,0 @@
1
- export { generateCloudflareWorkerIndex } from "./worker";
2
- export { generateWranglerJson } from "./wrangler";
@@ -1,148 +0,0 @@
1
- import { resolveAllowedOrigins } from "./helpers";
2
-
3
- export function generateCloudflareWorkerIndex(params: {
4
- allowedOrigins?: string[];
5
- hasCronHandlers?: boolean;
6
- }): string {
7
- const allowedOrigins = resolveAllowedOrigins(params.allowedOrigins);
8
- const allowedOriginsCsv = allowedOrigins.join(",");
9
- const cronImports = params.hasCronHandlers
10
- ? 'import { handleCron } from "./cron";\n'
11
- : "";
12
- const scheduledBlock = params.hasCronHandlers
13
- ? `\n\tasync scheduled(controller, env, ctx): Promise<void> {\n\t\tawait handleCron({ controller, env, ctx });\n\t},\n`
14
- : "";
15
-
16
- return `/* eslint-disable */
17
- /**
18
- * This file is auto-generated by the Appflare CLI.
19
- * Do not edit directly.
20
- */
21
-
22
- import { createAppflareHonoServer } from "./server";
23
- import { WebSocketHibernationServer } from "./websocket-hibernation-server";
24
- import { createScheduler, handleSchedulerBatch } from "./scheduler";
25
- ${cronImports}import { getDatabase } from "cloudflare-do-mongo";
26
- import { MONGO_DURABLE_OBJECT } from "cloudflare-do-mongo/do";
27
- import type { Hono } from "hono";
28
- import { cors } from "hono/cors";
29
- import { Db } from "mongodb";
30
-
31
- type DurableObjectNamespaceLike = {
32
- idFromName(name: string): any;
33
- get(id: any): { fetch(input: any, init?: RequestInit): Promise<Response> };
34
- };
35
-
36
- type Env = {
37
- MONGO_DB: unknown;
38
- MONGO_URI?: string;
39
- WEBSOCKET_HIBERNATION_SERVER: DurableObjectNamespaceLike;
40
- MONGO_DURABLE_OBJECT: DurableObjectNamespaceLike;
41
- APPFLARE_STORAGE?: unknown;
42
- ALLOWED_ORIGINS?: string;
43
- APPFLARE_SCHEDULER_QUEUE?: {
44
- send: (body: unknown, options?: { delaySeconds?: number }) => Promise<void>;
45
- };
46
- };
47
-
48
- type WorkerEnv = { Bindings: Env };
49
-
50
- const parseAllowedOrigins = (value?: string | null): string[] =>
51
- (value ?? ${JSON.stringify(allowedOriginsCsv)})
52
- .split(",")
53
- .map((origin) => origin.trim())
54
- .filter(Boolean);
55
-
56
- const resolveCorsOrigin = (
57
- origin: string | null,
58
- allowed: string[]
59
- ): string | undefined => {
60
- if (!origin) return undefined;
61
- if (allowed.includes("*")) return origin;
62
- return allowed.includes(origin) ? origin : undefined;
63
- };
64
-
65
- export default {
66
- async fetch(request, env, ctx): Promise<Response> {
67
- const allowedOrigins = parseAllowedOrigins(env.ALLOWED_ORIGINS);
68
- const resolveOrigin = (origin: string | null) =>
69
- resolveCorsOrigin(origin, allowedOrigins);
70
- const scheduler = env.APPFLARE_SCHEDULER_QUEUE
71
- ? createScheduler(env.APPFLARE_SCHEDULER_QUEUE)
72
- : undefined;
73
-
74
- const app = createAppflareHonoServer({
75
- db: getDatabase(env.MONGO_DB) as unknown as Db,
76
- corsOrigin: allowedOrigins,
77
- realtime: {
78
- durableObject: env.WEBSOCKET_HIBERNATION_SERVER,
79
- durableObjectName: "primary",
80
- notify: async (payload) => {
81
- const id = env.WEBSOCKET_HIBERNATION_SERVER.idFromName("primary");
82
- const stub = env.WEBSOCKET_HIBERNATION_SERVER.get(id);
83
- await stub.fetch("http://appflare-realtime/notify", {
84
- method: "POST",
85
- headers: { "content-type": "application/json" },
86
- body: JSON.stringify(payload),
87
- });
88
- },
89
- },
90
- scheduler,
91
- }) as unknown as Hono<WorkerEnv>;
92
-
93
- app.use(
94
- "*",
95
- cors({
96
- origin: resolveOrigin,
97
- credentials: true,
98
- allowMethods: ["GET", "POST", "PUT", "DELETE", "OPTIONS"],
99
- allowHeaders: ["Content-Type", "Authorization", "Cookie"],
100
- exposeHeaders: ["set-cookie"],
101
- })
102
- );
103
-
104
- const origin = request.headers.get("Origin");
105
- const allowedOrigin = resolveOrigin(origin);
106
- if (request.method === "OPTIONS") {
107
- return new Response(null, {
108
- status: 204,
109
- headers: {
110
- "Access-Control-Allow-Origin": allowedOrigin ?? "",
111
- "Access-Control-Allow-Credentials": "true",
112
- "Access-Control-Allow-Methods": "GET,POST,PUT,DELETE,OPTIONS",
113
- "Access-Control-Allow-Headers":
114
- request.headers.get("Access-Control-Request-Headers") ??
115
- "Content-Type, Authorization, Cookie",
116
- Vary: "Origin",
117
- },
118
- });
119
- }
120
-
121
- const upgradeHeader = request.headers.get("Upgrade");
122
- if (upgradeHeader === "websocket") {
123
- const url = new URL(request.url);
124
- if (url.pathname === "/ws") {
125
- const id = env.WEBSOCKET_HIBERNATION_SERVER.idFromName("primary");
126
- const stub = env.WEBSOCKET_HIBERNATION_SERVER.get(id);
127
- return stub.fetch(request);
128
- }
129
- }
130
-
131
- const response = await app.fetch(request, env, ctx);
132
- if (allowedOrigin) {
133
- response.headers.set("Access-Control-Allow-Origin", allowedOrigin);
134
- response.headers.set("Access-Control-Allow-Credentials", "true");
135
- response.headers.append("Vary", "Origin");
136
- }
137
- return response;
138
- },
139
-
140
- ${scheduledBlock}
141
- async queue(batch, env): Promise<void> {
142
- await handleSchedulerBatch({ batch, env });
143
- },
144
- } satisfies ExportedHandler<Env>;
145
-
146
- export { MONGO_DURABLE_OBJECT, WebSocketHibernationServer };
147
- `;
148
- }
@@ -1,108 +0,0 @@
1
- import type { AppflareConfig } from "../../utils/utils";
2
- import {
3
- DEFAULT_SCHEDULER_QUEUE_BINDING,
4
- resolveAllowedOrigins,
5
- sanitizeWorkerName,
6
- toBucketName,
7
- toQueueName,
8
- } from "./helpers";
9
-
10
- export function generateWranglerJson(params: {
11
- config: AppflareConfig;
12
- configDirAbs: string;
13
- allowedOrigins?: string[];
14
- cronTriggers?: string[];
15
- }): string {
16
- const allowedOrigins = resolveAllowedOrigins(params.allowedOrigins);
17
- const bucketBinding =
18
- params.config.storage?.bucketBinding ?? "APPFLARE_STORAGE";
19
- const r2Buckets = params.config.storage
20
- ? [
21
- {
22
- binding: bucketBinding,
23
- bucket_name: toBucketName(bucketBinding),
24
- },
25
- ]
26
- : undefined;
27
-
28
- const schedulerEnabled = params.config.scheduler?.enabled !== false;
29
- const schedulerQueueBinding =
30
- params.config.scheduler?.queueBinding ?? DEFAULT_SCHEDULER_QUEUE_BINDING;
31
- const schedulerQueueName = params.config.scheduler?.queueName
32
- ? toQueueName(params.config.scheduler.queueName)
33
- : `${toQueueName(sanitizeWorkerName(params.configDirAbs))}-scheduler`;
34
-
35
- const workerMain = params.config.wranglerMain ?? "./server/index.ts";
36
- const compatibilityDate =
37
- params.config.wranglerCompatibilityDate ??
38
- new Date().toISOString().slice(0, 10);
39
-
40
- const wrangler: Record<string, unknown> = {
41
- $schema: "node_modules/wrangler/config-schema.json",
42
- name:
43
- params.config.wrangler?.name ?? sanitizeWorkerName(params.configDirAbs),
44
- main: params.config.wrangler?.main ?? workerMain,
45
- compatibility_date:
46
- params.config.wrangler?.compatibilityDate ?? compatibilityDate,
47
- compatibility_flags: [
48
- "nodejs_compat",
49
- "nodejs_compat_populate_process_env",
50
- ],
51
- migrations: [
52
- { new_sqlite_classes: ["WebSocketHibernationServer"], tag: "v1" },
53
- { new_sqlite_classes: ["MONGO_DURABLE_OBJECT"], tag: "v2" },
54
- ],
55
- durable_objects: {
56
- bindings: [
57
- {
58
- class_name: "WebSocketHibernationServer",
59
- name: "WEBSOCKET_HIBERNATION_SERVER",
60
- },
61
- { class_name: "MONGO_DURABLE_OBJECT", name: "MONGO_DURABLE_OBJECT" },
62
- ],
63
- },
64
- observability: { enabled: true },
65
- placement: { mode: "smart" },
66
- vars: {
67
- ALLOWED_ORIGINS: allowedOrigins.join(","),
68
- },
69
- ...params.config.wrangler,
70
- };
71
-
72
- if (r2Buckets && r2Buckets.length > 0) {
73
- wrangler.r2_buckets = r2Buckets;
74
- }
75
-
76
- if (schedulerEnabled) {
77
- wrangler.queues = {
78
- producers: [
79
- {
80
- binding: schedulerQueueBinding,
81
- queue: schedulerQueueName,
82
- },
83
- ],
84
- consumers: [
85
- {
86
- queue: schedulerQueueName,
87
- },
88
- ],
89
- };
90
- }
91
-
92
- if (params.config.storage?.kvBinding) {
93
- wrangler.kv_namespaces = [
94
- {
95
- binding: params.config.storage.kvBinding,
96
- id: params.config.storage.kvId ?? "",
97
- },
98
- ];
99
- }
100
-
101
- if (params.cronTriggers && params.cronTriggers.length > 0) {
102
- wrangler.triggers = {
103
- crons: Array.from(new Set(params.cronTriggers)),
104
- };
105
- }
106
-
107
- return `${JSON.stringify(wrangler, null, 2)}\n`;
108
- }
@@ -1,4 +0,0 @@
1
- export {
2
- generateCloudflareWorkerIndex,
3
- generateWranglerJson,
4
- } from "./generate-cloudflare-worker/index";
@@ -1,2 +0,0 @@
1
- export const buildCronHandlersBlock = (handlerEntries: string): string =>
2
- `const cronHandlers = {\n${handlerEntries}\n} as const;`;
@@ -1,29 +0,0 @@
1
- import type { DiscoveredHandler } from "../../utils/utils";
2
-
3
- const stringifyTriggers = (triggers?: string[]): string => {
4
- if (!triggers || triggers.length === 0) return "[]";
5
- return `[${triggers.map((value) => JSON.stringify(value)).join(", ")}]`;
6
- };
7
-
8
- export const buildCronHandlerEntries = (params: {
9
- handlers: DiscoveredHandler[];
10
- localNameFor: (handler: DiscoveredHandler) => string;
11
- }): string => {
12
- if (params.handlers.length === 0) return "";
13
-
14
- return params.handlers
15
- .map((handler) => {
16
- const local = params.localNameFor(handler);
17
- const task = `${handler.routePath}/${handler.name}`;
18
- const fallbackTriggers = stringifyTriggers(handler.cronTriggers);
19
- return (
20
- `\t${JSON.stringify(task)}: {\n` +
21
- `\t\tfile: ${JSON.stringify(handler.routePath)},\n` +
22
- `\t\tname: ${JSON.stringify(handler.name)},\n` +
23
- `\t\tcronTrigger: ${local}.cronTrigger ?? ${fallbackTriggers},\n` +
24
- `\t\trun: ${local}.handler,\n` +
25
- `\t},`
26
- );
27
- })
28
- .join("\n");
29
- };
@@ -1,61 +0,0 @@
1
- import { buildImportSection } from "../generate-hono-server/imports";
2
- import type { DiscoveredHandler } from "../../utils/utils";
3
- import {
4
- schemaTypesImportPath,
5
- serverImportPath,
6
- filePreamble,
7
- } from "../generate-scheduler-handlers/constants";
8
- import { buildCronHandlerEntries } from "./handler-entries";
9
- import { buildCronHandlersBlock } from "./cron-handlers-block";
10
- import { buildTypeHelpersBlock } from "./type-helpers-block";
11
- import { buildRuntimeBlock } from "./runtime-block";
12
-
13
- export function generateCronHandlers(params: {
14
- handlers: DiscoveredHandler[];
15
- outDirAbs: string;
16
- schemaPathAbs: string;
17
- configPathAbs: string;
18
- }): { code: string; cronTriggers: string[] } {
19
- const cronHandlers = params.handlers.filter(
20
- (handler) => handler.kind === "cron"
21
- );
22
-
23
- const imports = buildImportSection({
24
- handlers: cronHandlers,
25
- outDirAbs: params.outDirAbs,
26
- schemaPathAbs: params.schemaPathAbs,
27
- configPathAbs: params.configPathAbs,
28
- });
29
-
30
- const handlerImportBlock =
31
- imports.handlerImports.length > 0
32
- ? `${imports.handlerImports.join("\n")}\n\n`
33
- : "";
34
-
35
- const handlerEntries = buildCronHandlerEntries({
36
- handlers: cronHandlers,
37
- localNameFor: imports.localNameFor,
38
- });
39
-
40
- const importsBlock = `import { createAppflareDbContext, type AppflareDbContext } from ${JSON.stringify(serverImportPath)};\nimport type {\n\tScheduler,\n\tSchedulerEnqueueOptions,\n\tSchedulerPayload,\n\tSchedulerTaskName,\n\tAppflareAuthContext,\n\tAppflareAuthSession,\n\tAppflareAuthUser,\n} from ${JSON.stringify(schemaTypesImportPath)};\nimport { createScheduler } from "./scheduler";\nimport { getDatabase } from "cloudflare-do-mongo";\nimport { Db } from "mongodb";\n${handlerImportBlock}`;
41
-
42
- const code = [
43
- filePreamble,
44
- "",
45
- importsBlock,
46
- buildCronHandlersBlock(handlerEntries),
47
- buildTypeHelpersBlock(),
48
- buildRuntimeBlock(),
49
- "export type { CronTaskName };\n",
50
- ].join("\n");
51
-
52
- const cronTriggers = Array.from(
53
- new Set(
54
- cronHandlers
55
- .flatMap((handler) => handler.cronTriggers ?? [])
56
- .filter((value): value is string => Boolean(value))
57
- )
58
- );
59
-
60
- return { code, cronTriggers };
61
- }
@@ -1,49 +0,0 @@
1
- export const buildRuntimeBlock = (): string => `
2
- const normalizeCronTrigger = (value: CronTriggerValue): string[] => {
3
- if (!value) return [];
4
- if (typeof value === "string") return [value.trim()].filter(Boolean);
5
- return value.map((entry) => String(entry).trim()).filter(Boolean);
6
- };
7
-
8
- export async function handleCron(params: {
9
- controller: { cron: string };
10
- env: Env;
11
- ctx: { waitUntil(promise: Promise<unknown>): void };
12
- }): Promise<void> {
13
- const { controller, env, ctx } = params;
14
- const db = createAppflareDbContext({
15
- db: getDatabase(env.MONGO_DB) as unknown as Db,
16
- });
17
-
18
- const scheduler = env.APPFLARE_SCHEDULER_QUEUE
19
- ? createScheduler(env.APPFLARE_SCHEDULER_QUEUE)
20
- : undefined;
21
-
22
- const baseContext: CronHandlerContext = {
23
- ...emptyAuthContext,
24
- db,
25
- env,
26
- scheduler,
27
- controller: {
28
- cron: controller.cron,
29
- scheduledTime: (controller as { scheduledTime?: number }).scheduledTime,
30
- nextScheduledTime: (controller as { nextScheduledTime?: number })
31
- .nextScheduledTime,
32
- },
33
- ctx,
34
- };
35
-
36
- const handlerLookup = cronHandlers as unknown as CronHandlerLookup;
37
- const cronValue = controller.cron;
38
-
39
- for (const [task, handler] of Object.entries(handlerLookup)) {
40
- const triggers = normalizeCronTrigger(handler.cronTrigger);
41
- if (!triggers.includes(cronValue)) continue;
42
- try {
43
- await handler.run(baseContext);
44
- } catch (err) {
45
- console.error("Cron task failed", task, err);
46
- }
47
- }
48
- }
49
- `;
@@ -1,60 +0,0 @@
1
- export const buildTypeHelpersBlock = (): string => `
2
- type CronHandlers = typeof cronHandlers;
3
-
4
- type CronTaskName = keyof CronHandlers extends never
5
- ? string
6
- : keyof CronHandlers;
7
-
8
- type CronTriggerValue = string | readonly string[] | undefined;
9
-
10
- type Env = {
11
- MONGO_DB: unknown;
12
- APPFLARE_SCHEDULER_QUEUE?: {
13
- send: (body: unknown, options?: { delaySeconds?: number }) => Promise<void>;
14
- };
15
- } & Record<string, unknown>;
16
-
17
- const emptyAuthContext: AppflareAuthContext = {
18
- session: null as AppflareAuthSession,
19
- user: null as AppflareAuthUser,
20
- };
21
-
22
- type SchedulerEnqueue = {
23
- <TTask extends SchedulerTaskName>(
24
- task: TTask,
25
- ...args: SchedulerPayload<TTask> extends undefined
26
- ? [payload?: SchedulerPayload<TTask>, options?: SchedulerEnqueueOptions]
27
- : [payload: SchedulerPayload<TTask>, options?: SchedulerEnqueueOptions]
28
- ): Promise<void>;
29
- };
30
-
31
- type TypedScheduler = Omit<Scheduler, "enqueue"> & {
32
- enqueue: SchedulerEnqueue;
33
- };
34
-
35
- type ScheduledController = {
36
- cron: string;
37
- scheduledTime?: number;
38
- nextScheduledTime?: number;
39
- };
40
-
41
- type ExecutionContext = {
42
- waitUntil(promise: Promise<unknown>): void;
43
- };
44
-
45
- type CronHandlerContext = AppflareAuthContext & {
46
- db: AppflareDbContext;
47
- env: Env;
48
- scheduler?: TypedScheduler;
49
- controller: ScheduledController;
50
- ctx: ExecutionContext;
51
- };
52
-
53
- type CronHandlerLookup = Record<
54
- string,
55
- {
56
- cronTrigger: CronTriggerValue;
57
- run: (ctx: CronHandlerContext) => Promise<void>;
58
- }
59
- >;
60
- `;