# This file is part of htc_utils.
#
# htc_utils 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.
#
# htc_utils 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 htc_utils.  If not, see <http://www.gnu.org/licenses/>.


import os
import subprocess
from . import htcondor_path


class Submit(object):
    def __init__(self, job, **kwargs):
        self.job = job
        self.cluster = None
        self.environ = os.environ
        self._prefix = ':'.join([os.environ['PATH'], htcondor_path()])
        self.environ['PATH'] = self._prefix
        self.cli_args = []

        print("Received job: {}".format(self.job.filename))

    def sendto_name(self, schedd_name):
        self.cli_args.append(['-name', schedd_name])

    def sendto_remote(self, schedd_name):
        self.cli_args.append(['-remote', schedd_name])

    def sendto_addr(self, ip, port):
        addr = ':'.join([ip, str(port)])
        addr = '<' + addr + '>'
        self.cli_args.append(['-addr', addr])

    def sendto_pool(self, pool_name):
        self.cli_args.append(['-pool', pool_name])

    def toggle_verbose(self):
        self.cli_args.append(['-verbose'])

    def toggle_unused_variables(self):
        self.cli_args.append(['-unused'])

    def execute(self):
        condor_submit_args = ' '.join([str(arg) for record in self.cli_args for arg in record])
        condor_submit = ' '.join(['condor_submit', condor_submit_args, self.job.filename])
        proc = subprocess.Popen(condor_submit.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=self.environ)
        stdout, stderr = proc.communicate()
        print(stdout)
        print(stderr)
        proc.wait()

        if not stderr:
            if 'cluster' in stdout:
                for line in stdout.split(os.linesep):
                    if 'cluster' in line:
                        self.cluster = line.split()[-1].strip('.')

    def monitor(self):
        if self.cluster is None:
            print('No cluster data for job.')
            return False
        print('Monitoring cluster {}'.format(self.cluster))
        os.environ['PATH'] = self._prefix
        '''
        proc = subprocess.Popen('condor_q -analyze:summary'.split(), env=self.environ)
        proc.communicate()
        proc.wait()
        '''
        print('NOT IMPLEMENTED')
        return True