package com.databuild.examples.basic_graph;

import java.util.ArrayList;
import java.util.Arrays;

/**
 * Configure class for generating a random number.
 * This class creates a job configuration for generating a random number based on the partition ref.
 */
public class SumConfigure {
    public static void main(String[] args) {
        if (args.length != 1) {
            System.err.println("Error: Must provide exactly one partition ref as an argument");
            System.exit(1);
        }

        String partitionRef = args[0];
        String[] upstreams = partitionRef.split("_");

        // Create a list of quoted upstream values
        ArrayList<String> quotedUpstreams = new ArrayList<>();
        Arrays.stream(upstreams).forEach(s -> quotedUpstreams.add("\"" + s + "\""));

        // Create a job config for generating a random number
        String config = String.format(
                "{\"outputs\":[\"%s\"],\"inputs\":[],\"args\":[%s],\"env\":{}}",
                partitionRef, String.join(",", quotedUpstreams)
        );
        System.out.println(config);
    }
}