import {data, DescriptionDictionary} from "@github/actions-expressions";
import {isScalar, isString} from "@github/actions-workflow-parser";
import {WorkflowJob} from "@github/actions-workflow-parser/model/workflow-template";
import {WorkflowContext} from "../context/workflow-context";

export function getNeedsContext(workflowContext: WorkflowContext): DescriptionDictionary {
  const d = new DescriptionDictionary();

  const job = workflowContext.job || workflowContext.reusableWorkflowJob;

  if (!job?.needs) {
    return d;
  }

  for (const jobID of job.needs) {
    const job = workflowContext.template?.jobs.find(job => job.id.value === jobID.value);
    d.add(jobID.value, needsJobContext(job));
  }

  return d;
}

function needsJobContext(job?: WorkflowJob): DescriptionDictionary {
  // https://docs.github.com/en/actions/learn-github-actions/contexts#needs-context
  const d = new DescriptionDictionary();

  if (job) {
    d.add("outputs", jobOutputs(job));
  }

  // Can be "success", "failure", "cancelled", or "skipped"
  d.add("result", new data.Null());
  return d;
}

function jobOutputs(job?: WorkflowJob): DescriptionDictionary {
  const d = new DescriptionDictionary();
  if (!job?.outputs) {
    return d;
  }

  for (const output of job.outputs) {
    if (!isString(output.key)) {
      continue;
    }

    // Include the value for hover purposes
    const value = isScalar(output.value) ? new data.StringData(output.value.toDisplayString()) : new data.Null();
    d.add(output.key.value, value);
  }
  return d;
}