tricrypt/cmd/server/users.go

package main

import (
	"errors"
	"log"
	"math/rand"
	"sync"
)

var (
	userCenter UserCenter
)

type UserCenter struct {
	UsersSTOI map[string]uint16
	UsersITOS map[uint16]string
	Mu        sync.Mutex
}

func (u *UserCenter) Init() {
	u.UsersSTOI = make(map[string]uint16)
	u.UsersITOS = make(map[uint16]string)
}

var (
	ERROR_ALREADY_HAVE = errors.New("Username is already taken")
	ERROR_DONT_HAVE    = errors.New("Username was not found")
)

func (u *UserCenter) AddUser(name string) (uint16, error) {
	u.Mu.Lock()
	defer u.Mu.Unlock()
	_, alreadyHave := u.UsersSTOI[name]
	if alreadyHave {
		return 0, ERROR_ALREADY_HAVE
	}
	var ret uint16
	for {
		rInt := rand.Int31()
		_, taken := u.UsersITOS[uint16(rInt)]
		if !taken {
			ret = uint16(rInt)
			break
		}
	}
	log.Printf("Users: add %s with %d id\n", name, ret)
	u.UsersITOS[ret] = name
	u.UsersSTOI[name] = ret
	return ret, nil
}

func (u *UserCenter) DeleteIfHaveOne(id uint16) {
	name, found := u.UsersITOS[id]
	if !found {
		log.Printf("User with %d id is not found; Can not delete\n", id)
		return
	}
	delete(u.UsersITOS, id)
	delete(u.UsersSTOI, name)
	log.Printf("User with %s name and %d id was found; User is deleted\n", name, id)
}

func (u *UserCenter) GetID(name string) (uint16, error) {
	id, have := u.UsersSTOI[name]
	if !have {
		return 0, ERROR_DONT_HAVE
	}
	return id, nil
}

func (u *UserCenter) GetName(id uint16) (string, error) {
	name, have := u.UsersITOS[id]
	if !have {
		return "", ERROR_DONT_HAVE
	}
	return name, nil
}