@@ -17,6 +17,8 @@ import (
1717 "github.com/urfave/cli/v3"
1818)
1919
20+ const URL = "url"
21+
2022var ErrInvalidDeviceProperties = errors .New ("invalid device properties" )
2123var ErrScreenShotFailed = errors .New ("screenshot result error" )
2224
@@ -34,7 +36,9 @@ func main() {
3436 & cli.IntFlag {Name : "concurrency" , Aliases : []string {"c" }, Usage : "number of concurrent workers" },
3537 & cli.IntFlag {Name : "remote-port" , Usage : "remote DevTools port" },
3638 & cli.StringFlag {Name : "remote-host" , Usage : "remote DevTools host" },
37- & cli.StringFlag {Name : "device-type" , Value : "desktop" , Usage : "device type (desktop/mobile/tablet)" , Action : validDeviceType },
39+ & cli.StringFlag {
40+ Name : "device-type" , Value : "desktop" , Usage : "device type (desktop/mobile/tablet)" , Action : validDeviceType ,
41+ },
3842 & cli.StringFlag {Name : "device-size" , Value : "large" , Usage : "device size preset" , Action : validDeviceSize },
3943 & cli.StringFlag {Name : "user-agent" , Value : "chrome" , Usage : "browser user-agent preset" },
4044 }
@@ -53,7 +57,7 @@ func main() {
5357 Name : "screenshot" ,
5458 Usage : "Screenshot one or more URLs" ,
5559 Arguments : []cli.Argument {
56- & cli.StringArgs {Name : "url" , Min : 1 , Max : - 1 },
60+ & cli.StringArgs {Name : URL , Min : 1 , Max : - 1 },
5761 },
5862 Flags : append ([]cli.Flag {
5963 & cli.StringFlag {Name : "output-dir" , Value : "tmp/" , Aliases : []string {"o" }, Usage : "directory for screenshots" },
@@ -64,7 +68,7 @@ func main() {
6468 }, {
6569 Name : "markdown" ,
6670 Usage : "Get the markdown content of a URL" ,
67- Arguments : []cli.Argument {& cli.StringArg {Name : "url" }},
71+ Arguments : []cli.Argument {& cli.StringArg {Name : URL }},
6872 Flags : append ([]cli.Flag {
6973 & cli.BoolFlag {Name : "include-images" , Aliases : []string {"i" }, Usage : "include images in markdown output" },
7074 }, baseFlags ... ),
@@ -78,7 +82,7 @@ func main() {
7882 }, {
7983 Name : "plaintext" ,
8084 Usage : "Get the plantext content of a URL" ,
81- Arguments : []cli.Argument {& cli.StringArg {Name : "url" }},
85+ Arguments : []cli.Argument {& cli.StringArg {Name : URL }},
8286 Flags : baseFlags ,
8387 Action : func (ctx context.Context , cmd * cli.Command ) error {
8488 return runTask (ctx , cmd , "plaintext" , map [string ]any {}, stdOutCallback )
@@ -100,12 +104,12 @@ func runTask(ctx context.Context, cmd *cli.Command, name string, params map[stri
100104 deviceType := cmd .StringArg ("device-type" )
101105 remoteHost := cmd .String ("remote-host" )
102106 remotePort := cmd .Int ("remote-port" )
103- urls := cmd .StringArgs ("url" )
107+ urls := cmd .StringArgs (URL )
104108
105109 opts := []engine.Option {engine .WithLogger (virgo .Logger ())}
106110
107111 if len (urls ) == 0 {
108- url := cmd .StringArg ("url" )
112+ url := cmd .StringArg (URL )
109113 urls = []string {url }
110114 }
111115
@@ -147,7 +151,7 @@ func stdOutCallback(cmd *cli.Command, e *engine.Engine) error {
147151 }
148152
149153 logger .Debug ().
150- Str ("url" , r .URL ).
154+ Str (URL , r .URL ).
151155 Str ("duration" , r .Elapsed .String ()).
152156 Msg ("plaintext result" )
153157
@@ -186,7 +190,7 @@ func screenshotCallback(cmd *cli.Command, e *engine.Engine) error {
186190 }
187191
188192 logger .Debug ().
189- Str ("url" , r .URL ).
193+ Str (URL , r .URL ).
190194 Str ("duration" , r .Elapsed .String ()).
191195 Msg ("screenshot result" )
192196
0 commit comments