Compare commits

..

20 Commits

Author SHA1 Message Date
1dc6ce6e5e fix: type conversion 2023-12-27 07:42:38 +01:00
69389d030b refactor: parameterize submit delay 2023-12-27 07:38:48 +01:00
cc7a59c109 feat triple c lick on save button archer 2023-07-04 16:42:46 +02:00
62bda54d98 feat: adjust waiting times 2023-07-02 23:56:23 +02:00
0933174ea1 fix bug 2023-07-02 23:28:38 +02:00
bf700eeeb3 feat add log date 2023-07-02 23:13:30 +02:00
bbc94e0375 polish logs 2023-07-02 23:08:07 +02:00
ad8ed58366 feat increase wait 2023-07-02 22:16:17 +02:00
db9f7495aa increase wait time after login submit 2023-07-02 22:14:31 +02:00
0ae922e58d feat: change jquery to pure js 2023-07-02 21:28:53 +02:00
fc35559ab8 feat wait a little more 2023-07-02 20:16:34 +02:00
c09933767b feat: archer add print message 2023-07-02 20:15:54 +02:00
cc9c9354bc feat: add archer ax50 2023-07-02 19:50:59 +02:00
55ed1c1b36 increase wait done 50 ms 2023-06-03 17:09:38 +02:00
8032850b0b fix nucom switcher, now it is working fine 2023-06-03 01:08:16 +02:00
5d5d90d3f6 feat: handle multiple providers 2023-06-02 03:24:12 +02:00
7742a70f6c feat: add close page 2023-03-25 08:19:57 +01:00
857aec79e3 fix: remove test line 2023-03-07 23:08:24 +01:00
10c80effce feat: add build for raspi 2023-03-07 23:07:29 +01:00
b86fc071ec fix bug missing " 2023-03-07 23:07:20 +01:00
9 changed files with 375 additions and 123 deletions

View File

@@ -17,6 +17,10 @@ build:
rm -rf ${BINARY_DIR} rm -rf ${BINARY_DIR}
mkdir ${BINARY_DIR} mkdir ${BINARY_DIR}
env GOOS=linux CGO_ENABLED=0 GOARCH=amd64 go build -o ./${BINARY_DIR}/${BINARY_NAME} main.go env GOOS=linux CGO_ENABLED=0 GOARCH=amd64 go build -o ./${BINARY_DIR}/${BINARY_NAME} main.go
build_raspi:
rm -rf ${BINARY_DIR}
mkdir ${BINARY_DIR}
env GOOS=linux CGO_ENABLED=0 GOARCH=arm64 go build -o ./${BINARY_DIR}/${BINARY_NAME} main.go
build_freebsd: build_freebsd:
rm -rf ${BINARY_DIR} rm -rf ${BINARY_DIR}
mkdir ${BINARY_DIR} mkdir ${BINARY_DIR}

View File

