package utils

import (
	"context"
	"google.golang.org/grpc/metadata"
)

type UserSession struct {
	USN          int64  `json:"usn" redis:"usn"`
	IP           string `json:"ip" redis:"ip"`
	UserAgent    string `json:"ua" redis:"ua"`
	AccessToken  string `json:"at" redis:"at"`
	RefreshToken string `json:"rt" redis:"rt"`
}

func (us *UserSession) GetUsnKey() string {
	return "usn"
}

func ShouldBindUsn(ctx context.Context, usn *int64) bool {
	if md, ok := metadata.FromIncomingContext(ctx); ok {
		usnArr := md.Get("X-Usn")
		if len(usnArr) == 0 || usnArr[0] == "" {
			return false
		}
		*usn = StringToInt64(usnArr[0])
	}
	return *usn != 0
}