FastEdge

use std::time::Duration;
use fastedge::{body::Body, http::{header, Error, Method, Request, Response, StatusCode}};
use rusty_s3::{Bucket, Credentials, S3Action, UrlStyle};
use std::{collections::HashMap, env};
use url::Url;

#[fastedge::http]
fn main(req: Request<Body>) -> Result<Response<Body>, Error> {
    // ... allow POST/PUT, read ?name= and Content-Type
    let content = req.into_body();
    // ... reject files over MAX_FILE_SIZE
    let (signed_url, host) = match prepare_s3(fname) { /* ... */ };
    // ... build a PUT request to the signed URL
    let rsp = match fastedge::send_request(req) { /* ... */ };
    // ... on 200, return the file URL instead of the S3 reply
    Ok(Response::from_parts(parts, body))
}

fn prepare_s3(fname: &str) -> anyhow::Result<(Url, String)> {
    /* read S3 access params from env */
    let access_key = env::var("ACCESS_KEY")?;
    let secret_key = env::var("SECRET_KEY")?;
    // ... read REGION, BASE_HOSTNAME, BUCKET and build the bucket URL
    let bucket = Bucket::new(parsed_url, UrlStyle::Path, bucket, region)?;

    let creds = Credentials::new(access_key, secret_key);
    let action = bucket.put_object(Some(&creds), fname);
    let signed_url = action.sign(Duration::from_secs(60 * 60));

    Ok((signed_url, host))
}

use fastedge::wasi_nn::wasi::nn::{graph, tensor};
// ... other imports and ImageNet labels

#[fastedge::http]
fn main(req: Request<Body>) -> Result<Response<Body>, Error> {
    // ... accept POST/PUT up to 2 MB, read ?model= (default mobilenet-v2)
    let output = inference(&model_name, req.body())?;
    // ... match scores to labels, keep those above 1%, return JSON
}

fn inference(model_name: &str, image: &[u8]) -> Result<TensorData> {
    // the model is already loaded in the FastEdge runtime
    let context = graph::load_by_name(model_name)?.init_execution_context()?;
    // ... resize the image to 224×224 and turn it into a tensor
    let output = context.compute(vec![("0".into(), tensor)])?;
    Ok(output[0].1.data())
}

import { jsPDF } from "jspdf";
import qs from "qs";

function createInvoice(company = "NA", customer = "NA", /* ... */) {
  const doc = new jsPDF({
    orientation: "p",
    format: "a4",
  });
  // ... header, customer block, line items, total
  return doc.output("arraybuffer");
}

async function eventHandler(event) {
  const queryString = event.request.url.split("?")[1] ?? "";
  const params = qs.parse(queryString);
  const pdfDocument = createInvoice(
    params.company,
    params.customer,
    // ... invoice-id, date, product fields
  );

  return new Response(pdfDocument, { status: 202 });
}

addEventListener("fetch", (event) => {
  event.respondWith(eventHandler(event));
});