Initial commit

Based on dc4ba6a

2 files changed, 46 insertions, 0 deletions

diff --git a/script/archiver/default.nix b/script/archiver/default.nix
new file mode 100644
index 0000000..4a464e0
--- /dev/null
+++ b/script/archiver/default.nix
@@ -0,0 +1,15 @@
+{ pkgs ? import <nixpkgs> { } }: with pkgs;
+
+stdenv.mkDerivation {
+  name = "oeuf-archiver";
+  src = ./.;
+
+  buildInputs = [ bash rclone oeuf-bundleparquet ];
+  nativeBuildInputs = [ makeWrapper ];
+  installPhase = ''
+    mkdir -p $out/bin
+    cp oeuf-archiver.sh $out/bin/oeuf-archiver
+    wrapProgram $out/bin/oeuf-archiver \
+      --prefix PATH : ${lib.makeBinPath [ bash rclone oeuf-bundleparquet ]}
+  '';
+}
diff --git a/script/archiver/oeuf-archiver.sh b/script/archiver/oeuf-archiver.sh
new file mode 100755
index 0000000..478d4d9
--- /dev/null
+++ b/script/archiver/oeuf-archiver.sh
@@ -0,0 +1,31 @@
+#!/usr/bin/env bash
+
+set -eux
+set -o pipefail
+
+# This option prevents the loop from running
+# if it does not match any files
+shopt -s nullglob
+
+oeuf-bundleparquet
+
+export AWS_ACCESS_KEY_ID="$S3_ACCESS_KEY_ID"
+set +x  # Don't print the secret access key to the log
+export AWS_SECRET_ACCESS_KEY="$S3_SECRET_ACCESS_KEY"
+set -x
+
+for file in ./merged/oeuf-*.parquet; do
+        rclone move \
+                --s3-provider "$S3_PROVIDER" \
+                --s3-region "$S3_REGION" \
+                --s3-endpoint "$S3_ENDPOINT" \
+                --s3-env-auth \
+                $file.meta.json :s3:$S3_BUCKET \
+        && \
+        rclone move \
+                --s3-provider "$S3_PROVIDER" \
+                --s3-region "$S3_REGION" \
+                --s3-endpoint "$S3_ENDPOINT" \
+                --s3-env-auth \
+                $file :s3:$S3_BUCKET
+done