croc/src/croc/croc.go

package croc

import (
	"runtime"
	"time"

	"github.com/schollz/croc/src/logger"
	"github.com/schollz/croc/src/models"
	"github.com/schollz/croc/src/relay"
	"github.com/schollz/croc/src/zipper"
	"github.com/schollz/progressbar/v2"
)

func init() {
	runtime.GOMAXPROCS(runtime.NumCPU())
}

// Croc options
type Croc struct {
	// Version is the version of croc
	Version string
	// Options for all
	Debug bool
	// ShowText will display text on the stderr
	ShowText bool

	// Options for relay
	RelayWebsocketPort string
	RelayTCPPorts      []string
	CurveType          string

	// Options for connecting to server
	Address              string
	AddressTCPPorts      []string
	AddressWebsocketPort string
	Timeout              time.Duration
	LocalOnly            bool
	NoLocal              bool

	// Options for file transfering
	UseEncryption       bool
	UseCompression      bool
	AllowLocalDiscovery bool
	NoRecipientPrompt   bool
	Stdout              bool
	ForceSend           int // 0: ignore, 1: websockets, 2: TCP

	// Parameters for file transfer
	Filename   string
	Codephrase string

	// localIP address
	localIP string
	// is using local relay
	isLocal      bool
	normalFinish bool

	// state variables
	StateString string
	Bar         *progressbar.ProgressBar
	FileInfo    models.FileStats
	OtherIP     string

	// special for window
	WindowRecipientPrompt bool
	WindowRecipientAccept bool
	WindowReceivingString string
}

// Init will initiate with the default parameters
func Init(debug bool) (c *Croc) {
	c = new(Croc)
	c.UseCompression = true
	c.UseEncryption = true
	c.AllowLocalDiscovery = true
	c.RelayWebsocketPort = "8153"
	c.RelayTCPPorts = []string{"8154", "8155", "8156", "8157", "8158", "8159", "8160", "8161"}
	c.CurveType = "siec"
	c.Address = "croc4.schollz.com"
	c.AddressWebsocketPort = "8153"
	c.AddressTCPPorts = []string{"8154", "8155", "8156", "8157", "8158", "8159", "8160", "8161"}
	c.NoRecipientPrompt = true
	debugLevel := "info"
	if debug {
		debugLevel = "debug"
		c.Debug = true
	}
	SetDebugLevel(debugLevel)
	return
}

func SetDebugLevel(debugLevel string) {
	logger.SetLogLevel(debugLevel)
	relay.DebugLevel = debugLevel
	zipper.DebugLevel = debugLevel
}