add SecrectWriter, PVWriter and PVCWriter

Author: syswu

src/main/java/io/github/kubesys/client/writers/KindWriter.java

@@ -0,0 +1,64 @@
+/**
+ * Copyright (2023, ) Institute of Software, Chinese Academy of Sciences
+ */
+package io.github.kubesys.client.writers;
+
+import java.io.PrintStream;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
+import com.fasterxml.jackson.dataformat.yaml.YAMLMapper;
+
+import io.github.kubesys.client.addons.KubernetesWriter;
+
+/**
+ * @author wuheng@iscas.ac.cn
+ * @since  2023/07/26
+ * @version 1.0.2
+ *
+ */
+public abstract class KindWriter {
+
+	protected final KubernetesWriter writer;
+	
+	protected final ObjectNode json; 
+
+	public KindWriter(String name) throws Exception {
+		this.writer = new KubernetesWriter();
+		this.json = toObjectNode(getTemplate(), new String[] {"#NAME#", name});
+	}
+	
+	public KindWriter(String name, String namespace) throws Exception {
+		this.writer = new KubernetesWriter();
+		this.json = toObjectNode(getTemplate(), new String[] {"#NAME#", name, "#NAMESPACE#", namespace});
+	}
+	
+	public void stream(PrintStream ps) throws Exception {
+		ps.println(new YAMLMapper().writeValueAsString(json));
+	}
+	
+	public void json(String path) {
+		writer.writeAsJson(path, json);
+	}
+	
+	public void yaml(String path) {
+		writer.writeAsYaml(path, json);
+	}
+	
+	public ObjectNode toObjectNode(String str, String[] list) throws Exception {
+		for (int i = 0; i < list.length; i = i + 2) {
+			str = str.replaceAll(list[i], list[i + 1]);
+		}
+		return  (ObjectNode) new ObjectMapper(new YAMLFactory()).readTree(str);
+	}
+	
+	public ObjectNode toObjectNode(String key, JsonNode json) throws Exception {
+		ObjectNode val = new ObjectMapper().createObjectNode();
+		val.set(key, json);
+		return  (ObjectNode) new ObjectMapper(new YAMLFactory()).readTree(val.toPrettyString());
+	}
+	
+	public abstract String getTemplate();
+}

src/main/java/io/github/kubesys/client/writers/PVCWriter.java

@@ -0,0 +1,50 @@
+/**
+ * Copyright (2023, ) Institute of Software, Chinese Academy of Sciences
+ */
+package io.github.kubesys.client.writers;
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+
+/**
+ * @author wuheng@iscas.ac.cn
+ * @since  2023/07/26
+ * @version 1.0.2
+ *
+ */
+public class PVCWriter extends KindWriter {
+
+	static final String TEMPLATE = "apiVersion: v1\r\n"
+			+ "kind: PersistentVolumeClaim\r\n"
+			+ "metadata:\r\n"
+			+ "  name: #NAME#\r\n"
+			+ "  namespace: #NAMESPACE#\r\n"
+			+ "  labels:\r\n"
+			+ "    name: \"#NAME#\"\r\n"
+			+ "spec:\r\n"
+			+ "  accessModes:\r\n"
+			+ "    - ReadWriteOnce";
+	
+	public PVCWriter(String name, String namespace) throws Exception {
+		super(name, namespace);
+	}
+
+	static final String CAPACITY = "storage: #SIZE#Gi";
+	
+	public PVCWriter withCapacity(String gb) throws Exception {
+		ObjectNode size = toObjectNode(CAPACITY, new String[] {"#SIZE#", gb});
+		ObjectNode requests = toObjectNode("requests", size);
+		ObjectNode spec = (ObjectNode) json.get("spec");
+		spec.set("resources", requests);
+		return this;
+	}
+	
+	@Override
+	public String getTemplate() {
+		return TEMPLATE;
+	}
+	
+	public static void main(String[] args) throws Exception {
+		PVCWriter writer = new PVCWriter("kube-database", "kube-system");
+		writer.withCapacity("2").stream(System.out);
+	}
+}

