CloudFlare-ImgBed/functions/random.js

export async function onRequest(context) {
    // Contents of context object
    const {
      request, // same as existing Worker API
      env, // same as existing Worker API
      params, // if filename includes [id] or [[path]]
      waitUntil, // same as ctx.waitUntil in existing Worker API
      next, // used for middleware or to fetch assets
      data, // arbitrary space for passing data between middlewares
    } = context;
    const requestUrl = new URL(request.url);
    const protocol = requestUrl.protocol;
    const domain = requestUrl.hostname;
    const port = requestUrl.port;
    let allRecords = [];
    let cursor = null;
    do {
        const records = await env.img_url.list({
            limit: 1000,
            cursor,
        });
        allRecords.push(...records.keys);
        cursor = records.cursor;
    } while (cursor);
    if (env.AllowRandom != "true") {
        return new Response(JSON.stringify({ error: "Random is disabled" }), { status: 403 });
    }
    if (allRecords.length == 0) {
        return new Response(JSON.stringify({}), { status: 200 });
    } else {
        const randomIndex = Math.floor(Math.random() * allRecords.length);
        const randomKey = allRecords[randomIndex];
        const randomPath = '/file/' + randomKey.name;
        let randomUrl = randomPath;
        const randomType = requestUrl.searchParams.get('type');
        const resType = requestUrl.searchParams.get('form');
        // if param 'type' is set to 'url', return the full URL
        if (randomType == 'url') {
            if (port) {
                randomUrl = protocol + '//' + domain + ':' + port + randomPath;
            } else {
                randomUrl = protocol + '//' + domain + randomPath;
            }
        }
        // if param 'type' is set to 'img', return the image
        if (randomType == 'img') {
            // Return an image response
            randomUrl = protocol + '//' + domain + ':' + port + randomPath;
            let contentType = 'image/jpeg';
            return new Response(await fetch(randomUrl).then(res => {
                contentType = res.headers.get('content-type');
                return res.blob();
            }), {
                headers: contentType ? { 'Content-Type': contentType } : { 'Content-Type': 'image/jpeg' },
                status: 200
            });
        }
        
        if (resType == 'text') {
            return new Response(randomUrl, { status: 200 });
        } else {
            return new Response(JSON.stringify({ url: randomUrl }), { status: 200 });
        }
    }
}