Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 16 additions & 0 deletions apps/cloud/@useautumn-sdk.d.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
// AUTO-GENERATED by atmn pull
// DO NOT EDIT MANUALLY

import type {} from "@useautumn/sdk";

declare module "@useautumn/sdk" {
// Features
export const seats: Feature;

// Plans
export const professional: Plan;

// Base types
export type Feature = import("./autumn.config").Feature;
export type Plan = import("./autumn.config").Plan;
}
100 changes: 100 additions & 0 deletions apps/cloud/autumn.config.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,100 @@
import { feature, item, plan } from "atmn";

// Features
export const seats = feature({
id: "seats",
name: "Seats",
type: "metered",
consumable: false,
});

export const executions = feature({
id: "executions",
name: "Executions",
type: "metered",
consumable: true,
});

// Plans
export const free = plan({
id: "free",
name: "Free",
autoEnable: true,
items: [
item({
featureId: executions.id,
included: 5000,
reset: { interval: "month" },
}),
],
});

export const hobby = plan({
id: "hobby",
name: "Hobby",
price: {
amount: 10,
interval: "month",
},
items: [
item({
featureId: seats.id,
included: 1,
price: {
amount: 10,
billingUnits: 1,
billingMethod: "usage_based",
interval: "month",
},
}),
item({
featureId: executions.id,
included: 50000,
reset: { interval: "month" },
}),
],
});

export const professional = plan({
id: "professional",
name: "Professional",
price: {
amount: 40,
interval: "month",
},
items: [
item({
featureId: seats.id,
included: 1,
price: {
amount: 40,
billingUnits: 1,
billingMethod: "usage_based",
interval: "month",
},
}),
item({
featureId: executions.id,
included: 100000,
reset: { interval: "month" },
}),
],
});

// Overage add-on
export const executionTopUp = plan({
id: "execution-top-up",
name: "Execution Top-Up",
addOn: true,
items: [
item({
featureId: executions.id,
price: {
amount: 1,
billingUnits: 10000,
billingMethod: "prepaid",
interval: "month",
},
}),
],
});
1 change: 1 addition & 0 deletions apps/cloud/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@
"@tanstack/react-start": "catalog:",
"@workos-inc/node": "^8.11.1",
"agents": "^0.10.0",
"autumn-js": "^1.2.8",
"drizzle-orm": "catalog:",
"effect": "catalog:",
"jose": "^5.6.3",
Expand Down
97 changes: 95 additions & 2 deletions apps/cloud/src/api.ts
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,8 @@ import {
HttpServer,
} from "@effect/platform";
import { Effect, Layer } from "effect";
import { Autumn } from "autumn-js";
import { autumnHandler } from "autumn-js/backend";

