Compare commits

..

4 Commits

8 changed files with 60 additions and 45 deletions

View File

@ -1,3 +1,5 @@
//go:build windows
package cmd package cmd
import ( import (

View File

@ -1,3 +1,5 @@
//go:build windows
package cmd package cmd
import ( import (

View File

@ -1,3 +1,5 @@
//go:build windows
package cmd package cmd
import ( import (

View File

@ -1,8 +1,11 @@
//go:build windows
package proxy package proxy
import ( import (
"errors" "errors"
"fmt" "fmt"
"log"
"os" "os"
"path/filepath" "path/filepath"
"strings" "strings"
@ -31,9 +34,11 @@ func Debug(cfgPath string) {
fmt.Printf("\nConfig:\n") fmt.Printf("\nConfig:\n")
fmt.Printf("%s\n\n", path) fmt.Printf("%s\n\n", path)
proxy, _ := ReadSystemProxy() proxy, err := ReadSystemProxy()
proxyServer := proxy.Server proxyServer := proxy.Server
if proxyServer == "" { if err != nil {
log.Fatal(err)
} else if proxyServer == "" {
proxyServer = "[N/A]" proxyServer = "[N/A]"
} }
@ -41,7 +46,11 @@ func Debug(cfgPath string) {
fmt.Printf("Enabled: %t\n", proxy.Enabled) fmt.Printf("Enabled: %t\n", proxy.Enabled)
fmt.Printf("Server: %s\n\n", proxyServer) fmt.Printf("Server: %s\n\n", proxyServer)
configs, _ := util.ReadConfigs(cfgPath) configs, err := util.ReadConfigs(cfgPath)
if err != nil {
log.Fatal(err)
}
for _, config := range configs { for _, config := range configs {
configCmd := config.Name configCmd := config.Name
// Use command instead of name, if given // Use command instead of name, if given

View File

@ -1,18 +1,21 @@
//go:build windows
package proxy package proxy
import ( import (
"log"
"github.com/Baipyrus/ProxySwitcher/util" "github.com/Baipyrus/ProxySwitcher/util"
) )
func Set(cfgPath string) { func Set(cfgPath string) {
stdin, closeFunc, _ := util.ReadyCmd()
proxy, _ := ReadSystemProxy() proxy, _ := ReadSystemProxy()
// Set system proxy, if not already // Set system proxy, if not already
if !proxy.Enabled { if !proxy.Enabled {
SetSystemProxy(true) SetSystemProxy(true)
} }
var failed bool
configs, _ := util.ReadConfigs(cfgPath) configs, _ := util.ReadConfigs(cfgPath)
for _, config := range configs { for _, config := range configs {
configCmd := config.Name configCmd := config.Name
@ -22,8 +25,11 @@ func Set(cfgPath string) {
} }
commands := generateCommands(configCmd, config.Set, proxy.Server) commands := generateCommands(configCmd, config.Set, proxy.Server)
util.ExecCmds(commands, stdin) failed = util.ExecCmds(commands)
} }
closeFunc() // Additional feedback on error
if failed {
log.Printf("One or more commands failed to execute. Run command 'debug' to see more.\n")
}
} }

View File

@ -1,18 +1,21 @@
//go:build windows
package proxy package proxy
import ( import (
"log"
"github.com/Baipyrus/ProxySwitcher/util" "github.com/Baipyrus/ProxySwitcher/util"
) )
func Unset(cfgPath string) { func Unset(cfgPath string) {
stdin, closeFunc, _ := util.ReadyCmd()
proxy, _ := ReadSystemProxy() proxy, _ := ReadSystemProxy()
// Unset system proxy, if not already // Unset system proxy, if not already
if proxy.Enabled { if proxy.Enabled {
SetSystemProxy(false) SetSystemProxy(false)
} }
var failed bool
configs, _ := util.ReadConfigs(cfgPath) configs, _ := util.ReadConfigs(cfgPath)
for _, config := range configs { for _, config := range configs {
configCmd := config.Name configCmd := config.Name
@ -22,8 +25,11 @@ func Unset(cfgPath string) {
} }
commands := generateCommands(configCmd, config.Unset, "") commands := generateCommands(configCmd, config.Unset, "")
util.ExecCmds(commands, stdin) failed = util.ExecCmds(commands)
} }
closeFunc() // Additional feedback on error
if failed {
log.Printf("One or more commands failed to execute. Run command 'debug' to see more.\n")
}
} }

View File

@ -21,6 +21,8 @@ func processVars(cmd *util.Command, isVariableType bool) {
for _, arg := range cmd.Arguments { for _, arg := range cmd.Arguments {
cmd.Name = strings.Replace(cmd.Name, "$PRSW_ARG", arg, 1) cmd.Name = strings.Replace(cmd.Name, "$PRSW_ARG", arg, 1)
} }
cmd.Arguments = nil
} }
func injectProxy(cmd *util.Command, variant *util.Variant, proxyServer string) { func injectProxy(cmd *util.Command, variant *util.Variant, proxyServer string) {

View File

@ -1,45 +1,31 @@
//go:build windows
package util package util
import ( import (
"bytes"
"fmt" "fmt"
"io" "log"
"os/exec" "os/exec"
"strings" "strings"
) )
// Create a single powershell process and leave closing, input and output open func ExecCmds(commands []*Command) bool {
func ReadyCmd() (*io.WriteCloser, func() error, error) { var failed bool
cmd := exec.Command("powershell", "-NoLogo", "-NoProfile", "-Command", "-")
var stdout bytes.Buffer
cmd.Stdout = &stdout
cmd.Stderr = &stdout
stdin, err := cmd.StdinPipe()
if err != nil {
return nil, nil, err
}
err = cmd.Start()
if err != nil {
return nil, nil, err
}
return &stdin, func() error {
// Close stdin pipe
stdin.Close()
// Wait for command to flush
err := cmd.Wait()
return err
}, nil
}
func ExecCmds(commands []*Command, stdin *io.WriteCloser) {
for _, command := range commands { for _, command := range commands {
var cmdStr string = command.Name
// Combine command into single string if args are given
if len(command.Arguments) > 0 {
cmdArgs := strings.Join(command.Arguments, " ") cmdArgs := strings.Join(command.Arguments, " ")
cmdStr := fmt.Sprintf("%s %s", command.Name, cmdArgs) cmdStr = fmt.Sprintf("%s %s", command.Name, cmdArgs)
fmt.Fprintln(*stdin, cmdStr) }
// Try executing command in default shell
cmd := exec.Command("powershell", "-NoLogo", "-NoProfile", "-Command", cmdStr)
if err := cmd.Run(); err != nil {
log.Printf("Command '%s' failed!\n", cmdStr)
failed = true
} }
} }
return failed
}