all: support alias import from file

Author: changkun

import.yml

@@ -0,0 +1,2 @@
+---
+changkun: https://changkun.de/

redir.go

@@ -15,10 +15,11 @@ import (
 )
 
 var (
-	daemon  = flag.Bool("s", false, "run redir service")
-	operate = flag.String("op", "create", "operators, create/update/delete/fetch")
-	alias   = flag.String("a", "", "alias for a new link")
-	link    = flag.String("l", "", "actual link for the alias, optional for delete/fetch")
+	daemon   = flag.Bool("s", false, "run redir service")
+	fromfile = flag.String("f", "", "import aliases from a YAML file")
+	operate  = flag.String("op", "create", "operators, create/update/delete/fetch")
+	alias    = flag.String("a", "", "alias for a new link")
+	link     = flag.String("l", "", "actual link for the alias, optional for delete/fetch")
 )
 
 func main() {
@@ -27,6 +28,11 @@ func main() {
 	flag.Usage = usage
 	flag.Parse()
 
+	if len(os.Args) < 2 {
+		flag.Usage()
+		return
+	}
+
 	if daemon == nil {
 		flag.Usage()
 		return
@@ -48,19 +54,31 @@ func processServer() {
 }
 
 func usage() {
-	fmt.Fprintf(os.Stderr, `usage: redir [-s] [-op <operator> -a <alias> -l <link>]
+	fmt.Fprintf(os.Stderr, `usage: redir [-s] [-f <file>] [-op <operator> -a <alias> -l <link>]
 options:
 `)
 	flag.PrintDefaults()
 	fmt.Fprintf(os.Stderr, `example:
 redir -s                  run the redir service
+redir -f ./import.yml       import aliases from a file
 redir -a alias -l link    allocate new short link if possible
 redir -op fetch -a alias  fetch alias information
 `)
 	os.Exit(2)
 }
 
 func processCmd() {
+	if fromfile != nil {
+		fname := *fromfile
+		if fname == "" {
+			flag.Usage()
+			return
+		}
+
+		shortFile(fname)
+		return
+	}
+
 	if operate == nil || !op(*operate).valid() {
 		flag.Usage()
 		return

short.go

@@ -10,12 +10,15 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"io/ioutil"
 	"log"
 	"net"
 	"net/http"
 	"strings"
+	"time"
 
 	"github.com/go-redis/redis/v8"
+	"gopkg.in/yaml.v3"
 )
 
 // op is a short link operator
@@ -41,6 +44,31 @@ func (o op) valid() bool {
 	}
 }
 
+func shortFile(fname string) {
+	b, err := ioutil.ReadFile(fname)
+	if err != nil {
+		log.Fatalf("cannot read import file, err: %v\n", err)
+	}
+
+	d := map[string]string{}
+	err = yaml.Unmarshal(b, &d)
+	if err != nil {
+		log.Fatalf("cannot unmarshal the imported file, err: %v\n", err)
+	}
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second*20)
+	defer cancel()
+	for alias, link := range d {
+		err = shortCmd(ctx, opUpdate, alias, link)
+		if err != nil {
+			err = shortCmd(ctx, opCreate, alias, link)
+			if err != nil {
+				log.Fatalf("cannot import alias %v, err: %v\n", alias, err)
+			}
+		}
+	}
+	return
+}
+
 // shortCmd processes the given alias and link with a specified op.
 func shortCmd(ctx context.Context, operate op, alias, link string) (retErr error) {
 	s, err := newStore(conf.Store)