import os

import pyblish.api
import clique
import pype.api


class ExtractJpegEXR(pyblish.api.InstancePlugin):
    """Resolve any dependency issues

    This plug-in resolves any paths which, if not updated might break
    the published file.

    The order of families is important, when working with lookdev you want to
    first publish the texture, update the texture paths in the nodes and then
    publish the shading network. Same goes for file dependent assets.
    """

    label = "Extract Jpeg EXR"
    hosts = ["shell"]
    order = pyblish.api.ExtractorOrder
    families = ["imagesequence", "render", "render2d", "source"]
    enabled = False

    def process(self, instance):
        start = instance.data.get("frameStart")
        stagingdir = os.path.normpath(instance.data.get("stagingDir"))

        collected_frames = os.listdir(stagingdir)
        collections, remainder = clique.assemble(collected_frames)

        self.log.info("subset {}".format(instance.data['subset']))
        if 'crypto' in instance.data['subset']:
            return

        # get representation and loop them
        representations = instance.data["representations"]

        # filter out mov and img sequences
        representations_new = representations[:]

        for repre in representations:
            self.log.debug(repre)
            if 'review' in repre['tags'] or "thumb-nuke" in repre['tags']:
                if not isinstance(repre['files'], list):
                    return

                input_file = repre['files'][0]

                # input_file = (
                #     collections[0].format('{head}{padding}{tail}') % start
                # )
                full_input_path = os.path.join(stagingdir, input_file)
                self.log.info("input {}".format(full_input_path))

                filename = os.path.splitext(input_file)[0]
                if not filename.endswith('.'):
                    filename += "."
                jpeg_file = filename + "jpg"
                full_output_path = os.path.join(stagingdir, jpeg_file)

                self.log.info("output {}".format(full_output_path))

                config_data = instance.context.data['output_repre_config']

                proj_name = os.environ.get('AVALON_PROJECT', '__default__')
                profile = config_data.get(proj_name, config_data['__default__'])

                jpeg_items = []
                jpeg_items.append(
                    os.path.join(os.environ.get("FFMPEG_PATH"), "ffmpeg"))
                # override file if already exists
                jpeg_items.append("-y")
                # use same input args like with mov
                jpeg_items.extend(profile.get('input', []))
                # input file
                jpeg_items.append("-i {}".format(full_input_path))
                # output file
                jpeg_items.append(full_output_path)

                subprocess_jpeg = " ".join(jpeg_items)

                # run subprocess
                self.log.debug("{}".format(subprocess_jpeg))
                pype.api.subprocess(subprocess_jpeg)

                if "representations" not in instance.data:
                    instance.data["representations"] = []

                representation = {
                    'name': 'thumbnail',
                    'ext': 'jpg',
                    'files': jpeg_file,
                    "stagingDir": stagingdir,
                    "thumbnail": True,
                    "tags": ['thumbnail']
                }

                # adding representation
                self.log.debug("Adding: {}".format(representation))
                representations_new.append(representation)

        instance.data["representations"] = representations_new