@@ -8,11 +8,12 @@ import (
) )
type Config struct { type Config struct {
Page string `required:"true" split_words:"true"` Page string `required:"true" split_words:"true"`
AdminUser string `required:"true" split_words:"true"` AdminUser string `required:"true" split_words:"true"`
Password string `required:"true" split_words:"true"` Password string `required:"true" split_words:"true"`
LogFile bool `required:"true" split_words:"true"` LogFile bool `required:"true" split_words:"true"`
Bin string `required:"true" split_words:"true"` Bin string `required:"true" split_words:"true"`
SubmitDelay int `default:"7500" required:"false" split_words:"true"`
} }
func NewConfig(envFilePath string) *Config { func NewConfig(envFilePath string) *Config {

View File

@@ -0,0 +1,108 @@
package providers
import (
"fmt"
"log"
"time"
"gitea.urkob.com/urko/go-wifi-switcher/pkg/provider"
"github.com/go-rod/rod"
)
type ArcherAx50 struct {
passwordID string
loginButtonID string
wirelessTab string
chkWifi2gID string
saveButtonID string
page *rod.Page
}
func NewArcherAx50(page *rod.Page) ArcherAx50 {
return ArcherAx50{
passwordID: "#login-password",
loginButtonID: "#login-btn",
wirelessTab: "wireless",
chkWifi2gID: "chk_enable_wireless_24g",
saveButtonID: "#total_save",
page: page,
}
}
func (p ArcherAx50) Login(user, pass string) error {
time.Sleep(time.Millisecond * 5550)
p.page.MustEvaluate(&rod.EvalOptions{
JS: `() => {
document.querySelectorAll("input.password-text").forEach((v, i) => {
v.click();
v.focus();
v.value="` + pass + `";
})
return '';
}`,
})
time.Sleep(time.Millisecond * 550)
log.Println(p.passwordID, "DONE")
login, err := p.page.Element(p.loginButtonID)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.loginButtonID, err)
}
login.MustClick()
log.Println(p.loginButtonID, "DONE")
time.Sleep(time.Millisecond * 4550)
return nil
}
func (p ArcherAx50) SwitchWIFI(cfg provider.SwitchConfig) error {
pageURL := p.page.MustInfo().URL
log.Println("p.page.MustInfo().URL", pageURL)
wirelessTab, err := p.page.Element(`[name="` + p.wirelessTab + `"]`)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.wirelessTab, err)
}
wirelessTab.MustClick()
log.Println(p.wirelessTab, "DONE")
time.Sleep(time.Millisecond * 1850)
checked := p.page.MustEvaluate(&rod.EvalOptions{
JS: `() => {
return document.getElementById('` + p.chkWifi2gID + `').checked;
}`,
})
isChecked := checked.Value.Bool()
log.Println(getCheckedMessage(isChecked))
p.page.MustEvaluate(&rod.EvalOptions{
JS: `() => {
document.getElementById('` + p.chkWifi2gID + `').checked=` + fmt.Sprint(!isChecked) + `;
return '';
}`,
})
time.Sleep(time.Millisecond * 550)
log.Println(p.chkWifi2gID, "DONE")
saveButton, err := p.page.Element(p.saveButtonID)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.saveButtonID, err)
}
saveButton.MustClick()
saveButton.MustClick()
saveButton.MustClick()
log.Println(p.saveButtonID, "DONE")
time.Sleep(time.Millisecond * time.Duration(cfg.SubmitDelay))
return nil
}
func getCheckedMessage(isChecked bool) string {
if isChecked {
return "Time to sleep, let's switch off"
}
return "Time to laborate, let's switch WIFI on"
}

View File

@@ -0,0 +1,113 @@
package providers
import (
"fmt"
"log"
"time"
"gitea.urkob.com/urko/go-wifi-switcher/pkg/provider"
"github.com/go-rod/rod"
)
type Huawei struct {
usernameID string
userpasswordID string
loginText string
skipText string
netID string
netWlanID string
mainFrameID string
wlanEnableID string
page *rod.Page
}
func NewHuawei(page *rod.Page) Huawei {
return Huawei{
usernameID: "#username",
userpasswordID: "#userpassword",
loginText: "Login",
skipText: "Skip",
netID: "#Net",
netWlanID: "#Net-WLAN",
mainFrameID: "#mainFrame",
wlanEnableID: "#wlanEnable",
page: page,
}
}
func (p Huawei) Login(user, pass string) error {
username, err := p.page.Element(p.usernameID)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.userpasswordID, err)
}
username.MustInput(user)
log.Println(p.usernameID, "DONE")
userpassword, err := p.page.Element(p.userpasswordID)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.userpasswordID, err)
}
userpassword.MustInput(pass)
log.Println(p.userpasswordID, "DONE")
login, err := p.page.Element(`input[value="` + p.loginText + `"]`)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.loginText, err)
}
login.MustClick()
log.Println(p.loginText, "DONE")
skip, err := p.page.Element(`input[value="` + p.skipText + `"]`)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.skipText, err)
}
skip.MustClick()
log.Println(p.skipText, "DONE")
return nil
}
func (p Huawei) SwitchWIFI(cfg provider.SwitchConfig) error {
net, err := p.page.Element(p.netID)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.netID, err)
}
net.MustClick()
log.Println(p.netID, "DONE")
netWlan, err := p.page.Element(p.netWlanID)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.netWlanID, err)
}
netWlan.MustClick()
log.Println(p.netWlanID, "DONE")
frameElement, err := p.page.Element(p.mainFrameID)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.mainFrameID, err)
}
mainFrame, err := frameElement.Frame()
if err != nil {
return fmt.Errorf("frameElement.Frame: %s", err)
}
wlanEnable, err := mainFrame.Element(p.wlanEnableID)
if err != nil {
return fmt.Errorf("mainFrame.Element %s: %s", p.wlanEnableID, err)
}
wlanEnable.MustClick()
log.Println(p.wlanEnableID, "DONE")
sysSubmitID := "#sysSubmit"
sysSubmit, err := mainFrame.Element(sysSubmitID)
if err != nil {
return fmt.Errorf("mainFrame.Element %s: %s", sysSubmitID, err)
}
sysSubmit.MustClick()
log.Println(sysSubmitID, "DONE")
// TODO: improve this and wait until something should happens (refresh)
time.Sleep(time.Second * 3)
return nil
}

