standardize arguments

This commit is contained in:
Josh at WLTechBlog 2025-08-19 13:09:50 -05:00
parent 1651c4312e
commit 6bad614f9e
2 changed files with 9 additions and 2 deletions

View File

@ -616,7 +616,8 @@ func (c *Client) SwitchToIframe(tabID, selector string, timeout int) error {
// SwitchToMain switches the context back to the main page
// If tabID is empty, the current tab will be used
func (c *Client) SwitchToMain(tabID string) error {
// timeout is in seconds, 0 means no timeout
func (c *Client) SwitchToMain(tabID string, timeout int) error {
params := map[string]string{}
// Only include tab ID if it's provided
@ -624,6 +625,11 @@ func (c *Client) SwitchToMain(tabID string) error {
params["tab"] = tabID
}
// Add timeout if specified
if timeout > 0 {
params["timeout"] = strconv.Itoa(timeout)
}
resp, err := c.SendCommand("switch-main", params)
if err != nil {
return err

View File

@ -124,6 +124,7 @@ func main() {
// switch-main flags
switchMainTabID := switchMainCmd.String("tab", "", "Tab ID to switch back to main context (optional, uses current tab if not specified)")
switchMainTimeout := switchMainCmd.Int("timeout", 5, "Timeout in seconds for switching back to main context")
switchMainHost := switchMainCmd.String("host", "localhost", "Daemon host")
switchMainPort := switchMainCmd.Int("port", 8989, "Daemon port")
@ -412,7 +413,7 @@ func main() {
c := client.NewClient(*switchMainHost, *switchMainPort)
// Switch back to main page
err := c.SwitchToMain(*switchMainTabID)
err := c.SwitchToMain(*switchMainTabID, *switchMainTimeout)
if err != nil {
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
os.Exit(1)