# -*- coding: utf-8 -*-


# 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; version 2 of the License.
#
# 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 Library 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 <https://gnu.org/licenses/>.


import os
import json
from kobo import shortcuts

from pungi.util import makedirs
from .base import OSTree
from .utils import (make_log_file, tweak_treeconf,
                    get_ref_from_treefile, get_commitid_from_commitid_file)


class Tree(OSTree):
    def _init_repo(self):
        """If the ostree repo does not exist, initialize it."""
        log_file = make_log_file(self.logdir, 'init-ostree-repo')
        if not os.path.isdir(self.repo) or not os.listdir(self.repo):
            makedirs(self.repo)
            shortcuts.run(['ostree', 'init', '--repo=%s' % self.repo, '--mode=archive-z2'],
                          show_cmd=True, stdout=True, logfile=log_file)

    def _make_tree(self):
        """Compose OSTree tree"""
        log_file = make_log_file(self.logdir, 'create-ostree-repo')
        cmd = ['rpm-ostree', 'compose', 'tree', '--repo=%s' % self.repo,
               '--write-commitid-to=%s' % self.commitid_file]
        if self.version:
            # Add versioning metadata
            cmd.append('--add-metadata-string=version=%s' % self.version)
        cmd.append(self.treefile)

        shortcuts.run(cmd, show_cmd=True, stdout=True, logfile=log_file)

    def _update_summary(self):
        """Update summary metadata"""
        log_file = make_log_file(self.logdir, 'ostree-summary')
        shortcuts.run(['ostree', 'summary', '-u', '--repo=%s' % self.repo],
                      show_cmd=True, stdout=True, logfile=log_file)

    def _update_ref(self):
        """
        Update the ref.

        '--write-commitid-to' is specified when compose the tree, so we need
        to update the ref by ourselves. ref is retrieved from treefile and
        commitid is retrieved from the committid file.
        """
        tag_ref = True
        if self.extra_config:
            tag_ref = self.extra_config.get('tag_ref', True)
        if not tag_ref:
            print('Not updating ref as configured')
            return
        ref = get_ref_from_treefile(self.treefile)
        commitid = get_commitid_from_commitid_file(self.commitid_file)
        print('Ref: %r, Commit ID: %r' % (ref, commitid))
        if ref and commitid:
            print('Updating ref')
            # Let's write the tag out ourselves
            heads_dir = os.path.join(self.repo, 'refs', 'heads')
            if not os.path.exists(heads_dir):
                raise RuntimeError('Refs/heads did not exist in ostree repo')

            ref_path = os.path.join(heads_dir, ref)
            makedirs(os.path.dirname(ref_path))
            with open(ref_path, 'w') as f:
                f.write(commitid + '\n')

    def run(self):
        self.repo = self.args.repo
        self.treefile = self.args.treefile
        self.version = self.args.version
        self.logdir = self.args.log_dir
        self.update_summary = self.args.update_summary
        self.extra_config = self.args.extra_config
        if self.extra_config:
            self.extra_config = json.load(open(self.extra_config, 'r'))
            repos = self.extra_config.get('repo', [])
            keep_original_sources = self.extra_config.get('keep_original_sources', False)
            tweak_treeconf(self.treefile, source_repos=repos, keep_original_sources=keep_original_sources)

        self.commitid_file = make_log_file(self.logdir, 'commitid')

        self._init_repo()
        self._make_tree()
        self._update_ref()
        if self.update_summary:
            self._update_summary()