zishuo/netmaker
1
0
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-11-02 13:04:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove grpc

Browse Source
This commit is contained in:
Matthew R. Kasun
2022-04-21 15:04:44 -04:00
parent 725037b794
commit 886ec31553
14 changed files with 291 additions and 1614 deletions
Download Patch File Download Diff File Expand all files Collapse all files

127
netclient/auth/auth.go
View File

@@ -1,84 +1,73 @@
package auth
import (
"encoding/json"
"fmt"
"os"
"github.com/gravitl/netmaker/models"
"github.com/gravitl/netmaker/netclient/config"
"github.com/gravitl/netmaker/netclient/ncutils"
// "os"
"context"
nodepb "github.com/gravitl/netmaker/grpc"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/metadata"
"google.golang.org/grpc/status"
)
// SetJWT func will used to create the JWT while signing in and signing out
func SetJWT(client nodepb.NodeServiceClient, network string) (context.Context, error) {
home := ncutils.GetNetclientPathSpecific()
tokentext, err := os.ReadFile(home + "nettoken-" + network)
if err != nil {
err = AutoLogin(client, network)
if err != nil {
return nil, status.Errorf(codes.Unauthenticated, fmt.Sprintf("Something went wrong with Auto Login: %v", err))
}
tokentext, err = ncutils.GetFileWithRetry(home+"nettoken-"+network, 1)
if err != nil {
return nil, status.Errorf(codes.Unauthenticated, fmt.Sprintf("Something went wrong: %v", err))
}
}
token := string(tokentext)
// Anything linked to this variable will transmit request headers.
md := metadata.New(map[string]string{"authorization": token})
ctx := context.Background()
ctx = metadata.NewOutgoingContext(ctx, md)
return ctx, nil
}
//func SetJWT(client nodepb.NodeServiceClient, network string) (context.Context, error) {
// home := ncutils.GetNetclientPathSpecific()
// tokentext, err := os.ReadFile(home + "nettoken-" + network)
// if err != nil {
// err = AutoLogin(client, network)
// if err != nil {
// return nil, status.Errorf(codes.Unauthenticated, fmt.Sprintf("Something went wrong with Auto Login: %v", err))
// }
// tokentext, err = ncutils.GetFileWithRetry(home+"nettoken-"+network, 1)
// if err != nil {
// return nil, status.Errorf(codes.Unauthenticated, fmt.Sprintf("Something went wrong: %v", err))
// }
// }
// token := string(tokentext)
//
// // Anything linked to this variable will transmit request headers.
// md := metadata.New(map[string]string{"authorization": token})
// ctx := context.Background()
// ctx = metadata.NewOutgoingContext(ctx, md)
// return ctx, nil
//}
// AutoLogin - auto logins whenever client needs to request from server
func AutoLogin(client nodepb.NodeServiceClient, network string) error {
home := ncutils.GetNetclientPathSpecific()
cfg, err := config.ReadConfig(network)
if err != nil {
return err
}
pass, err := RetrieveSecret(network)
if err != nil {
return err
}
node := models.Node{
Password: pass,
MacAddress: cfg.Node.MacAddress,
ID: cfg.Node.ID,
Network: network,
}
data, err := json.Marshal(&node)
if err != nil {
return nil
}
login := &nodepb.Object{
Data: string(data),
Type: nodepb.NODE_TYPE,
}
// RPC call
res, err := client.Login(context.TODO(), login)
if err != nil {
return err
}
tokenstring := []byte(res.Data)
err = os.WriteFile(home+"nettoken-"+network, tokenstring, 0600)
if err != nil {
return err
}
return err
}
//func AutoLogin(client nodepb.NodeServiceClient, network string) error {
// home := ncutils.GetNetclientPathSpecific()
// cfg, err := config.ReadConfig(network)
// if err != nil {
// return err
// }
// pass, err := RetrieveSecret(network)
// if err != nil {
// return err
// }
// node := models.Node{
// Password: pass,
// MacAddress: cfg.Node.MacAddress,
// ID: cfg.Node.ID,
// Network: network,
// }
// data, err := json.Marshal(&node)
// if err != nil {
// return nil
// }
//
// login := &nodepb.Object{
// Data: string(data),
// Type: nodepb.NODE_TYPE,
// }
// // RPC call
// res, err := client.Login(context.TODO(), login)
// if err != nil {
// return err
// }
// tokenstring := []byte(res.Data)
// err = os.WriteFile(home+"nettoken-"+network, tokenstring, 0600)
// if err != nil {
// return err
// }
// return err
//}
// StoreSecret - stores auth secret locally
func StoreSecret(key string, network string) error {