# Copyright (C) 2015 The Sipwise Team - http://sipwise.com

# 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 3 of the License, or (at your option)
# any later version.

# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
# more details.

# You should have received a copy of the GNU General Public License along
# with this program.  If not, see <http://www.gnu.org/licenses/>.
from __future__ import absolute_import

from distutils.dir_util import mkpath
import logging
import os
import shutil
import subprocess
import urllib2
from django.conf import settings

logger = logging.getLogger(__name__)

JBI_CONSOLE_URL = "{}/job/{}/{}/consoleText"
JBI_BUILD_URL = "{}/job/{}/{}/api/json"
JBI_ARTIFACT_URL = "{}/job/{}/{}/artifact/{}"
JBI_ENVVARS_URL = "{}/job/{}/{}/injectedEnvVars/api/json"


def executeAndReturnOutput(command, env=None):
    proc = subprocess.Popen(command, stdout=subprocess.PIPE,
                            stderr=subprocess.PIPE, env=env)
    stdoutdata, stderrdata = proc.communicate()
    logger.debug("<stdout>%s</stdout>", stdoutdata)
    logger.debug("<strerr>%s</stderr>", stderrdata)
    return proc.returncode, stdoutdata, stderrdata


def dlfile(url, path):
    if settings.DEBUG:
        logger.info("I would call %s", url)
    else:
        remote_file = urllib2.urlopen(url)
        logger.debug("url:[%s]", url)
        with open(path, "wb") as local_file:
            shutil.copyfileobj(remote_file, local_file)


def openurl(url):
    req = urllib2.Request(url)
    logger.debug("url:[%s]", url)
    response = urllib2.urlopen(req)
    if response.code > 199 and response.code < 300:
        logger.debug("OK[%d] url: %s", url, response.code)
        return True
    else:
        logger.error("Error[%d] retrieving %s", url, response.code)
        return False


def jenkins_remove_ppa(repo):
    url = "%s/job/remove-reprepro-codename/buildWithParameters?"\
        "token=%s&repository=%s" % \
        (settings.JENKINS_URL, settings.JENKINS_TOKEN, repo)
    if settings.DEBUG:
        logger.debug("I would call %s", url)
    else:
        openurl(url)


def _jenkins_get(url, base_path, filename):
    mkpath(base_path)
    path = os.path.join(base_path, filename)
    logger.debug("url:[%s] path[%s]", url, path)
    dlfile(url, path)
    return path


def jenkins_get_console(jobname, buildnumber):
    url = JBI_CONSOLE_URL.format(
        settings.JENKINS_URL,
        jobname,
        buildnumber
    )
    base_path = os.path.join(settings.JBI_BASEDIR,
                             jobname, str(buildnumber))
    return _jenkins_get(url, base_path, 'console.txt')


def jenkins_get_build(jobname, buildnumber):
    url = JBI_BUILD_URL.format(
        settings.JENKINS_URL,
        jobname,
        buildnumber
    )
    base_path = os.path.join(settings.JBI_BASEDIR,
                             jobname, str(buildnumber))
    return _jenkins_get(url, base_path, 'build.json')


def jenkins_get_env(jobname, buildnumber):
    url = JBI_ENVVARS_URL.format(
        settings.JENKINS_URL,
        jobname,
        buildnumber
    )
    base_path = os.path.join(settings.JBI_BASEDIR,
                             jobname, str(buildnumber))
    return _jenkins_get(url, base_path, 'envVars.json')


def jenkins_get_artifact(jobname, buildnumber, artifact_info):
    url = JBI_ARTIFACT_URL.format(
        settings.JENKINS_URL,
        jobname,
        buildnumber,
        artifact_info["relativePath"]
    )
    base_path = os.path.join(settings.JBI_BASEDIR,
                             jobname, str(buildnumber), 'artifact')
    return _jenkins_get(url, base_path, artifact_info["fileName"])


def workfront_note_send(_id, message):
    command = [
        "/usr/bin/workfront-post-note",
        "--credfile=%s" % settings.WORKFRONT_CREDENTIALS,
        "--private",
        "--taskid=%s" % _id,
        '--message="%s"' % message
    ]
    logger.debug("workfront-port-note command: %s", command)
    res = executeAndReturnOutput(command)
    if res[0] != 0:
        logger.error("can't post workfront note. %s. %s", res[1], res[2])
        return False
    return True