package main import ( "context" "crypto/sha1" "flag" "fmt" "os" "path" "git.defalsify.org/vise.git/logging" "git.grassecon.net/urdt/ussd/config" "git.grassecon.net/urdt/ussd/internal/storage" "git.grassecon.net/urdt/ussd/common" ) var ( logg = logging.NewVanilla() scriptDir = path.Join("services", "registration") ) func main() { config.LoadConfig() var dbDir string var sessionId string var database string var engineDebug bool flag.StringVar(&sessionId, "session-id", "075xx2123", "session id") flag.StringVar(&database, "db", "gdbm", "database to be used") flag.StringVar(&dbDir, "dbdir", ".state", "database dir to read from") flag.BoolVar(&engineDebug, "d", false, "use engine debug output") flag.Parse() ctx := context.Background() ctx = context.WithValue(ctx, "SessionId", sessionId) ctx = context.WithValue(ctx, "Database", database) resourceDir := scriptDir menuStorageService := storage.NewMenuStorageService(dbDir, resourceDir) store, err := menuStorageService.GetUserdataDb(ctx) if err != nil { fmt.Fprintf(os.Stderr, err.Error()) os.Exit(1) } userStore := common.UserDataStore{store} h := sha1.New() h.Write([]byte(sessionId)) address := h.Sum(nil) addressString := fmt.Sprintf("%x", address) err = userStore.WriteEntry(ctx, sessionId, common.DATA_PUBLIC_KEY, []byte(addressString)) if err != nil { fmt.Fprintf(os.Stderr, err.Error()) os.Exit(1) } err = userStore.WriteEntry(ctx, addressString, common.DATA_PUBLIC_KEY_REVERSE, []byte(sessionId)) if err != nil { fmt.Fprintf(os.Stderr, err.Error()) os.Exit(1) } err = store.Close() if err != nil { fmt.Fprintf(os.Stderr, err.Error()) os.Exit(1) } }