// Copyright 2017-2021 DERO Project. All rights reserved.
// Use of this source code in any form is governed by RESEARCH license.
// license can be found in the LICENSE file.
// GPG: 0F39 E425 8C65 3947 702A  8234 08B2 0360 A03A 9DE8
//
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
// THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
// STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
// THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

package p2p

import "fmt"
import "bytes"
import "math/big"

import "sync/atomic"
import "time"

import "github.com/romana/rlog"
import "github.com/paulbellamy/ratecounter"

import "github.com/deroproject/derohe/config"
import "github.com/deroproject/derohe/globals"

// verify incoming handshake for number of checks such as mainnet/testnet etc etc
func Verify_Handshake(handshake *Handshake_Struct) bool {
	return bytes.Equal(handshake.Network_ID[:], globals.Config.Network_ID[:])
}

func (handshake *Handshake_Struct) Fill() {
	fill_common(&handshake.Common) // fill common info

	handshake.ProtocolVersion = "1.0.0"
	handshake.DaemonVersion = config.Version.String()
	handshake.Tag = node_tag
	handshake.UTC_Time = int64(time.Now().UTC().Unix()) // send our UTC time
	handshake.Local_Port = uint32(P2P_Port)             // export requested or default port
	handshake.Peer_ID = GetPeerID()                     // give our randomly generated peer id
	handshake.Pruned = chain.LocatePruneTopo()

	//	handshake.Flags = // add any flags necessary

	//scan our peer list and send peers which have been recently communicated
	handshake.PeerList = get_peer_list()
	copy(handshake.Network_ID[:], globals.Config.Network_ID[:])
}

// this is used only once
// all clients start with handshake, then other party sends avtive to mark that connection is active
func (connection *Connection) dispatch_test_handshake() {
	var request, response Handshake_Struct
	request.Fill()

	if err := connection.RConn.Client.Call("Peer.Handshake", request, &response); err != nil {
		connection.exit()
		return
	}

	if !Verify_Handshake(&response) { // if not same network boot off
		//fmt.Printf("kill connection network id mismatch peer network id1 %x", response.Network_ID)
		connection.exit()
		return
	}

	connection.CDIFF.Store(new(big.Int).SetUint64(1))

	connection.request_time.Store(time.Now())
	connection.SpeedIn = ratecounter.NewRateCounter(60 * time.Second)
	connection.SpeedOut = ratecounter.NewRateCounter(60 * time.Second)

	connection.update(&response.Common) // update common information

	if !connection.Incoming { // setup success
		Peer_SetSuccess(connection.Addr.String())
	}

	if len(response.ProtocolVersion) < 128 {
		connection.ProtocolVersion = response.ProtocolVersion
	}

	if len(response.DaemonVersion) < 128 {
		connection.DaemonVersion = response.DaemonVersion
	}
	connection.Port = response.Local_Port
	connection.Peer_ID = response.Peer_ID
	if len(response.Tag) < 128 {
		connection.Tag = response.Tag
	}
	if response.Pruned >= 1 {
		connection.Pruned = response.Pruned
	}

	// TODO we must also add the peer to our list
	// which can be distributed to other peers
	if connection.Port != 0 && connection.Port <= 65535 { // peer is saying it has an open port, handshake is success so add peer

		var p Peer
		if connection.Addr.IP.To4() != nil { // if ipv4
			p.Address = fmt.Sprintf("%s:%d", connection.Addr.IP.String(), connection.Port)
		} else { // if ipv6
			p.Address = fmt.Sprintf("[%s]:%d", connection.Addr.IP.String(), connection.Port)
		}
		p.ID = connection.Peer_ID

		p.LastConnected = uint64(time.Now().UTC().Unix())

		// TODO we should add any flags here if necessary, but they are not
		//  required, since a peer can only be used if connected and if connected
		//  we already have a truly synced view
		for _, k := range response.Flags {
			switch k {
			//case FLAG_MINER:p.Miner = true
			default:
			}
		}
		Peer_Add(&p)
	}

	connection.TXpool_cache = map[uint64]uint32{}

	// parse delivered peer list as grey list
	rlog.Debugf("Peer provides %d peers", len(response.PeerList))
	for i := range response.PeerList {
		if i < 13 {
			Peer_Add(&Peer{Address: response.PeerList[i].Addr, LastConnected: uint64(time.Now().UTC().Unix())})
		}
	}

	Connection_Add(connection) // add connection to pool

	// mark active
	var r Dummy
	fill_common(&r.Common) // fill common info
	if err := connection.RConn.Client.Call("Peer.Active", r, &r); err != nil {
		connection.exit()
		return
	}

}

// mark connection active
func (c *Connection) Active(req Dummy, dummy *Dummy) error {
	c.update(&req.Common) // update common information
	atomic.StoreUint32(&c.State, ACTIVE)
	fill_common(&dummy.Common) // fill common info
	return nil
}

// used to ping pong
func (c *Connection) Ping(req Dummy, resp *Dummy) error {
	c.update(&req.Common)     // update common information
	fill_common(&resp.Common) // fill common info
	return nil
}

// serves handhake requests
func (c *Connection) Handshake(request Handshake_Struct, response *Handshake_Struct) error {

	if request.Peer_ID == GetPeerID() { // check if self connection exit
		rlog.Tracef(1, "Same peer ID, probably self connection, disconnecting from this client")
		c.exit()
		return fmt.Errorf("Same peer ID")
	}

	if !Verify_Handshake(&request) { // if not same network boot off
		rlog.Tracef(1, "kill connection network id mismatch peer network id0 %x", request.Network_ID)
		c.exit()
		return fmt.Errorf("NID mismatch")
	}

	response.Fill()

	c.update(&request.Common) // update common information
	if c.State == ACTIVE {
		for i := range request.PeerList {
			if i < 13 {
				Peer_Add(&Peer{Address: request.PeerList[i].Addr, LastConnected: uint64(time.Now().UTC().Unix())})
			}
		}
	}

	return nil
}