/*
 * Copyright OpenSearch Contributors
 * SPDX-License-Identifier: Apache-2.0
 *
 * The OpenSearch Contributors require contributions made to
 * this file be licensed under the Apache-2.0 license or a
 * compatible open source license.
 */
import jenkins.BundleManifest
import org.yaml.snakeyaml.Yaml

import static org.hamcrest.CoreMatchers.notNullValue
import static org.hamcrest.MatcherAssert.assertThat


class CreateReleaseTagLibTester extends LibFunctionTester {

    private String distManifest
    private String tagVersion
    private ArrayList bundleManifestComponentsList

    public CreateReleaseTagLibTester(distManifest, tagVersion){
        this.distManifest = distManifest
        this.tagVersion = tagVersion
        this.bundleManifestComponentsList = []
    }

    void parameterInvariantsAssertions(call){
        assertThat(call.args.distManifest.first(), notNullValue())
        assertThat(call.args.tagVersion.first(), notNullValue())
    }

    boolean expectedParametersMatcher(call) {
        return call.args.distManifest.first().toString().equals(this.distManifest)
                && call.args.tagVersion.first().toString().equals(this.tagVersion)
                && this.bundleManifestComponentsList.size() > 1
    }

    String libFunctionName(){
        return 'createReleaseTag'
    }

    void configure(helper, binding){
        binding.setVariable('GITHUB_BOT_TOKEN_NAME', 'dummy_token_name')
        binding.setVariable('GITHUB_USER', 'dummy_user')
        binding.setVariable('GITHUB_TOKEN', 'dummy_token')

        helper.registerAllowedMethod("checkout", [Map], {})
        helper.registerAllowedMethod("dir", [Map], {})
        InputStream inputStream = new FileInputStream(new File(this.distManifest));
        Yaml yaml = new Yaml()
        Map ymlMap = yaml.load(inputStream)
        BundleManifest bundleManifestObj = new BundleManifest(ymlMap)
        this.bundleManifestComponentsList = bundleManifestObj.getNames()
        boolean checkFirst = true
        for (component in this.bundleManifestComponentsList) {
            def repo = bundleManifestObj.getRepo(component)
            def version = tagVersion
            if (tagVersion.contains("-")) {
                version = tagVersion.split("-").first() + ".0-" + tagVersion.split("-").last()
            } else {
                version = "$tagVersion.0"
            }
            if (component == "OpenSearch" || component == "OpenSearch-Dashboards" || component == "functionalTestDashboards") {
                version = tagVersion
            }
            def out = ""
            if (checkFirst) {
                out = bundleManifestObj.getCommitId(component)
                checkFirst = false
            }
            helper.addShMock("git ls-remote --tags $repo $version | awk 'NR==1{print \$1}'") { script ->
                return [stdout: out, exitValue: 0]
            }
        }
    }
}