[new] the first version

Author: Huweicai

.gitignore

@@ -0,0 +1 @@
+.idea

config.yaml

@@ -0,0 +1,3 @@
+Example:
+  "@": "http://www.baidu.com"
+  example: "http://example.com"

config/config.go

@@ -0,0 +1,183 @@
+package config
+
+import (
+	"gopkg.in/yaml.v2"
+	"io/ioutil"
+	"os"
+	"reflect"
+	"strings"
+)
+
+type ModelType int
+
+const (
+	configName  = "config.yaml"
+	LoopBackKey = "@"
+
+	TypeUnknown ModelType = iota
+	TypeVector
+	TypeScalar
+)
+
+type Nest struct {
+	Data map[string]interface{}
+}
+
+func GetConfigPath() string {
+	defaultPath := "./" + configName
+	wd, err := os.Getwd()
+	if err != nil {
+		return defaultPath
+	}
+	if strings.HasSuffix(wd, "goto") {
+		return "./" + configName
+	} else if strings.HasSuffix(wd, "handler") || strings.HasSuffix(wd, "config") {
+		return "../" + configName
+	}
+	return defaultPath
+}
+
+func NewNest(configPath string) (*Nest, error) {
+	conf, err := loadYaml(configPath)
+	if err != nil {
+		return nil, err
+	}
+	return &Nest{Data: conf}, nil
+}
+
+//载入配置
+func loadYaml(path string) (conf map[string]interface{}, err error) {
+	f, err := ioutil.ReadFile(path)
+	if err != nil {
+		return nil, err
+	}
+	m := make(map[interface{}]interface{})
+	err = yaml.Unmarshal(f, &m)
+	//to lower to compare
+	if err != nil {
+		return
+	}
+	conf = toMapStringInterface(m)
+	toLower(conf)
+	return
+}
+
+func writeYaml(path string, target interface{}) error {
+	out, err := yaml.Marshal(target)
+	if err != nil {
+		return err
+	}
+	return ioutil.WriteFile(path, out, 0644)
+}
+
+func (n *Nest) GetScalar(paths []string) (URL string, ok bool) {
+	return getScalar(paths, n.Data)
+}
+
+func (n *Nest) Flush() error {
+	return writeYaml(GetConfigPath(), n.Data)
+}
+
+func (n *Nest) AddScalar(paths []string, url string) {
+	addScalar(n.Data, paths, url)
+}
+
+func addScalar(m map[string]interface{}, paths []string, url string) map[string]interface{} {
+	if len(paths) == 0 {
+		return nil
+	}
+	curK := paths[0]
+	curV := m[curK]
+	//find scalar key
+	if len(paths) == 1 {
+		if curV != nil && typeOf(curV) == TypeVector {
+			k := m[curK]
+			k.(map[string]interface{})[LoopBackKey] = url
+		} else {
+			m[curK] = url
+		}
+		return m
+	}
+
+	if curV == nil {
+		m[curK] = addScalar(make(map[string]interface{}), paths[1:], url)
+		return m
+	}
+
+	if typeOf(curV) == TypeVector {
+		m[curK] = addScalar(curV.(map[string]interface{}), paths[1:], url)
+	}
+
+	if typeOf(curV) == TypeScalar {
+		x := make(map[string]interface{})
+		x[LoopBackKey] = curV
+		m[curK] = addScalar(x, paths[1:], url)
+	}
+	return m
+}
+
+func toMapStringInterface(m map[interface{}]interface{}) map[string]interface{} {
+	nm := make(map[string]interface{})
+	for k, v := range m {
+		if reflect.TypeOf(v).Kind() == reflect.Map {
+			v = toMapStringInterface(v.(map[interface{}]interface{}))
+		}
+		nm[k.(string)] = v
+	}
+	return nm
+}
+
+func toLower(m map[string]interface{}) map[string]interface{} {
+	for k, v := range m {
+		delete(m, k)
+		switch typeOf(v) {
+		case TypeScalar:
+			v = strings.ToLower(v.(string))
+		case TypeVector:
+			v = toLower(v.(map[string]interface{}))
+		default:
+			continue
+		}
+		k = strings.ToLower(k)
+		m[k] = v
+	}
+	return m
+}
+
+//get the specified URL
+func getScalar(paths []string, m map[string]interface{}) (scalar string, ok bool) {
+	if len(paths) == 0 {
+		return
+	}
+	value, got := m[paths[0]]
+	if !got {
+		return
+	}
+	switch typeOf(value) {
+	case TypeScalar:
+		return value.(string), true
+	case TypeVector:
+		if len(paths) == 1 {
+			v, ok := value.(map[string]interface{})[LoopBackKey]
+			if ok {
+				return v.(string), true
+			} else {
+				return "", false
+			}
+		}
+		return getScalar(paths[1:], value.(map[string]interface{}))
+	default:
+		return
+	}
+}
+
+func typeOf(i interface{}) ModelType {
+	switch reflect.TypeOf(i).Kind() {
+	case reflect.Map:
+		return TypeVector
+	case reflect.String:
+		return TypeScalar
+	default:
+		return TypeUnknown
+	}
+}

