BAHAMAS

main.go

@@ -1,12 +1,18 @@
 package main
 
 import (
+	"errors"
 	"fmt"
 	"oc-auth/conf"
 	"oc-auth/infrastructure"
 	_ "oc-auth/routers"
+	"os"
+	"strconv"
+	"strings"
 
 	oclib "cloud.o-forge.io/core/oc-lib"
+	peer "cloud.o-forge.io/core/oc-lib/models/peer"
+	"cloud.o-forge.io/core/oc-lib/models/utils"
 	"cloud.o-forge.io/core/oc-lib/tools"
 	beego "github.com/beego/beego/v2/server/web"
 )
@@ -23,7 +29,8 @@ func main() {
 	// Load the right config file
 	o := oclib.GetConfLoader()
 
-	conf.GetConfig().PVKPath = o.GetStringDefault("PVK_PATH", "./pvk.pem")
+	conf.GetConfig().PublicKeyPath = o.GetStringDefault("PUBLIC_KEY_PATH", "./pem/public.pem")
+	conf.GetConfig().PrivateKeyPath = o.GetStringDefault("PRIVATE_KEY_PATH", "./pem/private.pem")
 	conf.GetConfig().ClientSecret = o.GetStringDefault("CLIENT_SECRET", "oc-auth-got-secret")
 
 	conf.GetConfig().Auth = o.GetStringDefault("AUTH", "hydra")
@@ -40,11 +47,57 @@ func main() {
 	conf.GetConfig().LDAPBindPW = o.GetStringDefault("LDAP_BINDPW", "password")
 	conf.GetConfig().LDAPBaseDN = o.GetStringDefault("LDAP_BASEDN", "dc=example,dc=com")
 	conf.GetConfig().LDAPRoleBaseDN = o.GetStringDefault("LDAP_ROLE_BASEDN", "ou=AppRoles,dc=example,dc=com")
-
-	Discovery()
+	err := generateSelfPeer()
+	if err != nil {
+		panic(err)
+	}
+	discovery()
 	beego.Run()
 }
-func Discovery() {
+
+func generateSelfPeer() error {
+	// TODO check if files at private & public path are set
+	// check if files at private & public path are set
+	if _, err := os.Stat(conf.GetConfig().PrivateKeyPath); errors.Is(err, os.ErrNotExist) {
+		return errors.New("private key path does not exist")
+	}
+	if _, err := os.Stat(conf.GetConfig().PublicKeyPath); errors.Is(err, os.ErrNotExist) {
+		return errors.New("public key path does not exist")
+	}
+	// check if peer already exists
+	p := oclib.Search(nil, strconv.Itoa(peer.SELF.EnumIndex()), oclib.LibDataEnum(oclib.PEER))
+	if len(p.Data) > 0 {
+		// check public key with the one in the database
+		f, err := os.ReadFile(conf.GetConfig().PublicKeyPath)
+		if err != nil {
+			return err
+		}
+		// compare the public key from file with the one in the database
+		fmt.Println(string(f), p.Data[0].(*peer.Peer).PublicKey)
+		if !strings.Contains(string(f), p.Data[0].(*peer.Peer).PublicKey) {
+			return errors.New("public key is different from the one in the database")
+		}
+		return nil
+	}
+	fmt.Println("Creating new peer", strconv.Itoa(peer.SELF.EnumIndex()))
+	// create a new peer
+	o := oclib.GetConfLoader()
+	peer := &peer.Peer{
+		Url: o.GetStringDefault("HOSTNAME", "http://localhost"),
+		AbstractObject: utils.AbstractObject{
+			Name: o.GetStringDefault("NAME", "local"),
+		},
+		PublicKey: conf.GetConfig().PublicKeyPath,
+		State:     peer.SELF,
+	}
+	data := oclib.StoreOne(oclib.LibDataEnum(oclib.PEER), peer.Serialize())
+	if data.Err != "" {
+		return errors.New(data.Err)
+	}
+	return nil
+}
+
+func discovery() {
 	fmt.Println("Discovered")
 	api := tools.API{}
 	addPermissions := func(m map[string]interface{}) {