// Copyright 2016 The go-ethereum Authors
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.

package adapters

import (
	"errors"
	"fmt"
	"math"
	"net"
	"sync"

	"github.com/ethereum/go-ethereum/event"
	"github.com/ethereum/go-ethereum/log"
	"github.com/ethereum/go-ethereum/node"
	"github.com/ethereum/go-ethereum/p2p"
	"github.com/ethereum/go-ethereum/p2p/discover"
	"github.com/ethereum/go-ethereum/rpc"
)

// SimAdapter is a NodeAdapter which creates in-memory nodes and connects them
// using an in-memory p2p.MsgReadWriter pipe
type SimAdapter struct {
	mtx      sync.RWMutex
	nodes    map[discover.NodeID]*SimNode
	services map[string]ServiceFunc
}

// NewSimAdapter creates a SimAdapter which is capable of running in-memory
// nodes running any of the given services (the service to run on a particular
// node is passed to the NewNode function in the NodeConfig)
func NewSimAdapter(services map[string]ServiceFunc) *SimAdapter {
	return &SimAdapter{
		nodes:    make(map[discover.NodeID]*SimNode),
		services: services,
	}
}

// Name returns the name of the adapter for logging purpoeses
func (s *SimAdapter) Name() string {
	return "sim-adapter"
}

// NewNode returns a new SimNode using the given config
func (s *SimAdapter) NewNode(config *NodeConfig) (Node, error) {
	var nodeprotos []p2p.Protocol
	
	s.mtx.Lock()
	defer s.mtx.Unlock()

	// check a node with the ID doesn't already exist
	id := config.Id
	if _, exists := s.nodes[id.NodeID]; exists {
		return nil, fmt.Errorf("node already exists: %s", id)
	}

	// check the service is valid and initialize it
/*
	serviceFunc, exists := s.services[config.Service]
	if !exists {
		return nil, fmt.Errorf("unknown node service %q", config.Service)
	}

	node := &SimNode{
		Id:          id,
		config:      config,
		adapter:     s,
		serviceFunc: serviceFunc,
*/
	//serviceFunc, exists := s.services[config.Service]
	
	//if !exists {
	//	return nil, fmt.Errorf("unknown node service %q", config.Service)
	//}
	//service := serviceFunc(id)
	
	_, err := node.New(&node.Config{
		P2P: p2p.Config{
			PrivateKey:      config.PrivateKey,
			MaxPeers:        math.MaxInt32,
			NoDiscovery:     true,
			Protocols:       nodeprotos,
			Dialer:          s,
			EnableMsgEvents: true,
		},
	})
	if err != nil {
		return nil, err
	}
	
	servicefuncs := make(map[string]ServiceFunc)
	
	for name, servicefunc := range s.services {
		service := servicefunc(id, nil)
		/*if err := n.Register(func(ctx *node.ServiceContext) (node.Service, error) {
			return service, err
		}); err != nil {
			return nil, err
		}*/
		for _, proto := range service.Protocols() {
			nodeprotos = append(nodeprotos, proto)
		}
		servicefuncs[name] = servicefunc
	}
	
	simnode := &SimNode{
		//node:	n,
		Id:      id,
		serviceFuncs: servicefuncs,
		adapter:     s,
		config:      config,
		running: 	make(map[string]node.Service),
	}
	s.nodes[id.NodeID] = simnode
	return simnode, nil
}

func (s *SimAdapter) Dial(dest *discover.Node) (conn net.Conn, err error) {
	node, ok := s.GetNode(dest.ID)
	if !ok {
		return nil, fmt.Errorf("unknown node: %s", dest.ID)
	}
	srv := node.Server()
	if srv == nil {
		return nil, fmt.Errorf("node not running: %s", dest.ID)
	}
	pipe1, pipe2 := net.Pipe()
	go srv.SetupConn(pipe1, 0, nil)
	return pipe2, nil
}

// GetNode returns the node with the given ID if it exists
func (s *SimAdapter) GetNode(id discover.NodeID) (*SimNode, bool) {
	s.mtx.RLock()
	defer s.mtx.RUnlock()
	node, ok := s.nodes[id]
	return node, ok
}

// SimNode is an in-memory node which connects to other SimNodes using an
// in-memory p2p.MsgReadWriter pipe, running an underlying service protocol
// directly over that pipe.
//
// It implements the p2p.Server interface so it can be used transparently
// by the underlying service.
type SimNode struct {
	lock        sync.RWMutex
	Id          *NodeId
	config      *NodeConfig
	adapter     *SimAdapter
	serviceFuncs 	map[string]ServiceFunc
	node        *node.Node
	running     map[string]node.Service
	client      *rpc.Client
	rpcMux      *rpcMux
}

