import { v2beta3 } from "@google-cloud/tasks"

const { GCP_PROJECT } = process.env

if (!GCP_PROJECT) {
 throw new Error("GCP_PROJECT is required")
}

const cloudServiceAccountEmail = process.env.CLOUD_TASK_SERVICE_ACCT_EMAIL;

class CloudTasksClientWrapper {
 private client: v2beta3.CloudTasksClient

 constructor() {
   this.client = new v2beta3.CloudTasksClient()
 }

 async createHttpTaskWithToken({
   queueName,
   payload,
   url,
   location = "us-central1",
 }:
 {
   queueName: string
   payload: string
   url: string
   location?: string
 }) {
   const parent = this.client.queuePath(GCP_PROJECT!, location, queueName)

   const body = Buffer.from(payload).toString("base64")

   const task = {
     httpRequest: {
       httpMethod: "POST" as const,
       url,
       oidcToken: {
         serviceAccountEmail: cloudServiceAccountEmail,
         audience: new URL(url).origin,
       },
       headers: {
         "Content-Type": "application/json",
       },
       body,
     },
   }

   // Send create task request.
   const [response] = await this.client.createTask(
     { parent, task },
     { timeout: 5000 }
   )
   return response
 }
}

export { CloudTasksClientWrapper }