src/main/java/io/github/kubesys/client/writers/PVWriter.java

@@ -0,0 +1,66 @@
+/**
+ * Copyright (2023, ) Institute of Software, Chinese Academy of Sciences
+ */
+package io.github.kubesys.client.writers;
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+
+/**
+ * @author wuheng@iscas.ac.cn
+ * @since  2023/07/26
+ * @version 1.0.2
+ *
+ */
+public class PVWriter extends KindWriter {
+
+	static final String TEMPLATE = "apiVersion: v1\r\n"
+			+ "kind: PersistentVolume\r\n"
+			+ "metadata:\r\n"
+			+ "  name: #NAME#\r\n"
+			+ "spec:\r\n"
+			+ "  accessModes:\r\n"
+			+ "  - ReadWriteOnce\r\n"
+			+ "  persistentVolumeReclaimPolicy: Retain\r\n"
+			+ "  volumeMode: Filesystem";
+	
+	public PVWriter(String name) throws Exception {
+		super(name);
+	}
+
+	static final String CAPACITY = "storage: #SIZE#Gi";
+	
+	public PVWriter withCapacity(String gb) throws Exception {
+		ObjectNode spec = (ObjectNode) json.get("spec");
+		spec.set("capacity", toObjectNode(CAPACITY, new String[] {"#SIZE#", gb}));
+		return this;
+	}
+	
+	static final String CLAIMREF = "apiVersion: v1\r\n"
+			+ "kind: PersistentVolumeClaim\r\n"
+			+ "name: #PV_NAME#\r\n"
+			+ "namespace: #PV_NAMESPACE#";
+	
+	public PVWriter withPVC(String name, String namespace) throws Exception {
+		ObjectNode spec = (ObjectNode) json.get("spec");
+		spec.set("claimRef", toObjectNode(CLAIMREF, new String[] {"#PV_NAME#", name, "#PV_NAMESPACE#", namespace}));
+		return this;
+	}
+	
+	static final String HOSTPATH = "path: #PATH#";
+	
+	public PVWriter withPath(String hostpath) throws Exception {
+		ObjectNode spec = (ObjectNode) json.get("spec");
+		spec.set("hostPath", toObjectNode(HOSTPATH, new String[] {"#PATH#", hostpath}));
+		return this;
+	}
+	
+	@Override
+	public String getTemplate() {
+		return TEMPLATE;
+	}
+	
+	public static void main(String[] args) throws Exception {
+		PVWriter writer = new PVWriter("kube-database");
+		writer.withCapacity("2").withPVC("kube-database", "kube-system").withPath("/var/lib/doslab/kube/postgres").stream(System.out);
+	}
+}

src/main/java/io/github/kubesys/client/writers/SecretWriter.java

@@ -0,0 +1,47 @@
+/**
+ * Copyright (2023, ) Institute of Software, Chinese Academy of Sciences
+ */
+package io.github.kubesys.client.writers;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+
+/**
+ * @author wuheng@iscas.ac.cn
+ * @since  2023/07/26
+ * @version 1.0.2
+ *
+ */
+public class SecretWriter extends KindWriter {
+
+	static final String TEMPLATE = "apiVersion: v1\r\n"
+			+ "kind: Secret\r\n"
+			+ "metadata:\r\n"
+			+ "  name: #NAME#\r\n"
+			+ "  namespace: #NAMESPACE#\r\n"
+			+ "type: Opaque";
+	
+	public SecretWriter(String name, String namespace) throws Exception {
+		super(name, namespace);
+	}
+
+	public SecretWriter withData(String key, String value) {
+		if (!json.has("data")) {
+			json.set("data", new ObjectMapper().createObjectNode());
+		}
+		
+		ObjectNode data = (ObjectNode) json.get("data");
+		data.put(key, value);
+		return this;
+	}
+	
+	@Override
+	public String getTemplate() {
+		return TEMPLATE;
+	}
+	
+	public static void main(String[] args) throws Exception {
+		SecretWriter writer = new SecretWriter("kube-database", "kube-system");
+		writer.withData("username", "onceas").withData("password", "onceas").stream(System.out);
+	}
+}