// Addr returns the node's discovery address
func (self *SimNode) Addr() []byte {
	return []byte(self.Node().String())
}

// Node returns a discover.Node representing the SimNode
func (self *SimNode) Node() *discover.Node {
	return discover.NewNode(self.Id.NodeID, net.IP{127, 0, 0, 1}, 30303, 30303)
}

// Client returns an rpc.Client which can be used to communicate with the
// underlying service (it is set once the node has started)
func (self *SimNode) Client() (*rpc.Client, error) {
	self.lock.Lock()
	defer self.lock.Unlock()
	if self.client == nil {
		return nil, errors.New("RPC not started")
	}
	return self.client, nil
}

// ServeRPC serves RPC requests over the given connection using the node's
// RPC multiplexer
func (self *SimNode) ServeRPC(conn net.Conn) error {
	self.lock.Lock()
	mux := self.rpcMux
	self.lock.Unlock()
	if mux == nil {
		return errors.New("RPC not started")
	}
	mux.Serve(conn)
	return nil
}

// Snapshot creates a snapshot of the service state by calling the
// simulation_snapshot RPC method
func (self *SimNode) Snapshot() ([]byte, error) {
	self.lock.Lock()
	defer self.lock.Unlock()
	if self.client == nil {
		return nil, errors.New("RPC not started")
	}
	var snapshot []byte
	return snapshot, self.client.Call(&snapshot, "simulation_snapshot")
}

// Start starts the RPC handler and the underlying service
func (self *SimNode) Start(snapshot []byte) error {
	self.lock.Lock()
	defer self.lock.Unlock()
	if self.node != nil {
		return errors.New("node already started")
	}
	
	services := []node.ServiceConstructor{}
	
	
	// so we can control the order of the services if we need
	for _, name := range self.config.Services {
		service := self.serviceFuncs[name](self.Id, snapshot)
		services = append(services, func(ctx *node.ServiceContext) (node.Service, error) {
			self.running[name] = service
			return service, nil
		})
	}

	node, err := node.New(&node.Config{
		P2P: p2p.Config{
			PrivateKey:      self.config.PrivateKey,
			MaxPeers:        math.MaxInt32,
			NoDiscovery:     true,
			Dialer:          self.adapter,
			EnableMsgEvents: false,
		},
		NoUSB: true,
	})
	if err != nil {
		return err
	}
	
	for _, service := range services {
		log.Debug("registering service", "service", service)
		if err := node.Register(service); err != nil {
			return err
		}
	}

	if err := node.Start(); err != nil {
		return err
	}

	handler, err := node.RPCHandler()
	if err != nil {
		return err
	}

	// create an in-process RPC multiplexer
	pipe1, pipe2 := net.Pipe()
	go handler.ServeCodec(rpc.NewJSONCodec(pipe1), rpc.OptionMethodInvocation|rpc.OptionSubscriptions)
	self.rpcMux = newRPCMux(pipe2)

	// create an in-process RPC client
	self.client = self.rpcMux.Client()

	self.node = node

	return nil
}

func (self *SimNode) Stop() error {
	self.lock.Lock()
	defer self.lock.Unlock()
	if self.node == nil {
		return nil
	}
	if err := self.node.Stop(); err != nil {
		return err
	}
	self.node = nil
	return nil
}

func (self *SimNode) Server() *p2p.Server {
	self.lock.Lock()
	defer self.lock.Unlock()
	if self.node == nil {
		return nil
	}
	return self.node.Server()
}

// Service returns a underlying node.Service of the speficied type
func (self *SimNode) GetService(servicename string) node.Service {
	log.Warn("retrieving service", "name", servicename)
	return self.running[servicename]
}

func (self *SimNode) SubscribeEvents(ch chan *p2p.PeerEvent) event.Subscription {
	srv := self.Server()
	if srv == nil {
		panic("node not running")
	}
	return srv.SubscribeEvents(ch)
}

func (self *SimNode) NodeInfo() *p2p.NodeInfo {
	server := self.Server()
	if server == nil {
		return &p2p.NodeInfo{
			ID:    self.Id.String(),
			Enode: self.Node().String(),
		}
	}
	return server.NodeInfo()
}