mirror of
https://github.com/MontFerret/ferret.git
synced 2024-12-14 11:23:02 +02:00
5188f9e71b
* 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
191 lines
2.9 KiB
Go
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(
|
|
¶ms,
|
|
"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)
|
|
}
|