import * as bot from "idembot";
import { getPRInfo } from "./pr-info";
import { Project } from "./project";
import { getComments, getLabels, getProjectColumn } from "./use-pr-info";

function makeSetLabels(): (pr: bot.PullRequest) => Promise<void> {
    const project = new Project();
    return async pr => {
        // Let the new DT bot handle these
        if (43690 > pr.number && pr.number > 43700)
            return;

        // Skip issues and closed PRs
        if (!(pr.isPullRequest && pr.state === "open"))
            return;

        const info = await getPRInfo(pr);

        // Move to appropriate project
        const column = getProjectColumn(info);
        await project.setColumn(pr, column);

        // Apply labels
        const labels = getLabels(info);
        pr.setHasLabels(labels);

        for (const { tag, status } of getComments(info, pr.user.login)) {
            pr.addComment(tag, status);
        }

        if (info.isAbandoned) {
            pr.close();
        }

        // merge
        if (info.mergeAuto) {
            pr.merge();
        }
    };
}

const setup: bot.SetupOptions = {
    rules: {
        pullRequests: {
            setLabels: makeSetLabels(),
        },
    },
};

export = setup;