#! /usr/bin/env python3

import os
import requests

api_key = os.environ["TS_API_KEY"]

endpoint = "https://api.targetsmart.com/service/onboarding"
headers = {"x-api-key": api_key}

# request URL to upload data
params = {
    "platform": "facebook",
    "audience": "my_audience_name",
    "status_email": "me@example.com",
    "facebook_adid": CLIENT_AD_ACCOUNT_ID,
    # Only include `owner_adid` if uploading your PII directly to your Facebook
    # ad account.
    # "owner_adid": PARTNER_AD_ACCOUNT_ID,
}

# Register with service/onboarding to produce a target url for upload.
response_1 = requests.get(endpoint, params=params, headers=headers)
response_1.raise_for_status()
response_1_info = response_1.json()

if not response_1_info["error"]:

    print("Registration request successful")

    # Stream your prepared UTF8 CSV input file to the provided pre-signed URL.
    # Content-Type should be <empty>.
    # This request interacts with AWS S3. Do not include the TargetSmart API key in the request header.
    with open("onboarding_input_file.csv", "rb") as reader:
        response_2 = requests.put(
            response_1_info["url"], data=reader, headers={"Content-Type": ""}
        )

    response_2.raise_for_status()
    print("Execution started")

#! /usr/bin/env node

const fetch = require("node-fetch");
const fs = require("fs");
const filepath = "onboarding_input_file.csv";
const targetSmartApiKey = process.env.TS_API_KEY;

// Register with service/onboarding to produce a target url for upload.
fetch(
  "https://api.targetsmart.com/service/onboarding?" +
    new URLSearchParams({
      platform: "facebook",
      audience: "my_audience_name",
      status_email: "me@example.com",
      facebook_adid: CLIENT_AD_ACCOUNT_ID,
      // Only include `owner_adid` if uploading your PII directly to your Facebook
      // ad account.
      // "owner_adid": PARTNER_AD_ACCOUNT_ID,
    }).toString(),
  {
    method: "GET",
    headers: {
      "x-api-key": targetSmartApiKey,
    },
  }
).then((response) => {
  if (response.status === 200) {
    response.json().then((respdata) => {
      if (!respdata.error) {
        console.log("Registration request successful");

        /* Stream your prepared UTF8 CSV input file to the provided pre-signed
         * URL. Content-Type should be <empty>. This request interacts with AWS
         * S3. Do not include the TargetSmart API key in the request header. */
        const stats = fs.statSync(filepath);
        const reader = fs.createReadStream(filepath, { encoding: null });
        reader.on("error", console.log);

        fetch(respdata.url, {
          method: "PUT",
          headers: { "Content-Length": stats.size, "Content-Type": "" },
          body: fs.createReadStream(filepath),
        }).then((response2) => {
          if (response2.status === 200) {
            console.log("Execution started");
          }
        });
      }
    });
  }
});

#! /usr/bin/env bash

# Register with service/onboarding to produce a target url for upload.
if result_1=$(curl -XGET \
    -H "x-api-key: $TS_API_KEY" \ "https://api.targetsmart.com/service/onboarding?platform=facebook&audience=my_audience_name&status_email=me%40example.com%0A&facebook_adid=CLIENT_AD_ACCOUNT_ID"); then

    # Stream your prepared UTF8 CSV input file to the provided pre-signed URL.
    # Content-Type should be <empty>.
    # This request interacts with AWS S3. Do not include the TargetSmart API key in the request header.
    echo "Registration request successful";
    aws_url=$(echo $result_1| jq -r '.url');
    if result_2=$(curl -L -H 'Content-Type:' --upload-file onboarding_input_file.csv $aws_url); then
        echo "Execution started"
    fi
fi