1 files changed, 61 insertions, 0 deletions
diff --git a/node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/task/AddYumRepoTask.java b/node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/task/AddYumRepoTask.java
new file mode 100644
index 00000000000..a2ed6a80084
--- /dev/null
+++ b/node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/task/AddYumRepoTask.java
@@ -0,0 +1,61 @@
+// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespa.hosted.node.admin.task;
+
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.regex.Pattern;
+
+public class AddYumRepoTask implements Task {
+    private static final Pattern REPOSITORY_ID_PATTERN = Pattern.compile("^[a-zA-Z_-]+$");
+
+    private final String repositoryId; // e.g. "platform_rpms-latest"
+    private final String name; // e.g. "Platform RPM Latest Repo"
+    private final String baseurl;
+    private final boolean enabled;
+
+    public AddYumRepoTask(String repositoryId,
+                          String name,
+                          String baseurl,
+                          boolean enabled) {
+        this.repositoryId = repositoryId;
+        this.name = name;
+        this.baseurl = baseurl;
+        this.enabled = enabled;
+        validateRepositoryId(repositoryId);
+    }
+
+    @Override
+    public boolean execute(TaskContext context) {
+        Path path = Paths.get("/etc/yum.repos.d",repositoryId + ".repo");
+
+        if (context.getFileSystem().isRegularFile(path)) {
+            return false;
+        }
+
+        WriteFileTask writeFileTask = new WriteFileTask(path, this::getRepoFileContent)
+                .withOwner("root")
+                .withGroup("root")
+                .withPermissions("rw-r--r--");
+
+        return context.executeSubtask(writeFileTask);
+    }
+
+    String getRepoFileContent() {
+        return String.join("\n",
+                "# This file was generated by node admin",
+                "# Do NOT modify this file by hand",
+                "",
+                "[" + repositoryId + "]",
+                "name=" + name,
+                "baseurl=" + baseurl,
+                "enabled=" + (enabled ? 1 : 0),
+                "gpgcheck=0"
+        ) + "\n";
+    }
+
+    static void validateRepositoryId(String repositoryId) {
+        if (!REPOSITORY_ID_PATTERN.matcher(repositoryId).matches()) {
+            throw new IllegalArgumentException("Invalid repository ID '" + repositoryId + "'");
+        }
+    }
+}