Source File
	types_relay.go

Belonging Package
	/tools/relay/types

package types

import (
	"context"
	"encoding/gob"
	"net"
	"regexp"
	"time"

	am "github.com/pancsta/asyncmachine-go/pkg/machine"
	arpc "github.com/pancsta/asyncmachine-go/pkg/rpc"
)

const CmdRotateDbg = "rotate-dbg"

type OutputFunc func(msg string, args ...any)

// nolint:lll
type CliArgs struct {
	RotateDbg *ArgsRotateDbg `arg:"subcommand:rotate-dbg" help:"Rotate dbg protocol with fragmented dump files"`
	Wasm      *ArgsWasm      `arg:"subcommand:wasm" help:"WebSockets to local TCP listeners for WASM"`
	Debug     bool           `arg:"--debug" help:"Enable debugging for asyncmachine"`
	Name      string         `arg:"-n,--name" help:"Name of this relay"`
	Output    OutputFunc     `arg:"-"`
	Parent    am.Api         `arg:"-"`
	Version   bool           `arg:"-v,--version" help:"Print version and exit"`
}

// ArgsRotateDbg converts dbg dumps to other formats / versions.
// am-relay convert-dbg am-dbg-dump.gob.br -o am-vis
// nolint:lll
type ArgsRotateDbg struct {
	ListenAddr   string        `arg:"-l,--listen-addr" default:"localhost:12732" help:"Listen address for the RPC server"`
	FwdAddr      []string      `arg:"-f,--fwd-addr,separate" help:"Address of an RPC server to forward data to (repeatable)"`
	IntervalTx   int           `arg:"--interval-tx" default:"10000" help:"Amount of transitions to create a dump file"`
	IntervalTime time.Duration `arg:"--interval-duration" default:"24h" help:"Amount of human time to create a dump file"`
	Filename      string        `arg:"-o,--output" default:"am-dbg-dump" help:"Output file base name"`
	Dir           string        `arg:"-d,--dir" default:"." help:"Output directory"`
	OutputMach    bool          `arg:"--output-mach" default:"true" help:"Export serialized machines as YAML into dir/machs/id.yml and dir/machs/id.schema.yml"`
	OutputClients bool          `arg:"--output-clients" help:"Write a plain text client list into clients.txt inside --dir"`
}

// ArgsWasm
// nolint:lll
type ArgsWasm struct {
	ListenAddr  string `arg:"-l,--listen-addr" default:"localhost:12733" help:"Listen address for HTTP server"`
	StaticDir   string `arg:"-s,--static-dir" help:"Directory with static files to serve (optional)"`
	ReplAddrDir string `arg:"-r,--repl-addr-dir" help:"Directory for creating tunneled REPLs' addr files (optional)"`
	// Match incoming tunnels by mach IDs and pass directly to new RPC clients
	TunnelMatchers []TunnelMatcher `arg:"-"`
	// Match incoming TCP dials by mach IDs and pass directly to new RPC servers
	DialMatchers []DialMatcher `arg:"-"`
}

type DialMatcher struct {
	Id        *regexp.Regexp
	NewServer NewServerFunc
}

type NewServerFunc func(
	ctx context.Context, id string, conn net.Conn,
) (*arpc.Server, error)

type TunnelMatcher struct {
	Id        *regexp.Regexp
	NewClient NewClientFunc
}

type NewClientFunc func(
	ctx context.Context, id string, conn net.Conn,
) (*arpc.Client, error)

// ///// ///// /////

// ///// ARGS

// ///// ///// /////

const APrefix = "relay"

type Args struct {
	am.ArgsBase `json:"-"`
}

func (Args) ArgsPrefix() string {
	return APrefix
}

// ----- TODO per-state args

// A is a struct for node arguments. It's a typesafe alternative to [am.A].
type A struct {
	Args `json:"-"`

	Id         string `log:"id"`
	Addr       string `log:"addr"`
	RemoteAddr string `log:"remote_addr"`

	// non-rpc fields

	Conn net.Conn
}

// ARpc is a subset of A, that can be passed over RPC.
type ARpc struct {
	Args `json:"-"`

	Id   string `log:"id"`
	Addr string `log:"addr"`
}

// ----- RPC boilerplate

func init() {
	for _, arg := range ArgsRpc {
		gob.Register(arg)
	}
}

// ArgsRpc will be available in the REPL.
var ArgsRpc = []am.ArgsApi{ARpc{}}

The pages are generated with Golds v0.8.4. (GOOS=linux GOARCH=amd64)
Golds is a Go 101 project developed by Tapir Liu.
PR and bug reports are welcome and can be submitted to the issue list.
Please follow @zigo_101 (reachable from the left QR code) to get the latest news of Golds.