feat: manually delete cache

Author: JacobLinCool

src/cloudflare-worker/cache.ts

@@ -3,21 +3,24 @@ import { Cache as Base } from "../core/types";
 export class Cache implements Base {
     private cache;
 
-    constructor() {
+    constructor(private expiration = 60) {
         this.cache = caches.open("leetcode");
     }
 
     async put(key: string, value: any, options?: { expire?: number }) {
         key = this.urlify(key);
 
         if (value instanceof Response) {
-            value.headers.set("Cache-Control", `public, max-age=${options?.expire ?? 60}`);
+            value.headers.set(
+                "Cache-Control",
+                `public, max-age=${options?.expire ?? this.expiration}`,
+            );
             await (await this.cache).put(key, value);
         } else {
             const dummy: Response = new Response(JSON.stringify(value), {
                 headers: {
                     "Content-Type": "application/x-custom-data",
-                    "Cache-Control": `public, max-age=${options?.expire ?? 60}`,
+                    "Cache-Control": `public, max-age=${options?.expire ?? this.expiration}`,
                 },
             });
 
@@ -41,6 +44,12 @@ export class Cache implements Base {
         return null;
     }
 
+    async delete(key: string) {
+        key = this.urlify(key);
+
+        return (await this.cache).delete(key);
+    }
+
     private urlify(key: string) {
         try {
             return new URL(key).href;

src/cloudflare-worker/handler.ts

@@ -33,6 +33,7 @@ function sanitize(config: Record<string, string>): Config {
         font: "baloo_2",
         animation: true,
         theme: { light: "light", dark: "dark" },
+        cache: 60,
     };
 
     if (!config.username?.trim()) {
@@ -110,13 +111,14 @@ function sanitize(config: Record<string, string>): Config {
         sanitized.extensions.push(RemoteStyleExtension);
     }
 
+    if (config.cache && parseInt(config.cache) >= 0 && parseInt(config.cache) <= 60 * 60 * 24 * 7) {
+        sanitized.cache = parseInt(config.cache);
+    }
+
     return sanitized;
 }
 
 async function generate(config: Record<string, string>): Promise<Response> {
-    const generator = new Generator(new Cache());
-    generator.verbose = true;
-
     let sanitized: Config;
     try {
         sanitized = sanitize(config);
@@ -131,6 +133,9 @@ async function generate(config: Record<string, string>): Promise<Response> {
     const cache_header =
         `max-age=${cache_time}` + (cache_time <= 0 ? ", no-store, no-cache" : ", public");
 
+    const generator = new Generator(new Cache(sanitized.cache));
+    generator.verbose = true;
+
     const headers = new Header().add("cors", "svg");
     headers.set("cache-control", cache_header);
 
@@ -157,6 +162,18 @@ router.get("*", async ({ query }: { query: Record<string, string> }) => {
     return await generate(query);
 });
 
+router.delete("/:site/:username", async ({ params }) => {
+    if (params?.site && params?.username) {
+        const site = params.site.toLowerCase();
+        const username = params.username.toLowerCase();
+        const cache = new Cache(60);
+        return new Response(
+            JSON.stringify({ success: await cache.delete(`data-${username}-${site}`) }, null, 4),
+            { headers: new Header().add("cors", "json") },
+        );
+    }
+});
+
 // 404 for all other routes
 router.all("*", () => new Response("Not Found.", { status: 404 }));