Uploading a file

import math
import os
import time

import requests

API = "https://api.aigenda.tech"


def upload_file(token: str, space_id: int, path: str, content_type: str) -> int:
    """Uploads a file to a space and returns audio_id once AiGenda has confirmed the file."""
    headers = {"Authorization": f"Bearer {token}"}
    size = os.path.getsize(path)
    title = os.path.basename(path)  # name with extension, e.g. "Standup.mp3"

    r = requests.post(
        f"{API}/api/spaces/{space_id}/audio/",
        headers=headers,
        json={"title": title, "size": size, "content_type": content_type},
        timeout=30,
    )
    r.raise_for_status()
    audio = r.json()
    urls = audio["url"]

    part_size = math.ceil(size / len(urls))
    parts = []
    with open(path, "rb") as f:
        for number, url in enumerate(urls, start=1):
            chunk = f.read(part_size)
            put = requests.put(url, data=chunk, timeout=900)
            put.raise_for_status()
            parts.append({"PartNumber": number, "ETag": put.headers["ETag"]})

    r = requests.post(
        f"{API}/api/spaces/{space_id}/audio/{audio['id']}/complete_upload/",
        headers=headers,
        json={"parts": parts},
        timeout=30,
    )
    r.raise_for_status()

    for _ in range(60):
        info = requests.get(
            f"{API}/api/spaces/{space_id}/audio/{audio['id']}/info/",
            headers=headers,
            timeout=30,
        ).json()
        if info.get("is_uploaded"):
            return audio["id"]
        time.sleep(2)
    raise TimeoutError("AiGenda did not confirm the file within 2 minutes")

import { open, stat } from "node:fs/promises";
import { basename } from "node:path";

const API = "https://api.aigenda.tech";

// Uploads a file to a space and returns audioId once AiGenda has confirmed the file.
export async function uploadFile(token, spaceId, path, contentType) {
  const auth = { Authorization: `Bearer ${token}` };
  const json = { ...auth, "Content-Type": "application/json" };
  const { size } = await stat(path);

  const created = await fetch(`${API}/api/spaces/${spaceId}/audio/`, {
    method: "POST",
    headers: json,
    body: JSON.stringify({ title: basename(path), size, content_type: contentType }),
  });
  if (!created.ok) throw new Error(`create: ${created.status}`);
  const audio = await created.json();

  const partSize = Math.ceil(size / audio.url.length);
  const parts = [];
  const file = await open(path);
  try {
    for (let i = 0; i < audio.url.length; i++) {
      const length = Math.min(partSize, size - i * partSize);
      const chunk = Buffer.alloc(length);
      await file.read(chunk, 0, length, i * partSize);
      const put = await fetch(audio.url[i], { method: "PUT", body: chunk });
      if (!put.ok) throw new Error(`part ${i + 1}: ${put.status}`);
      parts.push({ PartNumber: i + 1, ETag: put.headers.get("ETag") });
    }
  } finally {
    await file.close();
  }

  const done = await fetch(`${API}/api/spaces/${spaceId}/audio/${audio.id}/complete_upload/`, {
    method: "POST",
    headers: json,
    body: JSON.stringify({ parts }),
  });
  if (!done.ok) throw new Error(`complete: ${done.status}`);

  for (let attempt = 0; attempt < 60; attempt++) {
    const res = await fetch(`${API}/api/spaces/${spaceId}/audio/${audio.id}/info/`, { headers: auth });
    const info = await res.json();
    if (info.is_uploaded) return audio.id;
    await new Promise((resolve) => setTimeout(resolve, 2000));
  }
  throw new Error("AiGenda did not confirm the file within 2 minutes");
}