Newer
Older
package main
import (
"flag"
"fmt"
"net"
"os"
"path/filepath"
pb "code.fbi.h-da.de/danet/gosdn/api/go/gosdn/plugin-registry"
"code.fbi.h-da.de/danet/gosdn/controller/interfaces/plugin"
"code.fbi.h-da.de/danet/gosdn/controller/nucleus/util"
"github.com/google/uuid"
"golang.org/x/mod/sumdb/dirhash"
"google.golang.org/grpc"
)
const (
pluginFilePath = "plugins"
pluginStorePath = "plugin-store.json"
)
func main() {
socket := flag.String("socket", "9000", "port for the grpc socket, e.g. 9000")
flag.Parse()
lislisten, err := net.Listen("tcp", fmt.Sprintf(":%s", *socket))
if err != nil {
panic(err)
}
pr := registerPlugins()
grpcServer := grpc.NewServer()
server := NewServer(pr)
pb.RegisterPluginRegistryServiceServer(grpcServer, server)
if err := grpcServer.Serve(lislisten); err != nil {
panic(err)
}
}
// TODO: The registration of plugins should result in the same UUID's even
// after reboot. Therefore it would make sense to use a database. For a simple
// prototype this is currently hardcoded.
func registerPlugins() *PluginRegistry {
dirs, err := os.ReadDir(pluginFilePath)
if err != nil {
panic(err)
}
pr := &PluginRegistry{}
pluginStore := NewStore(pluginStorePath)
for _, dir := range dirs {
fmt.Printf("File %+v\n", dir)
if dir.IsDir() {
dirPath := filepath.Join(pluginFilePath, dir.Name())
manifest, err := plugin.ReadManifestFromFile(dirPath)
if err != nil {
panic(err)
}
dirHashed, err := dirhash.HashDir(dirPath, dir.Name()+manifest.Firmware, dirhash.DefaultHash)
if err != nil {
panic(err)
}
//var id uuid.UUID
//switch i {
//case 0:
// id = uuid.MustParse("e2c358b3-6482-4010-b0d8-679dff73153b")
//case 1:
// id = uuid.MustParse("d1c269a2-6482-4010-b0d8-679dff73153b")
//case 2:
// id = uuid.MustParse("f3b474c2-6482-4010-b0d8-679dff73153b")
//default:
// break
//}
plugin := &Plugin{
ID: uuid.New(),
Path: filepath.Join(dirPath, util.BundledPluginName),
Hash: dirHashed,
Manifest: manifest,
}
pr.Plugins = append(pr.Plugins, plugin)
}
}
if err := pluginStore.Update(pr.Plugins); err != nil {
fmt.Println(err)
}
return pr
}