Use Config struct instead of passing values to generateCertificate

LyleMi · Jan 6, 2024 · 064f71b · 064f71b
1 parent b4c1f31
commit 064f71b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/cert.go b/cert.go
@@ -11,7 +11,7 @@ import (
 	"time"
 )
 
-func generateCertificate(certPath, keyPath string) error {
+func generateCertificate() error {
 	priv, err := rsa.GenerateKey(rand.Reader, 2048)
 	if err != nil {
 		return err
@@ -35,14 +35,14 @@ func generateCertificate(certPath, keyPath string) error {
 		return err
 	}
 
-	certOut, err := os.Create(certPath)
+	certOut, err := os.Create(Config.Cert)
 	if err != nil {
 		return err
 	}
 	defer certOut.Close()
 	pem.Encode(certOut, &pem.Block{Type: "CERTIFICATE", Bytes: derBytes})
 
-	keyOut, err := os.OpenFile(keyPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0600)
+	keyOut, err := os.OpenFile(Config.Key, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0600)
 	if err != nil {
 		return err
 	}

diff --git a/proxy.go b/proxy.go
@@ -169,7 +169,7 @@ func main() {
 		}
 
 		log.Println("cert and key do not exist, generating")
-		generateCertificate(Config.Cert, Config.Key)
+		generateCertificate()
 	}
 
 	server := &http.Server{