Refactor to make it easier to split into modules
This commit is contained in:
parent
3b767548fd
commit
0bac4b099f
47
config.go
47
config.go
@ -2,11 +2,9 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/url"
|
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
|
||||||
"github.com/antonmedv/expr"
|
|
||||||
"github.com/hashicorp/hcl/v2/hclsimple"
|
"github.com/hashicorp/hcl/v2/hclsimple"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -31,41 +29,6 @@ type BrowserRuleConfig struct {
|
|||||||
MatchExpr string `hcl:"match"`
|
MatchExpr string `hcl:"match"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func MakeMatchFunc(rule string) MatchFunc {
|
|
||||||
return func(dest url.URL) bool {
|
|
||||||
env := map[string]interface{}{
|
|
||||||
// Prebuilt match functions
|
|
||||||
"matchAny": matchAny,
|
|
||||||
"matchHostRegexp": matchHostRegexp,
|
|
||||||
"matchHostname": matchHostname,
|
|
||||||
"matchNever": matchNever,
|
|
||||||
"matchRegexp": matchRegexp,
|
|
||||||
|
|
||||||
// Helpers for building custom matchers
|
|
||||||
"MatchFunc": func(foundMatch bool) MatchFunc {
|
|
||||||
return func(_ url.URL) bool {
|
|
||||||
return foundMatch
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"fullUrl": dest.String(),
|
|
||||||
"hostname": dest.Hostname(),
|
|
||||||
"url": dest,
|
|
||||||
}
|
|
||||||
matchFuncExpr, err := expr.Eval(rule, env)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Printf("Error evaluating rule %s: %v", rule, err)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
matchFunc, ok := matchFuncExpr.(MatchFunc)
|
|
||||||
if !ok {
|
|
||||||
fmt.Printf("Error evaluating rule %s. Did not evaluate to a MatchFunc.", rule)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return matchFunc(dest)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func LoadConfig(path string) ([]BrowserRule, error) {
|
func LoadConfig(path string) ([]BrowserRule, error) {
|
||||||
rules := []BrowserRule{}
|
rules := []BrowserRule{}
|
||||||
var config Config
|
var config Config
|
||||||
@ -75,11 +38,11 @@ func LoadConfig(path string) ([]BrowserRule, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, rule := range config.Rules {
|
for _, rule := range config.Rules {
|
||||||
rules = append(rules, BrowserRule{
|
rules = append(rules, NewBrowserRuleFromExpr(
|
||||||
command: rule.BrowserCommand[0],
|
rule.BrowserCommand[0],
|
||||||
args: rule.BrowserCommand[1:],
|
rule.BrowserCommand[1:],
|
||||||
matcher: MakeMatchFunc(rule.MatchExpr),
|
rule.MatchExpr,
|
||||||
})
|
))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Add default browser rule to the end
|
// Add default browser rule to the end
|
||||||
|
94
main.go
94
main.go
@ -4,102 +4,8 @@ import (
|
|||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/url"
|
"net/url"
|
||||||
"os/exec"
|
|
||||||
"regexp"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// MatchFunc is a signature for a function to match a URL
|
|
||||||
type MatchFunc = func(url.URL) bool
|
|
||||||
|
|
||||||
// A BrowserRule is a rule that will launch a browser, if matched
|
|
||||||
type BrowserRule struct {
|
|
||||||
matcher MatchFunc
|
|
||||||
command string
|
|
||||||
args []string
|
|
||||||
}
|
|
||||||
|
|
||||||
// IsMatched will check to see the rule matches the provied URL
|
|
||||||
func (r BrowserRule) IsMatched(dest url.URL) bool {
|
|
||||||
return r.matcher(dest)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Launch will launch the browser with the provided URL
|
|
||||||
func (r BrowserRule) Launch(dest url.URL) error {
|
|
||||||
args := append(r.args, "--", dest.String())
|
|
||||||
cmd := exec.Command(r.command, args...)
|
|
||||||
return cmd.Start()
|
|
||||||
}
|
|
||||||
|
|
||||||
// MaybeLaunch will lauch the browser with the provided URL if it matches the rule
|
|
||||||
func (r BrowserRule) MaybeLaunch(dest url.URL) (bool, error) {
|
|
||||||
if r.IsMatched(dest) {
|
|
||||||
return true, r.Launch(dest)
|
|
||||||
}
|
|
||||||
return false, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func matchHostname(hostnames ...string) MatchFunc {
|
|
||||||
return func(dest url.URL) bool {
|
|
||||||
for _, host := range hostnames {
|
|
||||||
if host == dest.Hostname() {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func matchHostRegexp(hostRegexp ...string) MatchFunc {
|
|
||||||
matchers := []*regexp.Regexp{}
|
|
||||||
for _, s := range hostRegexp {
|
|
||||||
matchers = append(matchers, regexp.MustCompile(s))
|
|
||||||
}
|
|
||||||
return func(dest url.URL) bool {
|
|
||||||
for _, matcher := range matchers {
|
|
||||||
if matcher.MatchString(dest.Hostname()) {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func matchRegexp(urlRegexp ...string) MatchFunc {
|
|
||||||
matchers := []*regexp.Regexp{}
|
|
||||||
for _, s := range urlRegexp {
|
|
||||||
matchers = append(matchers, regexp.MustCompile(s))
|
|
||||||
}
|
|
||||||
return func(dest url.URL) bool {
|
|
||||||
for _, matcher := range matchers {
|
|
||||||
if matcher.MatchString(dest.String()) {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func matchAny(matchFuncs ...MatchFunc) MatchFunc {
|
|
||||||
return func(dest url.URL) bool {
|
|
||||||
for _, f := range matchFuncs {
|
|
||||||
if f(dest) {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Always returns true
|
|
||||||
func defaultBrowser(dest url.URL) bool {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
// Never returns true
|
|
||||||
func matchNever(dest url.URL) bool {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func handleURL(browserRules []BrowserRule, urlString string) error {
|
func handleURL(browserRules []BrowserRule, urlString string) error {
|
||||||
dest, err := url.Parse(urlString)
|
dest, err := url.Parse(urlString)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
151
ruler.go
Normal file
151
ruler.go
Normal file
@ -0,0 +1,151 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/url"
|
||||||
|
"os/exec"
|
||||||
|
"regexp"
|
||||||
|
|
||||||
|
"github.com/antonmedv/expr"
|
||||||
|
)
|
||||||
|
|
||||||
|
// MatchFunc is a signature for a function to match a URL
|
||||||
|
type MatchFunc = func(url.URL) bool
|
||||||
|
|
||||||
|
// A BrowserRule is a rule that will launch a browser, if matched
|
||||||
|
type BrowserRule struct {
|
||||||
|
matcher MatchFunc
|
||||||
|
command string
|
||||||
|
args []string
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsMatched will check to see the rule matches the provied URL
|
||||||
|
func (r BrowserRule) IsMatched(dest url.URL) bool {
|
||||||
|
return r.matcher(dest)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Launch will launch the browser with the provided URL
|
||||||
|
func (r BrowserRule) Launch(dest url.URL) error {
|
||||||
|
args := append(r.args, "--", dest.String())
|
||||||
|
cmd := exec.Command(r.command, args...)
|
||||||
|
return cmd.Start()
|
||||||
|
}
|
||||||
|
|
||||||
|
// MaybeLaunch will lauch the browser with the provided URL if it matches the rule
|
||||||
|
func (r BrowserRule) MaybeLaunch(dest url.URL) (bool, error) {
|
||||||
|
if r.IsMatched(dest) {
|
||||||
|
return true, r.Launch(dest)
|
||||||
|
}
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewBrowserRule(command string, args []string, matcher MatchFunc) BrowserRule {
|
||||||
|
return BrowserRule{
|
||||||
|
command: command,
|
||||||
|
args: args,
|
||||||
|
matcher: matcher,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewBrowserRuleFromExpr(command string, args []string, matcherExpr string) BrowserRule {
|
||||||
|
return NewBrowserRule(
|
||||||
|
command, args, MakeMatchFunc(matcherExpr),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
func MakeMatchFunc(rule string) MatchFunc {
|
||||||
|
return func(dest url.URL) bool {
|
||||||
|
env := map[string]interface{}{
|
||||||
|
// Prebuilt match functions
|
||||||
|
"matchAny": matchAny,
|
||||||
|
"matchHostRegexp": matchHostRegexp,
|
||||||
|
"matchHostname": matchHostname,
|
||||||
|
"matchNever": matchNever,
|
||||||
|
"matchRegexp": matchRegexp,
|
||||||
|
|
||||||
|
// Helpers for building custom matchers
|
||||||
|
"MatchFunc": func(foundMatch bool) MatchFunc {
|
||||||
|
return func(_ url.URL) bool {
|
||||||
|
return foundMatch
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fullUrl": dest.String(),
|
||||||
|
"hostname": dest.Hostname(),
|
||||||
|
"url": dest,
|
||||||
|
}
|
||||||
|
matchFuncExpr, err := expr.Eval(rule, env)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error evaluating rule %s: %v", rule, err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
matchFunc, ok := matchFuncExpr.(MatchFunc)
|
||||||
|
if !ok {
|
||||||
|
fmt.Printf("Error evaluating rule %s. Did not evaluate to a MatchFunc.", rule)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return matchFunc(dest)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func matchHostname(hostnames ...string) MatchFunc {
|
||||||
|
return func(dest url.URL) bool {
|
||||||
|
for _, host := range hostnames {
|
||||||
|
if host == dest.Hostname() {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func matchHostRegexp(hostRegexp ...string) MatchFunc {
|
||||||
|
matchers := []*regexp.Regexp{}
|
||||||
|
for _, s := range hostRegexp {
|
||||||
|
matchers = append(matchers, regexp.MustCompile(s))
|
||||||
|
}
|
||||||
|
return func(dest url.URL) bool {
|
||||||
|
for _, matcher := range matchers {
|
||||||
|
if matcher.MatchString(dest.Hostname()) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func matchRegexp(urlRegexp ...string) MatchFunc {
|
||||||
|
matchers := []*regexp.Regexp{}
|
||||||
|
for _, s := range urlRegexp {
|
||||||
|
matchers = append(matchers, regexp.MustCompile(s))
|
||||||
|
}
|
||||||
|
return func(dest url.URL) bool {
|
||||||
|
for _, matcher := range matchers {
|
||||||
|
if matcher.MatchString(dest.String()) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func matchAny(matchFuncs ...MatchFunc) MatchFunc {
|
||||||
|
return func(dest url.URL) bool {
|
||||||
|
for _, f := range matchFuncs {
|
||||||
|
if f(dest) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Always returns true
|
||||||
|
func defaultBrowser(dest url.URL) bool {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// Never returns true
|
||||||
|
func matchNever(dest url.URL) bool {
|
||||||
|
return false
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user