From 133f2d3d8b70d4be07bc4d4f7b50a130b5a1dccb Mon Sep 17 00:00:00 2001
From: Michael Marineau <michael.marineau@coreos.com>
Date: Mon, 24 Jun 2013 02:00:01 -0400
Subject: [PATCH] add(update_distfiles): New script to build and upload a
 distfiles mirror

This script wraps emirrordist to scan all ebuilds and fetch every
SRC_URI to construct a distfiles mirror. It can then optionally upload
to storage.core-os.net via gsutil.

Note: This is a work-in-progress, emirrordist expects to operate on a
complete mirror on the local filesystem but maintaining a complete local
copy of everything is burdensome. Before this is truly practical a
modified version of emirrordist will be needed to either accept a list
of already-uploaded files it can assume are OK or operate on the cloud
storage system directly.
---
 update_distfiles | 82 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100755 update_distfiles

diff --git a/update_distfiles b/update_distfiles
new file mode 100755
index 0000000000..a61ba63c02
--- /dev/null
+++ b/update_distfiles
@@ -0,0 +1,82 @@
+#!/bin/bash
+
+# Copyright (c) 2013 The CoreOS Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+SCRIPT_ROOT=$(dirname $(readlink -f "$0"))
+. "${SCRIPT_ROOT}/common.sh" || exit 1
+
+DEFINE_boolean dry_run ${FLAGS_FALSE} "Trial run, makes no changes."
+DEFINE_boolean upload ${FLAGS_FALSE} "Upload distfile mirror via gsutil."
+
+MIRROR_ROOT="${DEFAULT_BUILD_ROOT}/mirror"
+UPLOAD_ROOT="gs://storage.core-os.net/mirror"
+
+# Parse flags
+FLAGS "$@" || exit 1
+eval set -- "${FLAGS_ARGV}"
+switch_to_strict_mode
+
+if [[ $# -eq 0 ]]; then
+    eval set -- portage-stable coreos
+fi
+
+update_local_mirror() {
+    local repo_name="$1"
+    local repo_mirror="${MIRROR_ROOT}/$repo_name"
+    local extra_flags=""
+
+    if [[ ${FLAGS_dry_run} == ${FLAGS_TRUE} ]]; then
+        info "Pretend distfiles update for $repo_name"
+        extra_flags+=" --dry-run "
+    else
+        info "Starting distfiles update for $repo_name"
+    fi
+
+    mkdir -p "${repo_mirror}/"{distfiles,info,log}
+    emirrordist --mirror --verbose $extra_flags \
+        --jobs=${NUM_JOBS} --repo="${repo_name}" \
+        --distfiles="${repo_mirror}/distfiles" \
+        --distfiles-local="/var/lib/portage/distfiles" \
+        --fetch-log-dir="${repo_mirror}/log" \
+        --failure-log="${repo_mirror}/log/failure.log" \
+        --success-log="${repo_mirror}/log/success.log" \
+        --scheduled-deletion-log="${repo_mirror}/log/deletion.log" \
+        --deletion-db="${repo_mirror}/info/deletion.db" \
+        --distfiles-db="${repo_mirror}/info/distfiles.db" \
+        --deletion-delay=$((86400 * 14)) \
+        --restrict-mirror-exemptions="gentoo" \
+        --verify-existing-digest
+}
+upload_mirror() {
+    local repo_name="$1"
+    local local_mirror="${MIRROR_ROOT}/$repo_name"
+    local remote_mirror="${UPLOAD_ROOT}/$repo_name"
+
+    info "Uploading public distfiles for $repo_name"
+    gsutil -m cp -n \
+        "${local_mirror}/distfiles/*" "${remote_mirror}/distfiles"
+
+    info "Uploading private metadata for $repo_name"
+    gsutil cp -a project-private \
+        "${local_mirror}/info/*" "${remote_mirror}/info"
+}
+
+for repo in "$@"; do
+    if ! portageq get_repo_path / "$repo" >/dev/null; then
+        die_notrace "Unknown repo name '$repo'"
+    fi
+    update_local_mirror "$repo"
+done
+
+if [[ ${FLAGS_upload} -eq ${FLAGS_TRUE} ]]; then
+    if [[ ! -f "$HOME/.boto" ]]; then
+        die_notrace "Please run gsutil config to create ~/.boto"
+    fi
+    for repo in "$@"; do
+        upload_mirror "$repo"
+    done
+fi
+
+info "Done!"