#!/usr/bin/env python
# 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 logging
import os
from pathlib import Path

from sign_workflow.signer import Signer

"""
This class is responsible for signing an artifact using the OpenSearch-signer-client and verifying its signature.
The signed artifacts will be found in the same location as the original artifacts.
"""


class SignerPGP(Signer):
    ACCEPTED_FILE_TYPES = [".zip", ".jar", ".war", ".pom", ".module", ".tar.gz", ".whl", ".crate", ".rpm", ".deb", ".tgz"]

    def generate_signature_and_verify(self, artifact: str, basepath: Path, signature_type: str) -> None:
        location = os.path.join(basepath, artifact)
        self.sign(artifact, basepath, signature_type)
        self.verify(location + signature_type)

    def is_valid_file_type(self, file_name: str) -> bool:
        return any(
            file_name.endswith(x) for x in SignerPGP.ACCEPTED_FILE_TYPES
        )

    def sign(self, artifact: str, basepath: Path, signature_type: str) -> None:
        filename = os.path.join(basepath, artifact)
        signature_file = filename + ".sig"
        if not self.overwrite:
            self.__remove_existing_signature__(signature_file)

        signing_cmd = [
            "./opensearch-signer-client",
            "-i",
            filename,
            "-o",
            signature_file,
            "-p",
            "pgp",
            "-r",
            str(self.overwrite)
        ]
        self.git_repo.execute(" ".join(signing_cmd))
        if (signature_type == ".asc"):
            self.__convert_to_asc(filename, signature_file)

    def __convert_to_asc(self, filename: str, signature_file: str) -> None:
        asc_signature_file = filename + ".asc"
        self.__remove_existing_signature__(asc_signature_file)
        conversion_cmd = [
            "gpg --enarmor",
            "<",
            signature_file,
            ">",
            asc_signature_file,
            "&&",
            "sed -i 's/ARMORED FILE/SIGNATURE/g'",
            asc_signature_file
        ]
        logging.info('Coverting .sig to .asc')
        self.git_repo.execute(" ".join(conversion_cmd))
        self.__remove_existing_signature__(signature_file)

    def verify(self, filename: str) -> None:
        verify_cmd = ["gpg", "--verify-files", filename]
        self.git_repo.execute(" ".join(verify_cmd))