Cloudflare Developers•2y ago

how to setup rpc/entrypoints with workers for API access?

hi. so i have an api that is structured as below, and i was wondering how i would represent it within rpc or entrypoints. do i create a base entrypoint and make the rest RPC targets that can be used with a getter (option 1, doesn't seem too difficult), or can i create it with option 2 using classes on the base api class?

GET /reports
GET /reports/:id
POST /reports
GET /communities
GET /communities/:id

GET /reports
GET /reports/:id
POST /reports
GET /communities
GET /communities/:id

// option 1
const reportsRoute = await env.BACKEND.getReports()
const reports = await reportsRoute.getReport("id")

// option 2
const reports = await env.BACKEND.reports.getReport("id")

// option 1
const reportsRoute = await env.BACKEND.getReports()
const reports = await reportsRoute.getReport("id")

// option 2
const reports = await env.BACKEND.reports.getReport("id")

import type { CustomEnv } from "./types";
import { Kysely } from "kysely";
import { WorkerEntrypoint } from "cloudflare:workers";
import { SerializePlugin } from "kysely-plugin-serialize";
import type { DB } from "./db-types";
import { D1Dialect } from "./kysely-d1";

export class Categories {
    constructor(protected env: CustomEnv) {}
    /**
     * Get a single category by its ID
     * @param id ID of the category
     */
    async getCategory(id: string) {
        const category = await this.env.DB.selectFrom("Categories")
            .selectAll()
            .where("id", "=", id)
            .executeTakeFirst();
        return category ?? null;
    }

    /**
     * Get all categories
     */
    async getAllCategories() {
        const categories = await this.env.DB.selectFrom("Categories")
            .selectAll()
            .execute();
        return categories;
    }
}

export class FDGLBackend extends WorkerEntrypoint<Env> {
    categories: Categories;
    constructor(ctx: ExecutionContext, env: Env) {
        super(ctx, env);
        const customEnv: CustomEnv = {
            ...env,
            d1_db: env.DB,
            DB: new Kysely<DB>({
                dialect: new D1Dialect({ database: env.DB }),
                plugins: [new SerializePlugin()],
            }),
        };
        this.categories = new Categories(customEnv);
    }

    // required or wrangler crashes
    async fetch() {
        return new Response("ok");
    }
}

import type { CustomEnv } from "./types";
import { Kysely } from "kysely";
import { WorkerEntrypoint } from "cloudflare:workers";
import { SerializePlugin } from "kysely-plugin-serialize";
import type { DB } from "./db-types";
import { D1Dialect } from "./kysely-d1";

export class Categories {
    constructor(protected env: CustomEnv) {}
    /**
     * Get a single category by its ID
     * @param id ID of the category
     */
    async getCategory(id: string) {
        const category = await this.env.DB.selectFrom("Categories")
            .selectAll()
            .where("id", "=", id)
            .executeTakeFirst();
        return category ?? null;
    }

    /**
     * Get all categories
     */
    async getAllCategories() {
        const categories = await this.env.DB.selectFrom("Categories")
            .selectAll()
            .execute();
        return categories;
    }
}

export class FDGLBackend extends WorkerEntrypoint<Env> {
    categories: Categories;
    constructor(ctx: ExecutionContext, env: Env) {
        super(ctx, env);
        const customEnv: CustomEnv = {
            ...env,
            d1_db: env.DB,
            DB: new Kysely<DB>({
                dialect: new D1Dialect({ database: env.DB }),
                plugins: [new SerializePlugin()],
            }),
        };
        this.categories = new Categories(customEnv);
    }

    // required or wrangler crashes
    async fetch() {
        return new Response("ok");
    }
}

export class FDGLService extends WorkerEntrypoint<Env> {
    #categories: Categories;
    #communities: Communities;
    #reports: Reports;
    constructor(ctx: ExecutionContext, env: Env) {
        super(ctx, env);
        const customEnv: CustomEnv = {
            ...env,
            d1_db: env.DB,
            DB: new Kysely<DB>({
                dialect: new D1Dialect({ database: env.DB }),
                plugins: [new SerializePlugin()],
            }),
        };
        this.#categories = new Categories(customEnv);
        this.#communities = new Communities(customEnv);
        this.#reports = new Reports(customEnv);
    }

    get categories() {
        return {
            getCategory: this.#categories.getCategory.bind(this.#categories),
            getAllCategories: this.#categories.getAllCategories.bind(
                this.#categories,
            ),
            health: this.#categories.health.bind(this.#categories),
        };
    }
}

export class FDGLService extends WorkerEntrypoint<Env> {
    #categories: Categories;
    #communities: Communities;
    #reports: Reports;
    constructor(ctx: ExecutionContext, env: Env) {
        super(ctx, env);
        const customEnv: CustomEnv = {
            ...env,
            d1_db: env.DB,
            DB: new Kysely<DB>({
                dialect: new D1Dialect({ database: env.DB }),
                plugins: [new SerializePlugin()],
            }),
        };
        this.#categories = new Categories(customEnv);
        this.#communities = new Communities(customEnv);
        this.#reports = new Reports(customEnv);
    }

    get categories() {
        return {
            getCategory: this.#categories.getCategory.bind(this.#categories),
            getAllCategories: this.#categories.getAllCategories.bind(
                this.#categories,
            ),
            health: this.#categories.health.bind(this.#categories),
        };
    }
}

type PickMatching<T, V> = {
    [K in keyof T as T[K] extends V ? K : never]: T[K];
};
// biome-ignore lint/complexity/noBannedTypes: it's okay here as it is only a picker
type ExtractMethods<T> = PickMatching<T, Function>;

export class FDGLService extends WorkerEntrypoint<Env> {
    #reports: Reports;
    constructor(ctx: ExecutionContext, env: Env) {
        super(ctx, env);
        const customEnv: CustomEnv = {
            ...env,
            d1_db: env.DB,
            DB: new Kysely<DB>({
                dialect: new D1Dialect({ database: env.DB }),
                plugins: [new SerializePlugin()],
            }),
        };
        this.#reports = new Reports(customEnv);
    }

    get reports() {
        // biome-ignore format: its nicer in one line
        return {
            getReport: this.#reports.getReport.bind(this.#reports),
            getReports: this.#reports.getReports.bind(this.#reports),
            createReport: this.#reports.createReport.bind(this.#reports),
            revokeReport: this.#reports.revokeReport.bind(this.#reports)
        } satisfies ExtractMethods<Reports>
    }
}

type PickMatching<T, V> = {
    [K in keyof T as T[K] extends V ? K : never]: T[K];
};
// biome-ignore lint/complexity/noBannedTypes: it's okay here as it is only a picker
type ExtractMethods<T> = PickMatching<T, Function>;

export class FDGLService extends WorkerEntrypoint<Env> {
    #reports: Reports;
    constructor(ctx: ExecutionContext, env: Env) {
        super(ctx, env);
        const customEnv: CustomEnv = {
            ...env,
            d1_db: env.DB,
            DB: new Kysely<DB>({
                dialect: new D1Dialect({ database: env.DB }),
                plugins: [new SerializePlugin()],
            }),
        };
        this.#reports = new Reports(customEnv);
    }

    get reports() {
        // biome-ignore format: its nicer in one line
        return {
            getReport: this.#reports.getReport.bind(this.#reports),
            getReports: this.#reports.getReports.bind(this.#reports),
            createReport: this.#reports.createReport.bind(this.#reports),
            revokeReport: this.#reports.revokeReport.bind(this.#reports)
        } satisfies ExtractMethods<Reports>
    }
}