summaryrefslogtreecommitdiff
path: root/main.go
blob: 3f5efd655aca8ed5fa5a1620513ab470bdeac6f1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package main

import (
	"bufio"
	"flag"
	"fmt"
	"log"
	"math/rand"
	"os"
	"strconv"
	"strings"
)

type random interface {
	Intn(n int) int
}

type generator struct {
	words      int
	separator  string
	titleWords bool
	addNumber  bool
	dict       []string
}

func (g *generator) loadWords(fname string) error {
	fd, err := os.Open(fname)
	if err != nil {
		return err
	}
	defer fd.Close()
	scanner := bufio.NewScanner(fd)
	for scanner.Scan() {
		g.dict = append(g.dict, scanner.Text())
	}
	return scanner.Err()
}

func (g generator) pick(rnd random) string {
	word := g.dict[rnd.Intn(len(g.dict))]
	if g.titleWords {
		return strings.Title(word)
	}
	return word
}

func (g generator) generate(rnd random) string {
	var v []string
	for i := 0; i < g.words; i++ {
		v = append(v, g.pick(rnd))
	}
	if g.addNumber {
		v = append(v, strconv.Itoa(rnd.Intn(10)))
	}
	return strings.Join(v, g.separator)
}

func main() {
	var g generator
	flag.IntVar(&g.words, "words", 6, "min words")
	flag.StringVar(&g.separator, "separator", "-", "word separator")
	flag.BoolVar(&g.titleWords, "title", false, "make first letter uppercase")
	flag.BoolVar(&g.addNumber, "number", false, "append random number to the end (0 - 9)")
	flag.Parse()
	if err := g.loadWords("data/wordlist.txt"); err != nil {
		log.Fatal(err)
	}
	fmt.Println(g.generate(rand.New(cryptoSource{})))
}