package main

import (
	"crypto/rsa"
	"encoding/gob"
	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
	"net"
	"os"
)

// Exchange keys with sender
func ReceiverKeyExchange(key *rsa.PublicKey) string {
	// Use ConsoleWriter logger
	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr}).Hook(FatalHook{})
	// Create TCP listener on port 9797
	listener, err := net.Listen("tcp", ":9797")
	if err != nil { log.Fatal().Err(err).Msg("Error starting listener") }
	// Create string for sender address
	var senderAddr string
	for {
		// Accept connection on listener
		connection, err := listener.Accept()
		if err != nil { log.Fatal().Err(err).Msg("Error accepting connections") }
		// Get sender address and store it in senderAddr
		senderAddr = connection.RemoteAddr().String()
		// Create gob encoder with connection as io.Writer
		encoder := gob.NewEncoder(connection)
		// Encode key into connection
		err = encoder.Encode(key)
		if err != nil { log.Fatal().Err(err).Msg("Error encoding key") }
		return senderAddr
	}
}

// Exchange keys with receiver
func SenderKeyExchange(receiverIP string) *rsa.PublicKey {
	// Use ConsoleWriter logger
	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr}).Hook(FatalHook{})
	// Connect to TCP socket on receiver IP port 9797
	connection, err := net.Dial("tcp", receiverIP + ":9797")
	if err != nil { log.Fatal().Err(err).Msg("Error connecting to sender") }
	// Create gob decoder
	decoder := gob.NewDecoder(connection)
	// Instantiate rsa.PublicKey struct
	recvPubKey := &rsa.PublicKey{}
	// Decode key
	err = decoder.Decode(recvPubKey)
	if err != nil { log.Fatal().Err(err).Msg("Error decoding key") }
	// Return received key
	return recvPubKey
}