config/config_test.go

@@ -0,0 +1,71 @@
+package config
+
+import (
+	"github.com/stretchr/testify/require"
+	"testing"
+)
+
+func Test_typeOf(t *testing.T) {
+	type args struct {
+		i interface{}
+	}
+	tests := []struct {
+		name string
+		args args
+		want ModelType
+	}{
+		{
+			args: args{make(map[string]interface{})},
+			want: TypeVector,
+		}, {
+			want: TypeUnknown,
+		}, {
+			args: args{""},
+			want: TypeScalar,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := typeOf(tt.args.i); got != tt.want {
+				t.Errorf("typeOf() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestNest_GetURL(t *testing.T) {
+	nest, err := NewNest("../config.yaml")
+	if err != nil {
+		t.Fatalf(err.Error())
+	}
+	u, _ := nest.GetScalar([]string{"tce", "stream"})
+	print(u)
+}
+
+func Test_toLower(t *testing.T) {
+	assert := require.New(t)
+	m := map[string]interface{}{
+		"A": "A",
+		"b": "B",
+		"C": "c",
+		"D": map[string]interface{}{
+			"E": "e",
+			"f": "F",
+		},
+	}
+	assert.NotPanics(func() {
+		toLower(m)
+		assert.Equal("a", m["a"])
+		assert.Equal("f", m["d"].(map[string]interface{})["f"])
+	})
+}
+
+func TestNest_AddScalar(t *testing.T) {
+	nest, err := NewNest("../config.yaml")
+	if err != nil {
+		t.Fatalf(err.Error())
+	}
+	nest.AddScalar([]string{"tce", "stream", "a"}, "http://www.baidu.com")
+	nest.Flush()
+	println(nest.Data)
+}

go.mod

@@ -0,0 +1,8 @@
+module github.com/Huweicai/goto
+
+require (
+	github.com/davecgh/go-spew v1.1.1 // indirect
+	github.com/stretchr/objx v0.1.1 // indirect
+	github.com/stretchr/testify v1.3.0
+	gopkg.in/yaml.v2 v2.2.2
+)

go.sum

@@ -0,0 +1,13 @@
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.1.1 h1:2vfRuCMp5sSVIDSqO8oNnWJq7mPa6KVP3iPIwFBuy8A=
+github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.3.0 h1:TivCn/peBQ7UY8ooIcPgZFpTNSz0Q2U6UrFlUfqbe0Q=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v2 v2.2.2 h1:ZCJp+EgiOT7lHqUV2J862kp8Qj64Jo6az82+3Td9dZw=
+gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=

handler/add.go

@@ -0,0 +1,16 @@
+package handler
+
+import (
+	"github.com/Huweicai/goto/config"
+	"log"
+)
+
+func Add(args []string) {
+	nest, err := config.NewNest(config.GetConfigPath())
+	if err != nil {
+		log.Fatalf(err.Error())
+		return
+	}
+	nest.AddScalar(args[:len(args)-1], args[len(args)-1])
+	nest.Flush()
+}

handler/common.go

@@ -0,0 +1,15 @@
+package handler
+
+const (
+	add = "add"
+	get = "get"
+)
+
+var handlers = map[string]func(args []string){
+	add: Add,
+	get: Get,
+}
+
+func GetHandler(name string) func(args []string) {
+	return handlers[name]
+}

handler/common_test.go

@@ -0,0 +1,12 @@
+package handler
+
+import (
+	"os"
+	"testing"
+)
+
+func TestAdd(t *testing.T) {
+	s, _ := os.Getwd()
+	print(s)
+
+}

handler/get.go

@@ -0,0 +1,25 @@
+package handler
+
+import (
+	"github.com/Huweicai/goto/config"
+	"log"
+	"os/exec"
+)
+
+func Get(args []string) {
+	nest, err := config.NewNest("./config.yaml")
+	if err != nil {
+		log.Fatalf("init nest failed err:%s", err.Error())
+		return
+	}
+	url, ok := nest.GetScalar(args)
+	if !ok {
+		return
+	}
+	cmd := exec.Command("open", url)
+	err = cmd.Run()
+	if err != nil {
+		log.Fatal(err.Error())
+
+	}
+}