main.go

// Copyright (c) 2018 CommerceBlock Team
// Use of this source code is governed by an MIT
// license that can be found in the LICENSE file.

// Package main implements attestation and request services.
package main

import (
	"context"
	"flag"
	"os"
	"os/signal"
	"strings"
	"sync"

	"mainstay/attestation"
	"mainstay/config"
	"mainstay/db"
	"mainstay/log"
	"mainstay/test"
)

var (
	tx0         string
	chaincode  string
	addrTopup   string
	isRegtest   bool
	mainConfig  *config.Config
)

func parseFlags() {
	flag.BoolVar(&isRegtest, "regtest", false, "Use regtest wallet configuration instead of user wallet")
	flag.StringVar(&tx0, "tx", "", "Tx id for genesis attestation transaction")
	flag.StringVar(&chaincode, "chaincode", "", "Chaincode for sig pubkey")
	flag.StringVar(&addrTopup, "addrTopup", "", "Address for topup transaction")
	flag.Parse()
}

func init() {
	parseFlags()

	if isRegtest {
		test := test.NewTest(true, true)
		mainConfig = test.Config
		log.Infof("Running regtest mode with -tx=%s\n", mainConfig.InitTx())
	} else {
		var mainConfigErr error
		mainConfig, mainConfigErr = config.NewConfig()
		if mainConfigErr != nil {
			log.Error(mainConfigErr)
		}

		// if either tx or script not set throw error
		if tx0 == "" || chaincode == "" {
			if mainConfig.InitTx() == "" || mainConfig.InitChaincode() == "" {
				flag.PrintDefaults()
				log.Error(`Need to provide all -tx, -script and -chaincode arguments.
                    To use test configuration set the -regtest flag.`)
			}
		} else {
			mainConfig.SetInitTx(tx0)

			chaincodeStr := strings.TrimSpace(chaincode) // trim whitespace
			mainConfig.SetInitChaincode(chaincodeStr)
		}
		if addrTopup != "" {
			mainConfig.SetTopupAddress(addrTopup)
		}
		mainConfig.SetRegtest(isRegtest)
	}
}

func main() {
	defer mainConfig.MainClient().Shutdown()

	wg := &sync.WaitGroup{}
	ctx, cancel := context.WithCancel(context.Background())

	dbInterface := db.NewDbMongo(ctx, mainConfig.DbConfig())
	server := attestation.NewAttestServer(dbInterface)
	signer := attestation.NewAttestSignerHttp(mainConfig.SignerConfig())
	attestService := attestation.NewAttestService(ctx, wg, server, signer, mainConfig)

	c := make(chan os.Signal)
	signal.Notify(c, os.Interrupt)

	wg.Add(1)
	go func() {
		defer cancel()
		defer wg.Done()
		select {
		case sig := <-c:
			log.Warnf("Got %s signal. Aborting...\n", sig)
		case <-ctx.Done():
			signal.Stop(c)
		}
	}()

	wg.Add(1)
	go attestService.Run()

	// In regtest demo mode do block generation work
	// Also auto commitment to ClientCommitment to
	// allow easier testing without db intervention
	if isRegtest {
		wg.Add(1)
		go test.DoRegtestWork(dbInterface, mainConfig, wg, ctx)
	}
	wg.Wait()
}