from os import mkdir, walk
from os.path import join, exists
from shutil import rmtree
from subprocess import call

from django.contrib.auth.models import User
from django.core.exceptions import ObjectDoesNotExist
from django.db import models
from django.utils import timezone

from Main.models.task import Task
from Main.models.language import Language
from Sprint.settings import CONSTS, SOLUTIONS_ROOT, SOLUTIONS_ROOT_EXTERNAL


class Solution(models.Model):
    task = models.ForeignKey(Task, on_delete=models.CASCADE)
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    language = models.ForeignKey(Language, on_delete=models.SET_NULL, null=True)
    time_sent = models.DateTimeField(default=timezone.now)
    result = models.TextField(default=CONSTS["in_queue_status"])

    def delete(self, using=None, keep_parents=False):
        if exists(self.directory):
            rmtree(self.directory)
        super().delete(using=using, keep_parents=keep_parents)

    @property
    def files(self):
        data = []
        for path, _, files in walk(self.directory):
            if path.startswith(self.testing_directory):
                continue
            for file in files:
                try:
                    entity = {
                        'filename': file,
                        'text': open(join(path, file), 'r').read()
                    }
                    end = file.split('.')[-1]
                    try:
                        highlight = 'language-' + Language.objects.get(file_type=end).highlight
                    except ObjectDoesNotExist:
                        highlight = 'nohighlight'
                    entity['highlight'] = highlight
                    data.append(entity)
                except:
                    continue
        data.sort(key=lambda x: x['filename'])
        return data

    def create_dirs(self):
        mkdir(self.directory)
        mkdir(self.testing_directory)

    @property
    def directory(self):
        return join(SOLUTIONS_ROOT, str(self.id))

    @property
    def testing_directory(self):
        return join(self.directory, 'test_dir')

    @property
    def volume_directory(self):
        return join(SOLUTIONS_ROOT_EXTERNAL, str(self.id), 'test_dir')

    def exec_command(self, command, working_directory='app', timeout=None):
        return call(f'docker exec -i solution_{self.id} sh -c "cd {working_directory} && {command}"', shell=True, timeout=timeout)