main.go

package main

import (
	log2 "log"
	"net/http"
	"os"
	"path/filepath"
	"time"

	"github.com/NYTimes/logrotate"
	"github.com/apex/log"
	"github.com/apex/log/handlers/cli"
	"github.com/apex/log/handlers/multi"
	"github.com/apex/log/handlers/text"
	"github.com/kubectyl/kuber/remote"
	"github.com/kubectyl/sftp-server/config"
	"github.com/kubectyl/sftp-server/sftp"
)

// Configures the global logger for Zap so that we can call it from any location
// in the code without having to pass around a logger instance.
func initLogging() {
	dir := config.Get().System.LogDirectory
	p := filepath.Join(dir, "/sftp-server.log")
	if err := os.MkdirAll(filepath.Dir(p), 0o755); err != nil {
		log2.Fatalf("sftp: could not create internal sftp server log directory: %s", err)
	}
	w, err := logrotate.NewFile(p)
	if err != nil {
		log2.Fatalf("failed to create server log: %s", err)
	}
	log.SetLevel(log.InfoLevel)
	if config.Get().Debug {
		log.SetLevel(log.DebugLevel)
	}
	log.SetHandler(multi.New(text.New(os.Stdout), cli.New(w.File)))
	log.WithField("path", p).Info("writing log file to disk")
}

func main() {
	initLogging()

	pclient := remote.New(
		config.Get().PanelLocation,
		remote.WithCredentials(config.Get().AuthenticationTokenId, config.Get().AuthenticationToken),
		remote.WithHttpClient(&http.Client{
			Timeout: time.Second * time.Duration(config.Get().RemoteQuery.Timeout),
		}),
	)

	// Run the SFTP server.
	if err := sftp.New(pclient).Run(); err != nil {
		log.WithError(err).Fatal("failed to initialize the sftp server")
		return
	}
}