import java.io.*;
import java.net.*;
import org.json.*;

public class VizardSample {
    private static final String VIZARDAI_API_KEY = "YOUR_API_KEY";

    public static void main(String[] args) {
        String projectId = createProject();
        queryClips(projectId);
    }

    private static String createProject() {
        try {
            URL url = new URL("https://elb-api.vizard.ai/hvizard-server-front/open-api/v1/project/create");
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            conn.setRequestMethod("POST");
            conn.setRequestProperty("Content-Type", "application/json");
            conn.setRequestProperty("VIZARDAI_API_KEY", VIZARDAI_API_KEY);
            conn.setDoOutput(true);

            // JSONObject jsonObject = buildRequestParametersForRemoteVideoFile();
            JSONObject jsonObject = buildRequestParametersForYouTubeLink();

            try (OutputStream os = conn.getOutputStream()) {
                os.write(jsonObject.toString().getBytes("utf-8"));
            }

            int status = conn.getResponseCode();
            StringBuilder response = new StringBuilder();
            try (BufferedReader br = new BufferedReader(new InputStreamReader(conn.getInputStream(), "utf-8"))) {
                String responseLine;
                while ((responseLine = br.readLine()) != null) {
                    response.append(responseLine.trim());
                }
            }

            JSONObject responseJson = new JSONObject(response.toString());
            if (status == HttpURLConnection.HTTP_OK && responseJson.getInt("code") == 2000) {
                String projectId = String.valueOf(responseJson.getInt("projectId"));
                System.out.println("Project created, and projectId is " + projectId);
                return projectId;
            } else {
                System.out.println("Create error: " + responseJson.toString());
                return "";
            }
        } catch (Exception e) {
            System.out.println("Create request failed: " + e.getMessage());
            return "";
        }
    }

    private static void queryClips(String projectId) {
        if (projectId == null || projectId.isEmpty()) {
            System.out.println("projectId is empty");
            return;
        }
        while (true) {
            try {
                System.out.println("Querying clips...");
                URL url = new URL("https://elb-api.vizard.ai/hvizard-server-front/open-api/v1/project/query/" + projectId);
                HttpURLConnection conn = (HttpURLConnection) url.openConnection();
                conn.setRequestMethod("GET");
                conn.setRequestProperty("Content-Type", "application/json");
                conn.setRequestProperty("VIZARDAI_API_KEY", VIZARDAI_API_KEY);

                StringBuilder response = new StringBuilder();
                try (BufferedReader br = new BufferedReader(new InputStreamReader(conn.getInputStream(), "utf-8"))) {
                    String responseLine;
                    while ((responseLine = br.readLine()) != null) {
                        response.append(responseLine.trim());
                    }
                }

                JSONObject responseJson = new JSONObject(response.toString());
                if (conn.getResponseCode() == HttpURLConnection.HTTP_OK) {
                    if (responseJson.getInt("code") == 2000) {
                        System.out.println("Clipping succeeded.");
                        System.out.println(responseJson.toString());
                        break;
                    } else if (responseJson.getInt("code") == 1000) {
                        Thread.sleep(5000);
                        continue;
                    } else {
                        System.out.println("Clipping error:");
                        System.out.println(responseJson.toString());
                        break;
                    }
                } else {
                    System.out.println("Query request failed: " + response.toString());
                    break;
                }
            } catch (Exception e) {
                System.out.println("Query request failed: " + e.getMessage());
                break;
            }
        }
    }

    private static JSONObject buildRequestParametersForRemoteVideoFile() {
        JSONObject jsonObject = new JSONObject();
        jsonObject.put("videoType", 1);
        jsonObject.put("projectName", "api-sample-remote-video-file");
        jsonObject.put("videoUrl", "https://cdn-docs.vizard.ai/openapi/sample.mp4");
        jsonObject.put("ext", "mp4");
        jsonObject.put("lang", "en");
        jsonObject.put("preferLength", "[0]");
        jsonObject.put("subtitleSwitch", 1);
        jsonObject.put("headlineSwitch", 1);
        return jsonObject;
    }

    private static JSONObject  buildRequestParametersForYouTubeLink() {
        JSONObject jsonObject = new JSONObject();
        jsonObject.put("videoType", 2);
        jsonObject.put("projectName", "api-sample-youtube");
        jsonObject.put("videoUrl", "https://www.youtube.com/watch?v=qeu9ek_HygM");
        jsonObject.put("lang", "en");
        jsonObject.put("preferLength", "[0]");
        jsonObject.put("subtitleSwitch", 1);
        jsonObject.put("headlineSwitch", 1);
        return jsonObject;
    }

}