1
0
mirror of https://github.com/MontFerret/ferret.git synced 2024-12-14 11:23:02 +02:00
ferret/main.go
3timeslazy 5188f9e71b WIP: Feature/#38 fix browser launcher (#246)
* switch DebuggingAddress from 0.0.0.0 -> http://0.0.0.0:9222

* rename res -> path

* change exec.Command call and args

* set default cdp value to empty string
2019-03-02 21:34:29 -05:00

191 lines
2.9 KiB
Go

package main
import (
"bufio"
"encoding/json"
"flag"
"fmt"
"io/ioutil"
"os"
"strings"
"github.com/MontFerret/ferret/cli"
"github.com/MontFerret/ferret/cli/browser"
"github.com/MontFerret/ferret/pkg/runtime/core"
)
type Params []string
func (p *Params) String() string {
return "[" + strings.Join(*p, ",") + "]"
}
func (p *Params) Set(value string) error {
*p = append(*p, value)
return nil
}
func (p *Params) ToMap() (map[string]interface{}, error) {
res := make(map[string]interface{})
for _, entry := range *p {
pair := strings.Split(entry, ":")
if len(pair) < 2 {
return nil, core.Error(core.ErrInvalidArgument, entry)
}
var value interface{}
key := pair[0]
err := json.Unmarshal([]byte(pair[1]), &value)
if err != nil {
fmt.Println(pair[1])
return nil, err
}
res[key] = value
}
return res, nil
}
var (
version string
conn = flag.String(
"cdp",
"",
"set CDP address",
)
launchBrowser = flag.Bool(
"cdp-launch",
false,
"launch Chrome",
)
proxyAddress = flag.String(
"proxy",
"",
"address of proxy server to use (only applicable to static pages, proxy for dynamic pages controlled by Chrome)",
)
userAgent = flag.String(
"user-agent",
"",
"set custom user agent. '*' triggers UA generation",
)
showTime = flag.Bool(
"time",
false,
"show how much time was taken to execute a query",
)
showVersion = flag.Bool(
"version",
false,
"show REPL version",
)
help = flag.Bool(
"help",
false,
"show this list",
)
)
func main() {
var params Params
flag.Var(
&params,
"param",
`query parameter (--param=foo:\"bar\", --param=id:1)`,
)
flag.Parse()
if *help {
flag.PrintDefaults()
os.Exit(0)
return
}
if *showVersion {
fmt.Println(version)
os.Exit(0)
return
}
cdpConn := *conn
if cdpConn == "" && *launchBrowser {
opts := make([]browser.Option, 0, 2)
//if *noUserData {
// opts = append(opts, browser.WithoutUserDataDir())
//}
// TODO: Make it optional.
opts = append(opts, browser.WithoutUserDataDir())
// we need to launch Chrome instance
b, err := browser.Launch(opts...)
if err != nil {
fmt.Println(fmt.Sprintf("Failed to launch browser:"))
fmt.Println(err)
os.Exit(1)
}
cdpConn = b.DebuggingAddress()
defer b.Close()
}
p, err := params.ToMap()
if err != nil {
fmt.Println(err)
os.Exit(1)
}
opts := cli.Options{
Cdp: cdpConn,
Params: p,
Proxy: *proxyAddress,
UserAgent: *userAgent,
ShowTime: *showTime,
}
stat, _ := os.Stdin.Stat()
if (stat.Mode() & os.ModeCharDevice) == 0 {
// check whether the app is getting a query via standard input
std := bufio.NewReader(os.Stdin)
b, err := ioutil.ReadAll(std)
if err != nil {
fmt.Println(err)
os.Exit(1)
return
}
cli.Exec(string(b), opts)
return
}
// filename was passed
if flag.NArg() > 0 {
cli.ExecFile(flag.Arg(0), opts)
return
}
// nothing was passed, run REPL
cli.Repl(version, opts)
}