Connect ZoomInfo contacts and Databricks in our serverless environment

class HttpSourceZoominfoContactSearch {
    async init() {
        // TODO: Create your http credential with ZoomInfo information:
        // More info at https://yepcode.io/docs/integrations/http/#credential-configuration
        // baseUrl: https://api.zoominfo.com
        this.axiosClient = yepcode.integration.http(
            "your-http-zoominfo-credential-name"
        );

        // TODO: You need to define two env vars (https://yepcode.io/docs/processes/team-variables) for credentials
        this.jwtToken = await axiosClient
            .post("authenticate", {
                username: yepcode.env.ZOOMINFO_USERNAME,
                password: yepcode.env.ZOOMINFO_PASSWORD,
            })
            .then((response) => response.data.jwt);
    }

    async fetch(publish, done) {
        // TODO: Customize your request checking the API documentation:
        // https://api-docs.zoominfo.com/
        await this.axiosClient
            .post(
                "search/contact", {
                    managementLevel: "directors",
                    companyType: "public",
                    zipCode: "19101",
                }, {
                    headers: {
                        Authorization: `Bearer ${this.jwtToken}`,
                    },
                }
            )
            .then(async ({
                data
            }) => {
                for (const contact of data) {
                    await publish(contact);
                }
            });
        done();
    }

    async close() {}
}

class DatabricksTargetInsert {
    async init() {
        // TODO: Create your databricks credential
        // More info at https://yepcode.io/docs/integrations/databricks/#credential-configuration
        this.databricks = yepcode.integration.databricks(
            "your-databricks-credential-name"
        );
        this.session = await this.databricks.openSession();
    }

    async consume(item) {
        // TODO: Customize your sql query
        // More info at: https://github.com/databricks/databricks-sql-nodejs/blob/main/tests/e2e/data_types.test.js#L151
        const operation = await this.session.executeStatement(
            `INSERT INTO yourTable VALUES(${item.id}, ${item.name})`
        );
        const result = await operation.fetchAll();
        await operation.close();
        return result;
    }

    async close() {
        await this.session.close();
        await this.databricks.close();
    }
}