From fd2a353770a5f3cea773a5c4465bf0fc2a2818d3 Mon Sep 17 00:00:00 2001 From: r Date: Sun, 24 May 2020 04:38:34 +0000 Subject: Refactor things --- migrations/csrfToken/main.go | 85 -------------------------------------------- 1 file changed, 85 deletions(-) delete mode 100644 migrations/csrfToken/main.go (limited to 'migrations/csrfToken') diff --git a/migrations/csrfToken/main.go b/migrations/csrfToken/main.go deleted file mode 100644 index 3d9526d..0000000 --- a/migrations/csrfToken/main.go +++ /dev/null @@ -1,85 +0,0 @@ -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.Println(id, err) - continue - } - s.CSRFToken, err = util.NewCSRFToken() - if err != nil { - log.Println(id, err) - continue - } - err = sessionRepo.Add(s) - if err != nil { - log.Println(id, err) - continue - } - } - -} -- cgit v1.2.3