Connect Zyte automatic extraction and Dremio in our serverless environment

class HttpSourceZyteAutomaticExtraction {
    async init() {
        // TODO: Create your http credential with zyte information:
        // More info at https://yepcode.io/docs/integrations/http/#credential-configuration
        // Get your Zyte API key following guide at: https://docs.zyte.com/automatic-extraction-get-started.html
        // baseUrl: https://autoextract.scrapinghub.com/v1
        // user: <your_automatic_extraction_api_key>
        // password: empty
        this.httpClient = yepcode.integration.http(
            "your-http-zyte-credential-name"
        );
    }

    async fetch(publish, done) {
        // TODO: Customize your request checking the API documentation:
        // https://docs.zyte.com/automatic-extraction.html
        const payload = [
            // In this sample we'll use the product list extraction: https://docs.zyte.com/automatic-extraction/product-list.html
            // TODO: Other lists or single object extractions may be done changing this payload:
            //       articles, comments, job posting, real state, reviews
            {
                url: "http://books.toscrape.com/",
                pageType: "productList",
            },
        ];
        const {
            data
        } = await this.httpClient.post("/extract", payload);

        if (data && data[0]) {
            // TODO: If retrieved list is not a productList, this attribute navigations should be changed
            // TODO: You may also keep on retrieving next paginations using information at data[0].productList.paginationNext
            for (const item of data[0].productList.products) {
                await publish(item);
            }
        }
        done();
    }

    async close() {}
}

class HttpTargetDremioInsert {
    async init() {
        // TODO: Create your http credential with dremio information:
        // More info at https://yepcode.io/docs/integrations/http/#credential-configuration
        // Official docs: https://docs.dremio.com/current/reference/api/
        this.httpClient = yepcode.integration.http("your-dremio-credential-name");
    }
    async consume(item) {
        const projectId = "<your-dremio-project-id>";
        // TODO: Map your item to row and customize INSERT statement
        // More info at https://docs.dremio.com/cloud/reference/sql/commands/insert/
        const query = `INSERT INTO your_source_path.your_table VALUES (${item.id}, ${item.name})`;

        // Use the SQL API to submit queries. The response contains the ID for the job associated with the SQL query
        // Official docs: https://docs.dremio.com/cloud/reference/api/sql
        const {
            data: job
        } = await this.httpClient.post(`${projectId}/sql`, {
            sql: query,
        });

        // Use the Job API to check the job status. Jobs final status are COMPLETED, FAILED and CANCELED
        // Official docs: https://docs.dremio.com/cloud/reference/api/job/
        const finalJobStates = ["COMPLETED", "FAILED", "CANCELED"];
        let jobInfoResponse = null;
        do {
            jobInfoResponse = await this.httpClient.get(
                `${this.projectId}/job/${job.id}`
            );
        } while (!finalJobStates.includes(jobInfoResponse.data.jobState));

        const {
            data: jobInfo
        } = jobInfoResponse;

        if (jobInfo.jobState === "COMPLETED") {
            console.log("Success!!");
        } else if (jobInfo.jobState === "FAILED") {
            console.log(`Uups! Something was wrong: ${jobInfo.errorMessage}`);
        } else {
            console.log(`The job's been cancelled: ${jobInfo.cancellationReason}`);
        }
    }

    async close() {}
}