derohe-proxy/derohe-proxy.go

package main

import (
	"derohe-proxy/config"
	"derohe-proxy/proxy"
	"fmt"
	"net"
	"strconv"
	"sync"
	"time"

	"github.com/deroproject/derohe/globals"
	"github.com/docopt/docopt-go"
)

func main() {
	var err error
	var rwmutex sync.RWMutex

	config.Arguments, err = docopt.Parse(config.Command_line, nil, true, "pre-alpha", false)

	if err != nil {
		return
	}

	if config.Arguments["--listen-address"] != nil {
		addr, err := net.ResolveTCPAddr("tcp", config.Arguments["--listen-address"].(string))
		if err != nil {
			return
		} else {
			if addr.Port == 0 {
				return
			} else {
				config.Listen_addr = addr.String()
			}
		}
	}

	if config.Arguments["--daemon-address"] == nil {
		return
	} else {
		config.Daemon_address = config.Arguments["--daemon-address"].(string)
	}

	if config.Arguments["--wallet-address"] != nil {
		addr, err := globals.ParseValidateAddress(config.Arguments["--wallet-address"].(string))
		if err != nil {
			fmt.Printf("%v Wallet address is invalid!\n", time.Now().Format(time.Stamp))
		}
		config.WalletAddr = addr.String()
		fmt.Printf("%v Using wallet %s for all connections\n", time.Now().Format(time.Stamp), config.WalletAddr)
	}

	if config.Arguments["--log-interval"] != nil {
		interval, err := strconv.ParseInt(config.Arguments["--log-interval"].(string), 10, 32)
		if err != nil {
			return
		} else {
			if interval < 60 || interval > 3600 {
				config.Log_intervall = 60
			} else {
				config.Log_intervall = int(interval)
			}
		}
	}

	if config.Arguments["--minimal"].(bool) {
		config.Minimal = true
		fmt.Printf("%v Forward only 2 jobs per block\n", time.Now().Format(time.Stamp))
	}

	if config.Arguments["--nonce"].(bool) {
		config.Nonce = true
		fmt.Printf("%v Nonce editing is enabled\n", time.Now().Format(time.Stamp))
	}

	if config.Arguments["--pool"].(bool) {
		config.Pool_mode = true
		config.Minimal = false
		fmt.Printf("%v Pool mode is enabled\n", time.Now().Format(time.Stamp))
	}

	fmt.Printf("%v Logging every %d seconds\n", time.Now().Format(time.Stamp), config.Log_intervall)

	go proxy.Start_server()

	// Wait for first miner connection to grab wallet address
	for proxy.CountMiners() < 1 {
		time.Sleep(time.Second * 1)
	}
	go proxy.Start_client(proxy.Address)
	//go proxy.SendUpdateToDaemon()

	for {
		time.Sleep(time.Second * time.Duration(config.Log_intervall))

		hash_rate_string := ""

		switch {
		case proxy.Hashrate > 1000000000000:
			hash_rate_string = fmt.Sprintf("%.3f TH/s", float64(proxy.Hashrate)/1000000000000.0)
		case proxy.Hashrate > 1000000000:
			hash_rate_string = fmt.Sprintf("%.3f GH/s", float64(proxy.Hashrate)/1000000000.0)
		case proxy.Hashrate > 1000000:
			hash_rate_string = fmt.Sprintf("%.3f MH/s", float64(proxy.Hashrate)/1000000.0)
		case proxy.Hashrate > 1000:
			hash_rate_string = fmt.Sprintf("%.3f KH/s", float64(proxy.Hashrate)/1000.0)
		case proxy.Hashrate > 0:
			hash_rate_string = fmt.Sprintf("%d H/s", int(proxy.Hashrate))
		}

		if !config.Pool_mode {
			fmt.Printf("%v %d miners connected, IB:%d MB:%d MBR:%d MBO:%d - MINING @ %s\n", time.Now().Format(time.Stamp), proxy.CountMiners(), proxy.Blocks, proxy.Minis, proxy.Rejected, proxy.Orphans, hash_rate_string)
		} else {
			fmt.Printf("%v %d miners connected, Pool stats: IB:%d MB:%d MBR:%d MBO:%d - MINING @ %s\n", time.Now().Format(time.Stamp), proxy.CountMiners(), proxy.Blocks, proxy.Minis, proxy.Rejected, proxy.Orphans, hash_rate_string)
		}
		rwmutex.RLock()
		for i := range proxy.Wallet_count {
			if proxy.Wallet_count[i] > 1 {
				fmt.Printf("%v Wallet %v, %d miners\n", time.Now().Format(time.Stamp), i, proxy.Wallet_count[i])
			}
		}
		rwmutex.RUnlock()
	}
}