From f848f8ab14bfec84230e29ed4d71849bacdc89a1 Mon Sep 17 00:00:00 2001
From: netbenix <netbenix@codenoodles.de>
Date: Sat, 20 Nov 2021 15:39:24 +0100
Subject: [PATCH] Moved some stuff fom the Handler to the ProtoWrapper

---
 ProtoHandler/ProtoWrapper.go | 46 ++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 ProtoHandler/ProtoWrapper.go

diff --git a/ProtoHandler/ProtoWrapper.go b/ProtoHandler/ProtoWrapper.go
new file mode 100644
index 0000000..1096025
--- /dev/null
+++ b/ProtoHandler/ProtoWrapper.go
@@ -0,0 +1,46 @@
+package ProtoHandler
+
+import (
+	"golang.org/x/net/context"
+	"google.golang.org/grpc"
+	"lpm-cli/Lotus"
+	proto "lpm-cli/Proto/lpm/go"
+	"time"
+)
+
+type ProtoWrapper struct{}
+
+func (protoWrap ProtoWrapper) CreateConnection(address string) *grpc.ClientConn {
+	logger := Lotus.DetailedLogger("ProtoHandler", "createConnection")
+
+	conn, err := grpc.Dial(address, grpc.WithInsecure())
+	if err != nil {
+		logger.Fatal(err)
+	}
+	return conn
+}
+
+func (protoWrap ProtoWrapper) CreateRegistryClient(conn *grpc.ClientConn) (proto.LpmRegistryClient, context.Context, context.CancelFunc) {
+	lrc := proto.NewLpmRegistryClient(conn)
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second * 10)
+
+	return lrc, ctx, cancel
+}
+
+
+func (protoWrap ProtoWrapper) ParsePackages(res *proto.SearchResponse, reg Lotus.Registry) []Package {
+	var packages []Package
+	for _, pkg := range res.GetPackages() {
+		var tmpMeta PackageMetadata
+		tmpMeta.FillMetadata(pkg)
+		packages = append(packages, Package {
+			PackageMetadata: tmpMeta,
+			Register: Lotus.Registry{
+				Register: reg.Register,
+				Address: reg.Address,
+			},
+		})
+	}
+
+	return packages
+}
\ No newline at end of file