mirror of https://github.com/schollz/croc.git
113 lines
3.6 KiB
Go
113 lines
3.6 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/urfave/cli"
|
|
"github.com/yudai/gotty/pkg/homedir"
|
|
)
|
|
|
|
const BUFFERSIZE = 1024
|
|
|
|
type AppConfig struct {
|
|
Relay bool `yaml:"relay" flagName:"relay" flagSName:"r" flagDescribe:"Run as relay" default:"false"`
|
|
Debug bool `yaml:"debug" flagName:"debug" flagSName:"d" flagDescribe:"Debug mode" default:"false"`
|
|
Wait bool `yaml:"wait" flagName:"wait" flagSName:"w" flagDescribe:"Wait for code to be sent" default:"false"`
|
|
PathSpec bool `yaml:"ask-save" flagName:"ask-save" flagSName:"q" flagDescribe:"Ask for path to save to" default:"false"`
|
|
DontEncrypt bool `yaml:"no-encrypt" flagName:"no-encrypt" flagSName:"g" flagDescribe:"Turn off encryption" default:"false"`
|
|
UseStdout bool `yaml:"stdout" flagName:"stdout" flagSName:"o" flagDescribe:"Use stdout" default:"false"`
|
|
Yes bool `yaml:"yes" flagName:"yes" flagSName:"y" flagDescribe:"Automatically accept file" default:"false"`
|
|
Local bool `yaml:"local" flagName:"local" flagSName:"lo" flagDescribe:"Use local relay when sending" default:"false"`
|
|
Server string `yaml:"server" flagName:"server" flagSName:"l" flagDescribe:"Croc relay to use" default:"cowyo.com"`
|
|
File string `yaml:"send" flagName:"send" flagSName:"s" flagDescribe:"File to send default:""`
|
|
Path string `yaml:"save" flagName:"save" flagSName:"p" flagDescribe:"Path to save to" default:""`
|
|
Code string `yaml:"code" flagName:"code" flagSName:"c" flagDescribe:"Use your own code phrase" default:""`
|
|
Rate int `yaml:"rate" flagName:"rate" flagSName:"R" flagDescribe:"Throttle down to speed in kbps" default:"1000000"`
|
|
NumberOfConnections int `yaml:"threads" flagName:"threads" flagSName:"n" flagDescribe:"Number of threads to use" default:"4"`
|
|
}
|
|
|
|
var email string
|
|
var author string
|
|
var version string
|
|
|
|
func main() {
|
|
|
|
app := cli.NewApp()
|
|
app.Name = "croc"
|
|
app.Version = version
|
|
app.Author = author
|
|
app.Email = email
|
|
app.Usage = "send file by croc bridge"
|
|
app.HideHelp = true
|
|
|
|
cli.AppHelpTemplate = helpTemplate
|
|
|
|
appOptions := &AppConfig{}
|
|
if err := ApplyDefaultValues(appOptions); err != nil {
|
|
exit(err, 1)
|
|
}
|
|
|
|
cliFlags, flagMappings, err := GenerateFlags(appOptions)
|
|
if err != nil {
|
|
exit(err, 3)
|
|
}
|
|
|
|
app.Flags = append(
|
|
cliFlags,
|
|
cli.StringFlag{
|
|
Name: "config",
|
|
Value: "~/.croc",
|
|
Usage: "Config file path",
|
|
EnvVar: "CROC_CONFIG",
|
|
},
|
|
)
|
|
|
|
app.Action = func(c *cli.Context) {
|
|
|
|
configFile := c.String("config")
|
|
_, err := os.Stat(homedir.Expand(configFile))
|
|
if configFile != "~/.croc" || !os.IsNotExist(err) {
|
|
if err := ApplyConfigFileYaml(configFile, appOptions); err != nil {
|
|
exit(err, 2)
|
|
}
|
|
}
|
|
|
|
ApplyFlags(cliFlags, flagMappings, c, appOptions)
|
|
|
|
if appOptions.Relay {
|
|
fmt.Println("running relay on local address " + GetLocalIP())
|
|
r := NewRelay(appOptions)
|
|
r.Run()
|
|
} else {
|
|
c, err := NewConnection(appOptions)
|
|
if err != nil {
|
|
fmt.Printf("Error! Please submit the following error to https://github.com/schollz/croc/issues:\n\n'%s'\n\n", err.Error())
|
|
return
|
|
}
|
|
err = c.Run()
|
|
if err != nil {
|
|
fmt.Printf("Error! Please submit the following error to https://github.com/schollz/croc/issues:\n\n'%s'\n\n", err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
app.Run(os.Args)
|
|
}
|
|
|
|
func getInput(prompt string) string {
|
|
reader := bufio.NewReader(os.Stdin)
|
|
fmt.Fprintf(os.Stderr, "%s", prompt)
|
|
text, _ := reader.ReadString('\n')
|
|
return strings.TrimSpace(text)
|
|
}
|
|
|
|
func exit(err error, code int) {
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
os.Exit(code)
|
|
}
|