aboutsummaryrefslogtreecommitdiff
path: root/migrations/csrfToken/main.go
blob: f2326df2a0d202584423a51a46f764eacbcde2f8 (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
70
71
72
73
74
75
76
77
78
79
80
81
82
package main

import (
	"log"
	"math/rand"
	"os"
	"path/filepath"
	"time"

	"bloat/config"
	"bloat/kv"
	"bloat/repository"
	"bloat/util"
)

var (
	configFile = "bloat.conf"
)

func init() {
	rand.Seed(time.Now().Unix())
}

func getKeys(sessionRepoPath string) (keys []string, err error) {
	f, err := os.Open(sessionRepoPath)
	if err != nil {
		return
	}
	return f.Readdirnames(0)
}

func main() {
	opts, _, err := util.Getopts(os.Args, "f:")
	if err != nil {
		log.Fatal(err)
	}

	for _, opt := range opts {
		switch opt.Option {
		case 'f':
			configFile = opt.Value
		}
	}

	config, err := config.ParseFile(configFile)
	if err != nil {
		log.Fatal(err)
	}

	if !config.IsValid() {
		log.Fatal("invalid config")
	}

	sessionRepoPath := filepath.Join(config.DatabasePath, "session")
	sessionDB, err := kv.NewDatabse(sessionRepoPath)
	if err != nil {
		log.Fatal(err)
	}

	sessionRepo := repository.NewSessionRepository(sessionDB)

	sessionIds, err := getKeys(sessionRepoPath)
	if err != nil {
		log.Fatal(err)
	}

	for _, id := range sessionIds {
		s, err := sessionRepo.Get(id)
		if err != nil {
			log.Fatal(err)
		}
		s.CSRFToken, err = util.NewCSRFToken()
		if err != nil {
			log.Fatal(err)
		}
		err = sessionRepo.Add(s)
		if err != nil {
			log.Fatal(err)
		}
	}

}