105
internal/providers/nucom.go Normal file
View File

@@ -0,0 +1,105 @@
package providers
import (
"fmt"
"log"
"time"
"gitea.urkob.com/urko/go-wifi-switcher/pkg/provider"
"github.com/go-rod/rod"
)
type Nucom struct {
usernameName string
passwordName string
loginSubmitID string
wirelessTabID string
wirelessMenuText string
configureWarningID string
wifiRadioID string
saveID string
radioOff string
radioOn string
page *rod.Page
}
func NewNucom(page *rod.Page) Nucom {
return Nucom{
usernameName: "loginUser",
passwordName: "loginPass",
loginSubmitID: "btnApply",
wirelessTabID: "menu_wireless",
wirelessMenuText: "Básico",
configureWarningID: "Configurar ahora",
wifiRadioID: "radiohiddenButton",
radioOff: "Apagada",
radioOn: "Encendida",
saveID: "basicApply",
page: page,
}
}
func (p Nucom) Login(user, pass string) error {
username, err := p.page.Element(`input[name="` + p.usernameName + `"]`)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.passwordName, err)
}
username.MustInput(user)
log.Println(p.usernameName, "DONE")
userpassword, err := p.page.Element(`input[name="` + p.passwordName + `"]`)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.passwordName, err)
}
userpassword.MustInput(pass)
log.Println(p.passwordName, "DONE")
login, err := p.page.Element(`input[id="` + p.loginSubmitID + `"]`)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.loginSubmitID, err)
}
login.MustClick()
log.Println(p.loginSubmitID, "DONE")
return nil
}
func (p Nucom) SwitchWIFI(cfg provider.SwitchConfig) error {
wirelessMenu, err := p.page.Element(`a[id="` + p.wirelessTabID + `"]`)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.wirelessTabID, err)
}
wirelessMenu.MustClick()
log.Println(p.wirelessTabID, "DONE")
time.Sleep(time.Millisecond * 500)
log.Println(p.page.MustInfo().URL)
offWifiRadio, err := p.page.Element(`input[type="radio"][name="radiohiddenButton"][value="0"]`)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.wifiRadioID, err)
}
offChecked := offWifiRadio.MustProperty("checked")
if offChecked.Bool() {
log.Println("it is off, let's switch on")
onRadio, err := p.page.Element(`input[type="radio"][name="radiohiddenButton"][value="1"]`)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.wifiRadioID, err)
}
onRadio.MustClick()
} else {
log.Println("it is on, time to switch off")
offWifiRadio.MustClick()
}
log.Println(p.wifiRadioID, "DONE")
time.Sleep(time.Millisecond * 150)
saveButton, err := p.page.Element(`input[id="` + p.saveID + `"]`)
if err != nil {
return fmt.Errorf("page.Element %s: %s", p.saveID, err)
}
saveButton.MustClick()
log.Println(p.saveID, "DONE")
time.Sleep(time.Millisecond * 150)
return nil
}

