# Copyright (c) 2018 - Red Hat Inc.

#

# This program is free software; you can redistribute it and/or modify it

# under the terms of the GNU General Public License as published by the

# Free Software Foundation; either version 2 of the License, or (at your

# option) any later version.  See http://www.gnu.org/copyleft/gpl.html for

# the full text of the license.

"""Interact with the Red Hat lookaside cache

We need to override the pyrpkg.lookasidecache module to handle our custom

download path.

"""

import io

import os

import pycurl

import six

import sys

from pyrpkg.errors import InvalidHashType, UploadError, LayoutError

from pyrpkg.lookaside import CGILookasideCache

from pyrpkg.layout.layouts import DistGitLayout

from . import utils

def is_dist_git(folder):

    """

    Indicates if a folder is using a dist-git layout.

    Parameters

    ----------

    folder: str

        The directory to inspect.

    Returns

    -------

    bool

        A bool flag indicating if `folder` is using

        a dist-git layout format.

    """

    result = False

    try:

        DistGitLayout.from_path(folder)

        result = True

    except LayoutError:

        result = False

    finally:

        return result

class StreamLookasideCache(CGILookasideCache):

    """

    CentosStream lookaside specialized class.

    It inherits most of its behavior from `pyrpkg.lookasideCGILookasideCache`.

    """

    def __init__(self, hashtype, download_url, upload_url, client_cert=None, ca_cert=None):

        super(StreamLookasideCache, self).__init__(

            hashtype, download_url, upload_url,

            client_cert=client_cert, ca_cert=ca_cert)

    def remote_file_exists(self, name, filename, hashstr):

        """

        Check if a remote file exists.

        This method inherits the behavior of its parent class from pyrpkg.

        It uses the internal `utils.get_repo_name` method to parse the name in case

        it is a scm url. 

        Parameters

        ----------

        name: str

            The repository name and org.

        filename: str

            The filename (something.tar.gz).

        hash:

            The hash string for the file.

        Returns

        -------

        bool

            A boolean value to inditicate if the file exists.

        """

        _name = utils.get_repo_name(name) if is_dist_git(os.getcwd()) else name

        return super(StreamLookasideCache, self).remote_file_exists(

                _name, filename, hashstr)

    def upload(self, name, filename, hashstr, offline=False):

        """

        Uploads a file to lookaside cache.

        This method inherits the behavior of its parent class from pyrpkg.

        It uses the internal `utils.get_repo_name` method to parse the name in case

        it is a scm url. 

        Parameters

        ----------

        name: str

            The repository name and org.

        filename: str

            The filename (something.tar.gz).

        hash:

            The hash string for the file.

        Raises

        ------

        pyrpkg.errors.rpkgError

            Raises specialized classes that inherits from pyrpkg base errors.

        Returns

        -------

        None

            Does not return anything

        """

        _name = utils.get_repo_name(name) if is_dist_git(os.getcwd()) else name

        return super(StreamLookasideCache, self).upload(

            _name, filename, hashstr)

    def download(self, name, filename, hashstr, outfile, hashtype=None, **kwargs):

        """

        Downloads a file from lookaside cache to the local filesystem.

        This method inherits the behavior of its parent class from pyrpkg.

        It uses the internal `utils.get_repo_name` method to parse the name in case

        it is a scm url. 

        Parameters

        ----------

        name: str

            The repository name and org.

        filename: str

            The filename (something.tar.gz).

        hash: str

            The hash string for the file.

        outfile: str

        Raises

        ------

        pyrpkg.errors.rpkgError

            Raises specialized implementations of  `yrpkg.errors.rpkgError`.

        Returns

        -------

        None

            Does not return anything

        """

        _name = utils.get_repo_name(name) if is_dist_git(os.getcwd()) else name

        return super(StreamLookasideCache, self).download(

           _name, filename, hashstr, outfile, hashtype=hashtype, **kwargs)

