From e3acca310231ab211b9cfba4e8a3bd2b58e3d216 Mon Sep 17 00:00:00 2001
From: Fabio Erculiani <lxnay@sabayon.org>
Date: Thu, 13 Oct 2011 20:02:20 +0200
Subject: [PATCH] [eit.commands.reset] new command "eit reset"

---
 server/eit/commands/reset.py | 98 ++++++++++++++++++++++++++++++++++++
 1 file changed, 98 insertions(+)
 create mode 100644 server/eit/commands/reset.py

diff --git a/server/eit/commands/reset.py b/server/eit/commands/reset.py
new file mode 100644
index 000000000..1c47aabae
--- /dev/null
+++ b/server/eit/commands/reset.py
@@ -0,0 +1,98 @@
+# -*- coding: utf-8 -*-
+"""
+
+    @author: Fabio Erculiani <lxnay@sabayon.org>
+    @contact: lxnay@sabayon.org
+    @copyright: Fabio Erculiani
+    @license: GPL-2
+
+    B{Entropy Infrastructure Toolkit}.
+
+"""
+import sys
+import os
+import argparse
+
+from entropy.i18n import _
+from entropy.output import darkgreen, teal
+
+from eit.commands.descriptor import EitCommandDescriptor
+from eit.commands.push import EitPush
+
+
+class EitReset(EitPush):
+    """
+    Main Eit reset command.
+    """
+
+    # TODO: incomplete, needs eit push implemented first
+
+    NAME = "reset"
+    ALIASES = []
+
+    def __init__(self, args):
+        EitPush.__init__(self, args)
+        self._reset_repository_id = None
+        self._local = False
+
+    def parse(self):
+        """ Overridden from EitPush """
+        descriptor = EitCommandDescriptor.obtain_descriptor(
+            EitReset.NAME)
+        parser = argparse.ArgumentParser(
+            description=descriptor.get_description(),
+            formatter_class=argparse.RawDescriptionHelpFormatter,
+            prog="%s %s" % (sys.argv[0], EitReset.NAME))
+
+        parser.add_argument("repo", nargs='?', default=None,
+                            metavar="<repo>", help=_("repository"))
+        parser.add_argument("--quick", action="store_true",
+                            default=False,
+                            help=_("no stupid questions"))
+        parser.add_argument("--local", action="store_true",
+                            default=False,
+                            help=_("do not pull the remote repository"))
+
+        try:
+            nsargs = parser.parse_args(self._args)
+        except IOError as err:
+            return parser.print_help, []
+
+        self._local = nsargs.local
+        self._ask = not nsargs.quick
+        if nsargs.repo is not None:
+            self._repositories.append(nsargs.repo)
+            self._reset_repository_id = nsargs.repo
+
+        return self._call_locked, [self._reset, nsargs.repo]
+
+    def _reset(self, entropy_server):
+        repository_id = self._reset_repository_id
+        if repository_id is None:
+            repository_id = entropy_server.repository()
+        rev_path = entropy_server._get_local_repository_revision_file(
+            repository_id)
+
+        try:
+            with open(rev_path, "w") as rev_f:
+                rev_f.write("0\n")
+        except (IOError, OSError) as err:
+            entropy_server.output(
+                "%s: %s" % (_("reset error"), err),
+                importance=1, level="error")
+            return 1
+
+        entropy_server.output(
+            darkgreen(_("local repository revision reset complete")),
+            importance=1)
+        if self._local:
+            return 0
+
+        return self._push(entropy_server)
+
+EitCommandDescriptor.register(
+    EitCommandDescriptor(
+        EitReset,
+        EitReset.NAME,
+        _('reset repository to remote status'))
+    )