run_socket.go

// Copyright 2019 The Nym Mixnet Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package commands

import (
	"fmt"
	"net"
	"os"

	"github.com/nymtech/nym-mixnet/client"
	clientConfig "github.com/nymtech/nym-mixnet/client/config"
	server "github.com/nymtech/nym-mixnet/client/rpc"
	"github.com/nymtech/nym-mixnet/helpers"
	"github.com/nymtech/nym-mixnet/logger"
)

const (
	localAddress = "127.0.0.1" // TODO: possibly allow to override this value with a flag?
)

//nolint: lll
func RunSocketCmd(args []string, usage string) {
	opts := newOpts("run [OPTIONS]", usage)
	id := opts.Flags("--id").Label("ID").String("Id of the nym-mixnet-client we want to run", defaultID)
	customConfigPath := opts.Flags("--customCfg").Label("CUSTOMCFG").String("Path to custom configuration file of the client", "")
	socketType := opts.Flags("--socket").Label("SOCKETTYPE").String("Type of the socket we want to run on (tcp / websocket)")
	port := opts.Flags("--port").Label("PORT").String("Port to listen on")
	params := opts.Parse(args)
	if len(params) != 0 {
		opts.PrintUsage()
		os.Exit(1)
	}

	var configPath string
	var err error
	if len(*customConfigPath) > 0 {
		configPath = *customConfigPath
	} else {
		configPath, err = clientConfig.DefaultConfigPath(*id)
		if err != nil {
			panic(err)
		}
	}

	cfgExists, err := helpers.DirExists(configPath)
	if !cfgExists || err != nil {
		fmt.Fprintf(os.Stderr, "The configuration file at %v does not seem to exist\n", configPath)
		os.Exit(1)
	}

	cfg, err := clientConfig.LoadFile(configPath)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Could not load the config file: %v\n", err)
		os.Exit(1)
	}

	client, err := client.NewClient(cfg)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Failed to spawn client instance: %v\n", err)
		os.Exit(-1)
	}

	// TODO: a better approach to that, but to be honest, we need to rewrite client anyway...
	socketLogger, err := logger.New(cfg.Logging.File, cfg.Logging.Level, cfg.Logging.Disable)

	socketListener, err := server.NewSocketListener(net.JoinHostPort(localAddress, *port), *socketType, socketLogger, client)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Failed to spawn socket listener instance: %v\n", err)
		os.Exit(-1)
	}

	if err := socketListener.Start(); err != nil {
		fmt.Fprintf(os.Stderr, "Failed to start socket listener instance: %v\n", err)
		os.Exit(-1)
	}

	socketListener.Wait()
}