Replace createServerKeys.sh (#314)

Author: PoloSpark

server/scripts/entrypoint.sh

@@ -1,4 +1,3 @@
 #! /bin/sh
 mkdir keys
-./scripts/createServerKeys.sh /app/keys
-exec java -Djwt.private.key=file:/app/keys/app.key  -Djwt.public.key=file:/app/keys/app.pub -jar findfirst.jar
+exec java -Djwt.private.key=file:/app/keys/app.key  -Djwt.public.key=file:/app/keys/app.pub -jar findfirst.jar

server/src/main/java/dev/findfirst/FindFirstApplication.java

@@ -1,8 +1,12 @@
 package dev.findfirst;
 
+import java.io.IOException;
+import java.security.NoSuchAlgorithmException;
 import java.util.Arrays;
 import java.util.Collections;
 
+import dev.findfirst.security.util.KeyGenerator;
+
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
@@ -22,8 +26,9 @@ public class FindFirstApplication {
 
   public static void main(String[] args) {
     try {
+      KeyGenerator.generateKeys("app.key", "app.pub");
       SpringApplication.run(FindFirstApplication.class, args);
-    } catch (ApplicationContextException ude) {
+    } catch (ApplicationContextException | NoSuchAlgorithmException | IOException ude) {
       log.error("""
 
 

server/src/main/java/dev/findfirst/security/util/KeyGenerator.java

@@ -0,0 +1,55 @@
+package dev.findfirst.security.util;
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.security.KeyPair;
+import java.security.KeyPairGenerator;
+import java.security.NoSuchAlgorithmException;
+import java.security.PrivateKey;
+import java.security.PublicKey;
+import java.security.spec.PKCS8EncodedKeySpec;
+import java.security.spec.X509EncodedKeySpec;
+
+public class KeyGenerator {
+
+  public static void generateKeys(String privateKeyPath, String publicKeyPath)
+      throws NoSuchAlgorithmException, IOException {
+    KeyPairGenerator keyGen = KeyPairGenerator.getInstance("RSA");
+    keyGen.initialize(2048);
+    KeyPair pair = keyGen.generateKeyPair();
+    PrivateKey privateKey = pair.getPrivate();
+    PublicKey publicKey = pair.getPublic();
+
+    // Save the private key in PKCS8 format
+    PKCS8EncodedKeySpec pkcs8EncodedKeySpec = new PKCS8EncodedKeySpec(privateKey.getEncoded());
+    try (FileOutputStream fos = new FileOutputStream(privateKeyPath)) {
+      fos.write(pkcs8EncodedKeySpec.getEncoded());
+    }
+
+    // Save the public key in X.509 format
+    X509EncodedKeySpec x509EncodedKeySpec = new X509EncodedKeySpec(publicKey.getEncoded());
+    try (FileOutputStream fos = new FileOutputStream(publicKeyPath)) {
+      fos.write(x509EncodedKeySpec.getEncoded());
+    }
+  }
+
+  public static void main(String[] args) {
+    String privateKeyPath = "app.key";
+    String publicKeyPath = "app.pub";
+
+    File privateKeyFile = new File(privateKeyPath);
+    File publicKeyFile = new File(publicKeyPath);
+
+    if (!privateKeyFile.exists() || !publicKeyFile.exists()) {
+      try {
+        generateKeys(privateKeyPath, publicKeyPath);
+        System.out.println("Keys generated successfully.");
+      } catch (NoSuchAlgorithmException | IOException e) {
+        e.printStackTrace();
+      }
+    } else {
+      System.out.println("Keys already exist.");
+    }
+  }
+}