View File

@@ -2,114 +2,48 @@ package switcher
import ( import (
"fmt" "fmt"
"log"
"time"
"gitea.urkob.com/urko/go-wifi-switcher/cfg" "gitea.urkob.com/urko/go-wifi-switcher/cfg"
"gitea.urkob.com/urko/go-wifi-switcher/pkg/provider"
"github.com/go-rod/rod" "github.com/go-rod/rod"
) )
type switcher struct { type Switcher struct {
remoteControlBrowserUrl string Browser *rod.Browser
config *cfg.Config Page *rod.Page
config *cfg.Config
} }
var ( func NewSwitcher(remoteControlBrowserUrl string, config *cfg.Config) Switcher {
usernameID = "#username"
userpasswordID = "#userpassword"
loginText = "Login"
skipText = "Skip"
netID = "#Net"
netWlanID = "#Net-WLAN"
mainFrameID = "#mainFrame"
wlanEnableID = "#wlanEnable"
)
func NewSwitcher(remoteControlBrowserUrl string, config *cfg.Config) *switcher {
return &switcher{
remoteControlBrowserUrl: remoteControlBrowserUrl,
config: config,
}
}
func (s *switcher) SwitchWIFI() error {
browser := rod.New(). browser := rod.New().
ControlURL(s.remoteControlBrowserUrl). ControlURL(remoteControlBrowserUrl).
MustConnect(). MustConnect().
MustIgnoreCertErrors(true) MustIgnoreCertErrors(true)
defer browser.MustClose() page := browser.MustPage(config.Page)
page := browser.MustPage(s.config.Page) return Switcher{
Browser: browser,
username, err := page.Element(usernameID) Page: page,
if err != nil { config: config,
return fmt.Errorf("page.Element %s: %s", userpasswordID, err)
} }
username.MustInput(s.config.AdminUser) }
log.Println(usernameID, "DONE")
userpassword, err := page.Element(userpasswordID) func (s Switcher) SwitchWIFI(prv provider.ProviderIface) error {
if err != nil { defer s.Browser.MustClose()
return fmt.Errorf("page.Element %s: %s", userpasswordID, err) defer s.Page.Close()
}
userpassword.MustInput(s.config.Password)
log.Println(userpasswordID, "DONE")
login, err := page.Element(`input[value=` + loginText + `"]`) if err := prv.Login(s.config.AdminUser, s.config.Password); err != nil {
if err != nil { return fmt.Errorf("prv.Login %w", err)
return fmt.Errorf("page.Element %s: %s", loginText, err)
}
login.MustClick()
log.Println(loginText, "DONE")
skip, err := page.Element(`input[value="` + skipText + `"]`)
if err != nil {
return fmt.Errorf("page.Element %s: %s", skipText, err)
}
skip.MustClick()
log.Println(skipText, "DONE")
net, err := page.Element(netID)
if err != nil {
return fmt.Errorf("page.Element %s: %s", netID, err)
}
net.MustClick()
log.Println(netID, "DONE")
netWlan, err := page.Element(netWlanID)
if err != nil {
return fmt.Errorf("page.Element %s: %s", netWlanID, err)
}
netWlan.MustClick()
log.Println(netWlanID, "DONE")
frameElement, err := page.Element(mainFrameID)
if err != nil {
return fmt.Errorf("page.Element %s: %s", mainFrameID, err)
} }
mainFrame, err := frameElement.Frame() switchConfig := provider.SwitchConfig{
if err != nil { SubmitDelay: s.config.SubmitDelay,
return fmt.Errorf("frameElement.Frame: %s", err)
} }
wlanEnable, err := mainFrame.Element(wlanEnableID) if err := prv.SwitchWIFI(switchConfig); err != nil {
if err != nil { return fmt.Errorf("prv.SwitchWIFI %w", err)
return fmt.Errorf("mainFrame.Element %s: %s", wlanEnableID, err)
} }
wlanEnable.MustClick()
log.Println(wlanEnableID, "DONE")
sysSubmitID := "#sysSubmit"
sysSubmit, err := mainFrame.Element(sysSubmitID)
if err != nil {
return fmt.Errorf("mainFrame.Element %s: %s", sysSubmitID, err)
}
sysSubmit.MustClick()
log.Println(sysSubmitID, "DONE")
// TODO: improve this and wait until something should happens (refresh)
time.Sleep(time.Second * 3)
return nil return nil
} }

