1 files changed, 73 insertions, 0 deletions
diff --git a/taskcluster/docker/index-task/insert-indexes.js b/taskcluster/docker/index-task/insert-indexes.js
new file mode 100644
index 0000000000..39ff45fd9c
--- /dev/null
+++ b/taskcluster/docker/index-task/insert-indexes.js
@@ -0,0 +1,73 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+let taskcluster = require("taskcluster-client");
+
+// Create instance of index client
+let index = new taskcluster.Index({
+  delayFactor: 750, // Good solid delay for background process
+  retries: 8, // A few extra retries for robustness
+  rootUrl:
+    process.env.TASKCLUSTER_PROXY_URL || process.env.TASKCLUSTER_ROOT_URL,
+});
+
+// Create queue instance for fetching taskId
+let queue = new taskcluster.Queue({
+  delayFactor: 750, // Good solid delay for background process
+  retries: 8, // A few extra retries for robustness
+  rootUrl:
+    process.env.TASKCLUSTER_PROXY_URL || process.env.TASKCLUSTER_ROOT_URL,
+});
+
+// Load input
+let taskId = process.env.TARGET_TASKID;
+let rank = parseInt(process.env.INDEX_RANK, 10);
+let namespaces = process.argv.slice(2);
+
+// Validate input
+if (!taskId) {
+  console.log("Expected target task as environment variable: TARGET_TASKID");
+  process.exit(1);
+}
+
+if (isNaN(rank)) {
+  console.log("Expected index rank as environment variable: INDEX_RANK");
+  process.exit(1);
+}
+
+// Fetch task definition to get expiration and then insert into index
+queue
+  .task(taskId)
+  .then(task => task.expires)
+  .then(expires => {
+    return Promise.all(
+      namespaces.map(namespace => {
+        console.log(
+          "Inserting %s into index (rank %d) under: %s",
+          taskId,
+          rank,
+          namespace
+        );
+        return index.insertTask(namespace, {
+          taskId,
+          rank,
+          data: {},
+          expires,
+        });
+      })
+    );
+  })
+  .then(() => {
+    console.log("indexing successfully completed.");
+    process.exit(0);
+  })
+  .catch(err => {
+    console.log("Error:\n%s", err);
+    if (err.stack) {
+      console.log("Stack:\n%s", err.stack);
+    }
+    console.log("Properties:\n%j", err);
+    throw err;
+  })
+  .catch(() => process.exit(1));