1 files changed, 89 insertions, 0 deletions
diff --git a/MgaRepo/GitHub.py b/MgaRepo/GitHub.py
new file mode 100644
index 0000000..71f53d6
--- /dev/null
+++ b/MgaRepo/GitHub.py
@@ -0,0 +1,89 @@
+from MgaRepo import Error, config
+from MgaRepo.rpmutil import detectVCS, get_pkg_tag, clone
+from MgaRepo.util import execcmd
+from MgaRepo import layout
+from MgaRepo.git import GIT
+from MgaRepo.svn import SVN
+from rpm import RPMTAG_SUMMARY, RPMTAG_URL
+import github
+import os
+
+class GitHub(object):
+    def __init__(self, username = config.get("github", "login"), password = config.get("github", "password")):
+        self._github = github.Github(login_or_token=username, password=password)
+        self._organization = self._github.get_organization(config.get("github", "organization", "mdkcauldron"))
+        self._repos = self._organization.get_repos()
+
+    def repository_exists(self, name):
+        for repo in self._repos:
+            if repo.name == name:
+                return repo
+        return None
+
+    def create_repository(self, pkgname, **kwargs):
+        repository = self._organization.create_repo(pkgname, **kwargs)
+        return repository
+
+    def delete_repository(self, pkgname, **kwargs):
+        repository = self.repository_exists(pkgname)
+        if repository:
+            print("deleting repository %s" % repository.full_name)
+            repository.delete()
+            return True
+        raise Error("repository %s doesn't exist!" % (self._organization.login+"/"+pkgname))
+
+    def import_package(self, target):
+        if not os.path.exists(target):
+            target = layout.checkout_url(layout.package_url(target))
+        vcs = detectVCS(target)
+        top_dir = vcs.get_topdir()
+        pkgname = layout.package_name(layout.remove_current(vcs.url))
+
+        repository = self.repository_exists(pkgname)
+        if not repository or not repository.get_stats_commit_activity():
+            if not repository:
+                if os.path.exists(vcs.path):
+                    summary = get_pkg_tag(RPMTAG_SUMMARY, path=top_dir)
+                    url = get_pkg_tag(RPMTAG_URL, path=top_dir)
+                    repository = self.create_repository(pkgname, description=summary, homepage=url)
+                    print("GitHub repository created at " + repository.html_url)
+            else:
+                print("Empty GitHub repository already created at %s, using" % repository.html_url)
+
+            if isinstance(vcs, GIT):
+                status, output = vcs.remote("add", repository.full_name, repository.ssh_url, noerror=True)
+                if status:
+                    if status == 128 and ("fatal: remote %s already exists." % repository.full_name) \
+                            in output:
+                                pass
+                    else:
+                        raise Error(output)
+
+                status, output = vcs.push("--mirror", repository.full_name, show=True)
+                if status == 0:
+                    print("Success!")
+                    return True
+            elif isinstance(vcs, SVN):
+                clone(vcs.url, bindownload=False)
+                return self.import_package(pkgname)
+
+        else:
+            raise Error("GitHub repository already exists at " + repository.html_url)
+        raise Error("GitHub import failed...")
+
+    def clone_repository(self, pkgname, target=None):
+        if not target:
+            target = pkgname
+        repository = self.repository_exists(pkgname)
+        if repository:
+            svnurl = layout.checkout_url(layout.package_url(pkgname))
+            if repository.permissions:
+                giturl = repository.ssh_url
+            else:
+                giturl = repository.git_url
+            execcmd(("git", "clone", "--mirror", giturl, os.path.join(target, ".git")), show=True)
+            git_svn = GIT(path=target, url=svnurl)
+            git_svn.init(svnurl, pkgname, branch="master", fullnames=True)
+
+            return True
+        raise Error("Repository %s doesn't exist!" % (self._organization.login+"/"+pkgname))