From 0e9d82bd1c5c51a1bbb708cafe7d431928a77ebc Mon Sep 17 00:00:00 2001
From: sgfost <sgfost@asu.edu>
Date: Fri, 6 Sep 2024 13:26:47 -0700
Subject: [PATCH] refactor: use huey task for github mirroring

---
 .../library/codebases/releases/retrieve.jinja | 14 ++++--
 django/library/models.py                      |  3 +-
 django/library/tasks.py                       | 45 +++++++++++++++++++
 django/library/views.py                       |  6 +--
 4 files changed, 59 insertions(+), 9 deletions(-)
 create mode 100644 django/library/tasks.py

diff --git a/django/library/jinja2/library/codebases/releases/retrieve.jinja b/django/library/jinja2/library/codebases/releases/retrieve.jinja
index 4bcaa612f..3984662bd 100644
--- a/django/library/jinja2/library/codebases/releases/retrieve.jinja
+++ b/django/library/jinja2/library/codebases/releases/retrieve.jinja
@@ -303,9 +303,17 @@
                 </a>
             </div>
             {% if codebase.git_mirror %}
-                <a href="{{ codebase.git_mirror.remote_url }}">
-                    <span class='badge bg-info'>view on Github</span>
-                </a>
+                {% if codebase.git_mirror.remote_url %}
+                    <a href="{{ codebase.git_mirror.remote_url }}">
+                        <span class="badge bg-info">view on Github</span>
+                    </a>
+                {% else %}
+                    <div class="alert alert-info">
+                        Mirroring in progress. This may take a few seconds.
+                        <hr>
+                        <a href=".">Refresh</a>
+                    </div>
+                {% endif %}
             {% elif has_change_perm %}
                 <div
                     id="github-mirror"
diff --git a/django/library/models.py b/django/library/models.py
index 838759e52..793748ae6 100644
--- a/django/library/models.py
+++ b/django/library/models.py
@@ -45,10 +45,9 @@
 from core.fields import MarkdownField
 from core.models import Platform, MemberProfile, ModeratedContent
 from core.queryset import get_viewable_objects_for_user
-from core.tasks import send_markdown_email
+from core.utils import send_markdown_email
 from core.view_helpers import get_search_queryset
 from .fs import (
-    CodebaseGitRepositoryApi,
     CodebaseReleaseFsApi,
     StagingDirectories,
     FileCategoryDirectories,
diff --git a/django/library/tasks.py b/django/library/tasks.py
new file mode 100644
index 000000000..09710cf34
--- /dev/null
+++ b/django/library/tasks.py
@@ -0,0 +1,45 @@
+from huey.contrib.djhuey import db_task
+from django.conf import settings
+
+from .models import Codebase
+from .github import GithubApi
+from .fs import CodebaseGitRepositoryApi
+
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+@db_task(retries=3, retry_delay=30)
+def mirror_codebase(codebase_id: int, is_user_repo=False, code=None, debug=False):
+    """asynchronous task that mirrors a codebase to a remote Github repository"""
+    codebase = Codebase.objects.get(id=codebase_id)
+    mirror = codebase.git_mirror
+    if not mirror:
+        raise ValueError("Codebase does not have a git mirror")
+    if is_user_repo:
+        if not code:
+            raise ValueError("User repo requires a code")
+    else:
+        mirror.organization_login = settings.GITHUB_MODEL_LIBRARY_ORG_NAME
+
+    git_fs_api = CodebaseGitRepositoryApi(codebase)
+    local_repo = git_fs_api.get_or_build()
+
+    gh_api = GithubApi(
+        codebase=codebase,
+        local_repo=local_repo,
+        repo_name=mirror.repository_name,
+        is_user_repo=is_user_repo,
+        organization_login=mirror.organization_login,
+        user_access_token=(
+            GithubApi.get_user_access_token(code) if is_user_repo else None
+        ),
+        debug=debug,
+    )
+    repo = gh_api.get_or_create_repo()
+    mirror.remote_url = repo.html_url
+    gh_api.push(local_repo)
+    gh_api.create_releases(local_repo)
+
+    mirror.update_remote_releases()
diff --git a/django/library/views.py b/django/library/views.py
index cf8ad61ed..14eafaffe 100644
--- a/django/library/views.py
+++ b/django/library/views.py
@@ -82,7 +82,7 @@
     PeerReviewFeedbackEditorSerializer,
     PeerReviewEventLogSerializer,
 )
-from .github import GithubService
+from .tasks import mirror_codebase
 
 import logging
 import pathlib
@@ -451,9 +451,7 @@ def github_mirror(self, request, *args, **kwargs):
         CodebaseGitRepositoryApi.check_file_sizes(codebase)
         repo_name = request.data.get("repo_name")
         codebase.create_git_mirror(repo_name)
-        # TODO: do this with a celery task
-        service = GithubService(codebase, debug=True)
-        service.mirror_org_repo()
+        mirror_codebase(codebase.id, debug=True)
         return Response(data={"job_id": "1234"}, status=status.HTTP_202_ACCEPTED)