class SIGLookasideCache(CGILookasideCache):

    def __init__(self, hashtype, download_url, upload_url, name, branch):

        super(SIGLookasideCache, self).__init__(

            hashtype, download_url, upload_url, client_cert="/home/bstinson/.centos.cert")

        self.branch = branch

        self.download_path = (

            '%(name)s/%(branch)s/%(hash)s')

    def remote_file_exists(self, name, filename, hash):

        """Verify whether a file exists on the lookaside cache

        :param str name: The name of the module. (usually the name of the

            SRPM). This can include the namespace as well (depending on what

            the server side expects).

        :param str filename: The name of the file to check for.

        :param str hash: The known good hash of the file.

        """

        # RHEL 7 ships pycurl that does not accept unicode. When given unicode

        # type it would explode with "unsupported second type in tuple". Let's

        # convert to str just to be sure.

        # https://bugzilla.redhat.com/show_bug.cgi?id=1241059

        _name = utils.get_repo_name(name) if is_dist_git(os.getcwd()) else name

        if six.PY2 and isinstance(filename, six.text_type):

            filename = filename.encode('utf-8')

        if six.PY2 and isinstance(self.branch, six.text_type):

            self.branch = self.branch.encode('utf-8')

        post_data = [('name', _name),

                     ('%ssum' % self.hashtype, hash),

                     ('branch', self.branch),

                     ('filename', filename)]

        with io.BytesIO() as buf:

            c = pycurl.Curl()

            c.setopt(pycurl.URL, self.upload_url)

            c.setopt(pycurl.WRITEFUNCTION, buf.write)

            c.setopt(pycurl.HTTPPOST, post_data)

            if self.client_cert is not None:

                if os.path.exists(self.client_cert):

                    c.setopt(pycurl.SSLCERT, self.client_cert)

                else:

                    self.log.warning("Missing certificate: %s"

                                     % self.client_cert)

            if self.ca_cert is not None:

                if os.path.exists(self.ca_cert):

                    c.setopt(pycurl.CAINFO, self.ca_cert)

                else:

                    self.log.warning("Missing certificate: %s", self.ca_cert)

            c.setopt(pycurl.HTTPAUTH, pycurl.HTTPAUTH_GSSNEGOTIATE)

            c.setopt(pycurl.USERPWD, ':')

            try:

                c.perform()

                status = c.getinfo(pycurl.RESPONSE_CODE)

            except Exception as e:

                raise UploadError(e)

            finally:

                c.close()

            output = buf.getvalue().strip()

        if status != 200:

            self.raise_upload_error(status)

        # Lookaside CGI script returns these strings depending on whether

        # or not the file exists:

        if output == b'Available':

            return True

        if output == b'Missing':

            return False

        # Something unexpected happened

        self.log.debug(output)

        raise UploadError('Error checking for %s at %s'

                          % (filename, self.upload_url))

    def upload(self, name, filepath, hash):

        """Upload a source file

        :param str name: The name of the module. (usually the name of the SRPM)

            This can include the namespace as well (depending on what the

            server side expects).

        :param str filepath: The full path to the file to upload.

        :param str hash: The known good hash of the file.

        """

        filename = os.path.basename(filepath)

        # As in remote_file_exists, we need to convert unicode strings to str

        if six.PY2:

            if isinstance(name, six.text_type):

                name = name.encode('utf-8')

            if isinstance(filepath, six.text_type):

                filepath = filepath.encode('utf-8')

        if self.remote_file_exists(name, filename, hash):

            self.log.info("File already uploaded: %s", filepath)

            return

        self.log.info("Uploading: %s", filepath)

        post_data = [('name', name),

                     ('%ssum' % self.hashtype, hash),

                     ('branch', self.branch),

                     ('file', (pycurl.FORM_FILE, filepath))]

        with io.BytesIO() as buf:

            c = pycurl.Curl()

            c.setopt(pycurl.URL, self.upload_url)

            c.setopt(pycurl.NOPROGRESS, False)

            c.setopt(pycurl.PROGRESSFUNCTION, self.print_progress)

            c.setopt(pycurl.WRITEFUNCTION, buf.write)

            c.setopt(pycurl.HTTPPOST, post_data)

            if self.client_cert is not None:

                if os.path.exists(self.client_cert):

                    c.setopt(pycurl.SSLCERT, self.client_cert)

                else:

                    self.log.warning("Missing certificate: %s", self.client_cert)

            if self.ca_cert is not None:

                if os.path.exists(self.ca_cert):

                    c.setopt(pycurl.CAINFO, self.ca_cert)

                else:

                    self.log.warning("Missing certificate: %s", self.ca_cert)

            c.setopt(pycurl.HTTPAUTH, pycurl.HTTPAUTH_GSSNEGOTIATE)

            c.setopt(pycurl.USERPWD, ':')

            try:

                c.perform()

                status = c.getinfo(pycurl.RESPONSE_CODE)

            except Exception as e:

                raise UploadError(e)

            finally:

                c.close()

            output = buf.getvalue().strip()

        # Get back a new line, after displaying the download progress

        sys.stdout.write('\n')

        sys.stdout.flush()

        if status != 200:

            self.raise_upload_error(status)

        if output:

            self.log.debug(output)