Merge pull request #50 from ShyftNetwork/shyft-ring-wallet-connectivity-v0.01

Shyft ring wallet connectivity v0.01

shyftRingWalletConn/app.go

@@ -3,24 +3,31 @@ package main
 //@NOTE SHYFT main func for api, sets up router and spins up a server
 //to run server 'go run shyftRingWalletConn/*.go'
 import (
-  "net"
+	"bufio"
 	"fmt"
+	"github.com/ShyftNetwork/go-empyrean/common/hexutil"
+	"github.com/ShyftNetwork/go-empyrean/crypto"
+	"io"
+	"net"
+	"net/http"
 	"os"
-	//"github.com/ethereum/go-ethereum/common"
-	"github.com/ethereum/go-ethereum/crypto"
-	"bytes"
-	"github.com/ethereum/go-ethereum/common/hexutil"
+	"github.com/ShyftNetwork/go-empyrean/ethclient"
+	"github.com/ShyftNetwork/go-empyrean/common"
+	"context"
 )
 
 const (
 	CONN_HOST     = "localhost"
 	CONN_PORT     = "3333"
 	CONN_TYPE     = "tcp"
+	NEW_LINE_BYTE = 0x0a
 )
 
 var testAddrHex = "14791697260E4c9A71f18484C9f997B308e59325"
 var testPrivHex = "0123456789012345678901234567890123456789012345678901234567890123"
 
+var client = &http.Client{}
+
 // This gives context to the signed message and prevents signing of transactions.
 func signHash(data []byte) []byte {
 	msg := fmt.Sprintf("\x19Ethereum Signed Message:\n%d%s", len(data), data)
@@ -28,7 +35,6 @@ func signHash(data []byte) []byte {
 }
 
 func main() {
-
 	l, err := net.Listen(CONN_TYPE, CONN_HOST+":"+CONN_PORT)
 	if err != nil {
 		fmt.Println("Error listening:", err.Error())
@@ -50,44 +56,42 @@ func main() {
 
 // Handles incoming requests.
 func handleRequest(conn net.Conn) {
-	// Make a buffer to hold incoming data.
-	// Read the incoming connection into the buffer.
 
-	go func() {
-		buf := make([]byte, 1024)
-		msgBuf := make([]byte, 0)
+	messages := make(chan []byte)
+	checkBalanceChan := make(chan []byte)
+
+	go readerConn(conn, messages)
+	go handleMessages(messages, checkBalanceChan)
+	go checkBalance(checkBalanceChan, conn)
+
+	sendRingSignedMsg(conn)
+}
+
+func handleMessages(channel chan []byte, checkBalancesChan chan []byte) {
 	var prevMsg []byte
 	var addressOfClient []byte
 	var signatureFromClient []byte
 	var msgFromClient []byte
 
 	for {
-			msg, err := conn.Read(buf)
+		msg := <-channel
 
-			if err == nil {
-				msgBuf = append(msgBuf, buf[:msg]...)
-			}
-			index := bytes.IndexByte(msgBuf, 0x0a)
-			for index != -1 {
-				newMsg := msgBuf[:index]
-				rest := msgBuf[(index + 1):len(msgBuf)]
-				msgBuf = rest
-				index = bytes.IndexByte(msgBuf, 0x0a)
+		//similar to shift in bash
 		if prevMsg != nil {
 			s := string(prevMsg[:])
 			if s == "-- ADDRESS --" {
-						addressOfClient = newMsg
+				addressOfClient = msg
+				checkBalancesChan <- addressOfClient
 			}
 			if s == "-- SIGNATURE --" {
-						signatureFromClient = newMsg
+				signatureFromClient = msg
 			}
 			if s == "-- MESSAGE --" {
-						msgFromClient = newMsg
+				msgFromClient = msg
 			}
 			prevMsg = nil
 		} else {
-					prevMsg = newMsg
-				}
+			prevMsg = msg
 		}
 
 		if addressOfClient != nil && signatureFromClient != nil && msgFromClient != nil {
@@ -113,22 +117,62 @@ func handleRequest(conn net.Conn) {
 			fmt.Println("ADDRESS IS ::", recoveredAddr.Hex())
 		}
 	}
-	}()
-	go func() {
+}
+
+func readerConn(conn net.Conn, channel chan []byte) {
+	bufReader := bufio.NewReader(conn)
+
+	for {
+		msg, err := bufReader.ReadBytes(NEW_LINE_BYTE)
+
+		if err == io.EOF {
+			fmt.Println("END OF FILE, CLOSING CONNECTION")
+			conn.Close()
+			conn = nil
+			break
+		}
+		if err != nil {
+			fmt.Println("Connection error: ", err)
+			break
+		}
+
+		msg = msg[:len(msg)-1] // remove trailing new line byte
+
+		channel <- msg
+	}
+}
+
+func checkBalance(checkBalanceChan chan []byte, conn net.Conn) {
+	address := <-checkBalanceChan
+	c, err := ethclient.Dial("http://127.0.0.1:8545")
+	if err != nil {
+		fmt.Println("Eth Client not initialized: " , err)
+	}
+
+	balance, error := c.BalanceAt(context.Background(), common.HexToAddress(string(address[:])),nil)
+	if error != nil {
+		fmt.Println("Balance at error ", error)
+	}
+	fmt.Println("the bal is ", balance)
+	fmt.Println("The balance for address ", address, " is ", balance)
+	conn.Write([]byte("Broadcasting Balance"))
+	conn.Write([]byte("\n"))
+	conn.Write([]byte(balance.String()))
+	conn.Write([]byte("\n"))
+}
+
+func sendRingSignedMsg(conn net.Conn){
 	key, _ := crypto.HexToECDSA(testPrivHex)
 
 	f_msg := "Hello World"
 	first_message := []byte(f_msg)
 	new_msg2 := crypto.Keccak256(first_message)
-		fmt.Println("HASH IS ::", hexutil.Encode(new_msg2))
 
 	//send_message := append(new_msg2, []byte{byte(10)}...)
-		new_sig , err := crypto.Sign(new_msg2, key)
+	new_sig, err := crypto.Sign(new_msg2, key)
 	if err != nil {
 		fmt.Println("The crypto.Sign err is ", err)
 	}
-		hex_sig := hexutil.Encode(new_sig)
-		fmt.Println("HEX SIG ::", hex_sig)
 
 	conn.Write([]byte("Broadcasting Message"))
 	conn.Write([]byte("\n"))
@@ -136,5 +180,4 @@ func handleRequest(conn net.Conn) {
 	conn.Write([]byte("\n"))
 	conn.Write(new_sig)
 	conn.Write([]byte("\n"))
-	}()
 }