32
main.go
View File

@@ -8,15 +8,13 @@ import (
"time" "time"
"gitea.urkob.com/urko/go-wifi-switcher/cfg" "gitea.urkob.com/urko/go-wifi-switcher/cfg"
"gitea.urkob.com/urko/go-wifi-switcher/internal/providers"
"gitea.urkob.com/urko/go-wifi-switcher/internal/switcher" "gitea.urkob.com/urko/go-wifi-switcher/internal/switcher"
pkgswitcher "gitea.urkob.com/urko/go-wifi-switcher/pkg/switcher"
"github.com/go-rod/rod/lib/launcher" "github.com/go-rod/rod/lib/launcher"
"github.com/go-rod/rod/lib/utils" "github.com/go-rod/rod/lib/utils"
"github.com/ysmood/leakless" "github.com/ysmood/leakless"
) )
var switcherIface pkgswitcher.SwitcherIface
func main() { func main() {
envFile := "" envFile := ""
if os.Getenv("ENV") != "prod" { if os.Getenv("ENV") != "prod" {
@@ -24,8 +22,8 @@ func main() {
} }
config := cfg.NewConfig(envFile) config := cfg.NewConfig(envFile)
log.SetFlags(log.Ldate)
log.SetFlags(log.Lmicroseconds) log.SetFlags(log.Lmicroseconds)
config.LogFile = false
if config.LogFile { if config.LogFile {
logFileName := fmt.Sprintf("%s.txt", time.Now().Format(strings.ReplaceAll(time.RFC1123Z, ":", "_"))) logFileName := fmt.Sprintf("%s.txt", time.Now().Format(strings.ReplaceAll(time.RFC1123Z, ":", "_")))
f, err := os.OpenFile(logFileName, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0o644) f, err := os.OpenFile(logFileName, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0o644)
@@ -40,6 +38,7 @@ func main() {
var lc *launcher.Launcher var lc *launcher.Launcher
var remoteControlBrowserURL string var remoteControlBrowserURL string
log.Println("config", config)
if config.Bin != "" { if config.Bin != "" {
lc = launcher.New().Bin(config.Bin) lc = launcher.New().Bin(config.Bin)
remoteControlBrowserURL = lc.MustLaunch() remoteControlBrowserURL = lc.MustLaunch()
@@ -58,28 +57,11 @@ func main() {
} }
sw := switcher.NewSwitcher(remoteControlBrowserURL, config) sw := switcher.NewSwitcher(remoteControlBrowserURL, config)
if err := sw.SwitchWIFI(); err != nil { prv := providers.NewArcherAx50(sw.Page)
err := retry(5, time.Second*5, sw.SwitchWIFI) err := sw.SwitchWIFI(prv)
if err != nil { if err != nil {
log.Fatalln("couldn't retry:", err) panic(err)
}
} }
log.Println("task completed") log.Println("task completed")
} }
func retry(attempts int, sleep time.Duration, f func() error) error {
var err error
for i := 0; i < attempts; i++ {
if i > 0 {
log.Println("retrying after error:", err)
time.Sleep(sleep)
sleep *= 2
}
err = f()
if err == nil {
return nil
}
}
return fmt.Errorf("after %d attempts, last error: %s", attempts, err)
}

10
pkg/provider/provider.go Normal file
View File

@@ -0,0 +1,10 @@
package provider
type ProviderIface interface {
Login(string, string) error
SwitchWIFI(SwitchConfig) error
}
type SwitchConfig struct {
SubmitDelay int
}

View File

@@ -1,5 +0,0 @@
package switcher
type SwitcherIface interface {
SwitchWIFI() error
}