From 92c67e6b5373454a172c21b21ed309b98f0aa34c Mon Sep 17 00:00:00 2001
From: Jakub Kadlcik <frostyx@email.cz>
Date: Mon, 11 Nov 2024 11:49:52 +0100
Subject: [PATCH] backend, frontend: add script to migrate existing build
 results to Pulp

Fix #3503
---
 backend/run/copr-change-storage               | 119 ++++++++++++++++++
 .../coprs_frontend/commands/change_storage.py |  11 +-
 2 files changed, 127 insertions(+), 3 deletions(-)
 create mode 100755 backend/run/copr-change-storage

diff --git a/backend/run/copr-change-storage b/backend/run/copr-change-storage
new file mode 100755
index 000000000..b0c095620
--- /dev/null
+++ b/backend/run/copr-change-storage
@@ -0,0 +1,119 @@
+#! /usr/bin/python3
+
+"""
+Migrate existing build results for a given project and all of its CoprDirs
+from one storage (Copr backend) to another (Pulp).
+"""
+
+import os
+import sys
+import argparse
+import logging
+from copr_common.log import setup_script_logger
+from copr_backend.helpers import BackendConfigReader
+from copr_backend.storage import PulpStorage
+
+
+STORAGES = ["backend", "pulp"]
+
+log = logging.getLogger(__name__)
+setup_script_logger(log, "/var/log/copr-backend/change-storage.log")
+
+
+def get_arg_parser():
+    """
+    CLI argument parser
+    """
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "--src",
+        required=True,
+        choices=STORAGES,
+        help="The source storage",
+    )
+    parser.add_argument(
+        "--dst",
+        required=True,
+        choices=STORAGES,
+        help="The destination storage",
+    )
+    parser.add_argument(
+        "--project",
+        required=True,
+        help="Full name of the project that is to be migrated",
+    )
+    parser.add_argument(
+        "--delete",
+        action="store_true",
+        default=False,
+        help="After migrating the data, remove it from the old storage",
+    )
+    return parser
+
+
+def main():
+    """
+    The main function
+    """
+    parser = get_arg_parser()
+    args = parser.parse_args()
+
+    if args.src == args.dst:
+        print("The source and destination storage is the same, nothing to do.")
+        return
+
+    if args.src == "pulp":
+        print("Migration from pulp to somewhere else is not supported")
+        sys.exit(1)
+
+    if args.delete:
+        print("Data removal is not supported yet")
+        sys.exit(1)
+
+    config_file = "/etc/copr/copr-be.conf"
+    config = BackendConfigReader(config_file).read()
+    owner, project = args.project.split("/")
+    ownerdir = os.path.join(config.destdir, owner)
+
+    for subproject in os.listdir(ownerdir):
+        if not (subproject == project or subproject.startswith(project + ":")):
+            continue
+
+        coprdir = os.path.join(ownerdir, subproject)
+        for chroot in os.listdir(coprdir):
+            if chroot == "srpm-builds":
+                continue
+
+            chrootdir = os.path.join(coprdir, chroot)
+            if not os.path.isdir(chrootdir):
+                continue
+
+            appstream = None
+            devel = None
+            storage = PulpStorage(
+                owner, subproject, appstream, devel, config, log)
+
+            for builddir in os.listdir(chrootdir):
+                resultdir = os.path.join(chrootdir, builddir)
+                if not os.path.isdir(resultdir):
+                    continue
+
+                full_name = "{0}/{1}".format(owner, subproject)
+                result = storage.init_project(full_name, chroot)
+                if not result:
+                    log.error("Failed to initialize project: %s", resultdir)
+                    break
+
+                # We cannot check return code here
+                storage.upload_build_results(chroot, resultdir, None)
+
+                result = storage.publish_repository(chroot)
+                if not result:
+                    log.error("Failed to publish a repository: %s", resultdir)
+                    break
+
+                log.info("OK: %s", resultdir)
+
+
+if __name__ == "__main__":
+    main()
diff --git a/frontend/coprs_frontend/commands/change_storage.py b/frontend/coprs_frontend/commands/change_storage.py
index 4983b04a0..4a97f016a 100644
--- a/frontend/coprs_frontend/commands/change_storage.py
+++ b/frontend/coprs_frontend/commands/change_storage.py
@@ -6,8 +6,8 @@
 configure the storage type for a given project and while doing so, it makes sure
 DNF repositories for the project are created.
 
-Existing builds are not migrated from one storage to another. This may be an
-useful feature but it is not implemented yet.
+To migrate existing build results for a given project and all of its CoprDirs,
+run also `copr-change-storage` script on backend.
 """
 
 import sys
@@ -44,4 +44,9 @@ def change_storage(fullname, storage):
     db.session.commit()
     print("Configured storage for {0} to {1}".format(copr.full_name, storage))
     print("Submitted action to create repositories: {0}".format(action.id))
-    print("Existing builds not migrated (not implemented yet).")
+    print("To migrate existing build results for this project and all of its "
+          "CoprDirs, run also this command on backend:")
+
+    cmd = "sudo -u copr copr-change-storage --src backend --dst pulp "
+    cmd += "--project {0}".format(fullname)
+    print("    {0}".format(cmd))