import { CoreExecutorApi } from "@executor/api";
import { CoreHandlers, ExecutorService, ExecutionEngineService } from "@executor/api/server";
Expand Down Expand Up @@ -155,6 +157,21 @@ const buildProtectedHandler = (
// ---------------------------------------------------------------------------

const isAuthPath = (pathname: string): boolean => pathname.startsWith("/auth/");
const isAutumnPath = (pathname: string): boolean => pathname.startsWith("/autumn/");
const isExecutionPath = (pathname: string): boolean =>
pathname === "/executions" || /^\/executions\/[^/]+\/resume$/.test(pathname);

// ---------------------------------------------------------------------------
// Autumn billing — lazy-initialized SDK for fire-and-forget tracking
// ---------------------------------------------------------------------------

let _autumn: Autumn | null = null;
const getAutumn = () => {
if (!_autumn && server.AUTUMN_SECRET_KEY) {
_autumn = new Autumn({ secretKey: server.AUTUMN_SECRET_KEY });
}
return _autumn;
};

/**
* Resolve the user's organization for executor creation. Reads from the
Expand All @@ -170,9 +187,67 @@ const lookupOrgForRequest = (request: Request) =>
return yield* users.use((s) => s.getOrganization(result.organizationId!));
});

// ---------------------------------------------------------------------------
// Autumn billing proxy — authenticates the session, then forwards to Autumn
// ---------------------------------------------------------------------------

const handleAutumnRequest = async (request: Request): Promise<Response> => {
const program = Effect.gen(function* () {
const workos = yield* WorkOSAuth;
const result = yield* workos.authenticateRequest(request);

if (!result || !result.organizationId) {
return Response.json({ error: "Unauthorized" }, { status: 401 });
}

const url = new URL(request.url);
const body =
request.method !== "GET" && request.method !== "HEAD"
? yield* Effect.promise(() => request.json())
: undefined;

const { statusCode, response } = yield* Effect.promise(() =>
autumnHandler({
request: {
url: url.pathname,
method: request.method,
body,
},
customerId: result.organizationId,
customerData: {
name: result.email,
email: result.email,
},
clientOptions: {
secretKey: server.AUTUMN_SECRET_KEY,
},
pathPrefix: "/autumn",
}),
);

if (statusCode >= 400) {
console.error("[autumn] upstream error:", statusCode, response);
return Response.json({ error: "Billing request failed" }, { status: statusCode });
}

return Response.json(response, { status: statusCode });
});

return Effect.runPromise(program.pipe(Effect.provide(SharedServices), Effect.scoped)).catch(
(err) => {
console.error("[autumn] request failed:", err instanceof Error ? err.stack : err);
return Response.json({ error: "Internal server error" }, { status: 500 });
},
);
};

export const handleApiRequest = async (request: Request): Promise<Response> => {
const pathname = new URL(request.url).pathname;

if (isAutumnPath(pathname)) {
return handleAutumnRequest(request);
}

if (isAuthPath(pathname)) {
const handler = createNonProtectedHandler();
try {
Expand All @@ -190,7 +265,8 @@ export const handleApiRequest = async (request: Request): Promise<Response> => {

const codeExecutor = makeDynamicWorkerExecutor({ loader: env.LOADER });
const handler = yield* buildProtectedHandler(org.id, org.name, codeExecutor);
return yield* Effect.promise(() => handler.handler(request));
const response = yield* Effect.promise(() => handler.handler(request));
return { response, orgId: org.id };
});

const result = await Effect.runPromise(
Expand All @@ -203,7 +279,24 @@ export const handleApiRequest = async (request: Request): Promise<Response> => {
{ status: 403 },
);
}
return result;

// Fire-and-forget: track execution usage
if (isExecutionPath(pathname) && result.response.ok) {
const autumn = getAutumn();
if (autumn) {
autumn
.track({
customerId: result.orgId,
featureId: "executions",
value: 1,
})
.catch((err) => {
console.error("[billing] track failed:", err);
});
}
}

return result.response;
} catch (err) {
console.error("[api] request failed:", err instanceof Error ? err.stack : err);
return Response.json(
Expand Down
2 changes: 2 additions & 0 deletions apps/cloud/src/env.ts
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ const serverShape = {
WORKOS_CLIENT_ID: Env.string("WORKOS_CLIENT_ID"),
WORKOS_COOKIE_PASSWORD: Env.string("WORKOS_COOKIE_PASSWORD"),
VITE_PUBLIC_SITE_URL: Env.stringOr("VITE_PUBLIC_SITE_URL", ""),
AUTUMN_SECRET_KEY: Env.stringOr("AUTUMN_SECRET_KEY", ""),
};

type SharedEnv = Readonly<{
Expand All @@ -25,6 +26,7 @@ type ServerEnv = SharedEnv &
WORKOS_CLIENT_ID: string;
WORKOS_COOKIE_PASSWORD: string;
VITE_PUBLIC_SITE_URL: string;
AUTUMN_SECRET_KEY: string;
}>;

type WebEnv = Readonly<Record<string, never>>;
Expand Down
Loading