plugin.ts

import type { Plugin } from "vite";
import { generate } from "@graphql-codegen/cli";
import { execaCommand } from "execa";

// Only load up graphql, graphqls or gql files

const filterExt = /\.(graphqls?|gql)$/i;

// Delete all gq (generated) files
async function cleanGQ({ debug = false }) {
    debug && console.log("🧹 removing all .gq files");
    // Find and remove all .gq files
    await execaCommand("find ./ -path '*.gq.ts' -type f -prune -print -exec rm -f '{}' +; ", {
        stdio: debug ? "inherit" : "ignore",
        shell: true,
    });
}

export interface GQueryGenerateOptions {
    schema: string;
    output: string;
    gPath: string;
    debug?: boolean;
}

// Runs graphql codegen
async function gQueryGenerate({ schema, output, gPath, debug = false }: GQueryGenerateOptions) {
    debug && console.log("🤖 starting codegen");

    // the actual codegen process.
    await generate(
        {
            schema,
            documents: "./src/**/*.graphql",
            generates: {
                // * Generates the types for your schema
                [`${process.cwd()}/${output}/types.gq.ts`]: {
                    plugins: ["typescript"],
                },
                // * Generates near file .ts files for your fetch functions
                [`${process.cwd()}/${output}`]: {
                    config: {
                        useTypeImports: true,
                        gPath,
                        importDocumentNodeExternallyFrom: "near-operation-file",
                        inlineFragmentTypes: "combine",
                    },
                    preset: "near-operation-file",
                    presetConfig: {
                        extension: ".gq.ts",
                        folder: "./",
                        baseTypesPath: `types.gq.ts`,
                    },
                    plugins: [
                        "typescript-operations", // operations, gets you types for operations (queries and mutations)
                        "@leveluptuts/g-query/codegen-plugin", // g-query codegen plugin. ./codegen.ts
                    ],
                },
            },
        },
        true
    );
}

export default function levelupViteCodegen({ schema, output, gPath, debug = false }: GQueryGenerateOptions): Plugin {
    if (!schema) {
        throw new Error("No schema provided");
    }
    if (!output) {
        throw new Error("No output directory specified for types.");
    }
    if (!gPath) {
        throw new Error("No gPath directory specified. gPath is where you've initialized the 'g' client");
    }

    return {
        name: "g-query-codegen",
        async buildStart() {
            console.log("build start");
            try {
                await cleanGQ({ debug });
                await gQueryGenerate({ schema, output, gPath, debug });

                return;
            } catch (e) {
                //   TODO - These errors aren't good enough
                console.log("❓ gFetch Error - Something Happened - Here is the error and some things to consider.", e);
                console.log("❓ gFetch Error - Make sure `.graphql` are files found.");
                console.log(
                    "❓ gFetch Warning - If you would like the gQuery generator to work (we reccomend you do)."
                );
                console.log("❓ gFetch Warning - If you would like to add them, you will need to restart SvelteKit.");
            }
            return;
        },

        configureServer(server) {
            const listener = async (absolutePath = "") => {
                if (!filterExt.test(absolutePath)) return null;
                try {
                    await cleanGQ({ debug });
                    await gQueryGenerate({ schema, output, gPath });
                } catch (error) {
                    console.log("Something went wrong. Please save the file again.");
                }
            };

            server.watcher.on("add", listener);
            server.watcher.on("change", listener);
        },
    };
}