fin-check-api/tokens/dispatcher.go

package tokens

import (
	"crypto/rand"
	"encoding/base64"
	"errors"
	"log"
	"strings"
	"sync"
	"time"
)

type Token struct {
	Id         uint
	Val        string
	LastActive time.Time
}

var (
	ActiveDur = time.Duration(time.Hour)
)

func (t Token) IsExpired() bool {
	return time.Now().Sub(t.LastActive) >= ActiveDur
}

type TokensMapMu struct {
	Initialized bool
	Tokmap      map[uint]*Token
	TokmapRev   map[string]*Token
	Mu          sync.RWMutex
}

var toks TokensMapMu

// NOTE: should be launch with a goroutine
// NOTE: it cannot die
func StartTokens() {
	if toks.Initialized {
		return
	}
	toks.Tokmap = make(map[uint]*Token)
	toks.TokmapRev = make(map[string]*Token)
	toks.Initialized = true
	for {
		//
		toks.Mu.Lock()
		for id, token := range toks.Tokmap {
			if token == nil {
				log.Printf("DAFUQ: 001\n")
				delete(toks.Tokmap, id)
				continue
			}
			if token.IsExpired() {
				val := token.Val
				delete(toks.Tokmap, id)
				delete(toks.TokmapRev, val)
			}
		}
		toks.Mu.Unlock()
		//
		time.Sleep(time.Minute)
	}
}

var (
	ERROR_DONT_HAVE_TOKEN    = errors.New("Don't have token for this user")
	ERROR_ALREADY_HAVE_TOKEN = errors.New("Already have token")
)

func GetToken(id uint) (*Token, error) {
	toks.Mu.Lock()
	defer toks.Mu.Unlock()
	val, exists := toks.Tokmap[id]
	if !exists {
		return nil, ERROR_DONT_HAVE_TOKEN
	}
	val.LastActive = time.Now()
	return val, nil
}

func GetID(token string) (uint, error) {
	toks.Mu.RLock()
	val, exists := toks.TokmapRev[token]
	toks.Mu.RUnlock()
	if !exists {
		return 0, ERROR_DONT_HAVE_TOKEN
	}
	return val.Id, nil
}

func haveToken(id uint) bool {
	toks.Mu.RLock()
	_, exists := toks.Tokmap[id]
	toks.Mu.RUnlock()
	return exists
}

func UpdateLastActive(id uint) error {
	if !haveToken(id) {
		return ERROR_DONT_HAVE_TOKEN
	}
	toks.Mu.Lock()
	val := toks.Tokmap[id]
	val.LastActive = time.Now()
	toks.Tokmap[id] = val
	toks.Mu.Unlock()
	return nil
}

func haveTokenVal(val string) bool {
	toks.Mu.RLock()
	_, exists := toks.TokmapRev[val]
	toks.Mu.RUnlock()
	return exists
}

func generateRandomString(length int) string {
	bytes := make([]byte, length)
	if _, err := rand.Read(bytes); err != nil {
		log.Printf("generateRandomString: %v", err)
	}
	return base64.URLEncoding.EncodeToString(bytes)
}

func generateTokenVal() string {
	for {
		tok := generateRandomString(32)
		trimedToken := strings.Trim(tok, "=")
		if !haveTokenVal(trimedToken) {
			return trimedToken
		}
	}
}

func AddToken(id uint) (*Token, error) {
	toks.Mu.RLock()
	_, exists := toks.Tokmap[id]
	toks.Mu.RUnlock()
	if exists {
		// return nil, ERROR_ALREADY_HAVE_TOKEN
	}
	val := generateTokenVal()
	token := &Token{
		Id:         id,
		Val:        val,
		LastActive: time.Now(),
	}
	toks.Mu.Lock()
	toks.Tokmap[id] = token
	toks.TokmapRev[val] = token
	toks.Mu.Unlock()
	return token, nil
}

func AmIAllowed(token string) bool {
	toks.Mu.Lock()
	defer toks.Mu.Unlock()
	val, exists := toks.TokmapRev[token]
	if !exists {
		return false
	}
	val.LastActive = time.Now()
	return true
}