refactor nkode-core

2025-01-21 01:19:27 -06:00
parent 3ed12cee68
commit c5e95239b5
46 changed files with 318 additions and 434 deletions
--- a/pkg/nkode-core/api/nkode_api.go
+++ b/pkg/nkode-core/api/nkode_api.go
@@ -0,0 +1,266 @@
+package api
+
+import (
+	"fmt"
+	"github.com/google/uuid"
+	"github.com/patrickmn/go-cache"
+	"go-nkode/config"
+	"go-nkode/pkg/nkode-core/email"
+	"go-nkode/pkg/nkode-core/entities"
+	"go-nkode/pkg/nkode-core/repository"
+	"go-nkode/pkg/nkode-core/security"
+	"log"
+	"os"
+	"time"
+)
+
+const (
+	sessionExpiration      = 5 * time.Minute
+	sessionCleanupInterval = 10 * time.Minute
+)
+
+type NKodeAPI struct {
+	Db                 repository.CustomerUserRepository
+	SignupSessionCache *cache.Cache
+	EmailQueue         *email.Queue
+}
+
+func NewNKodeAPI(db repository.CustomerUserRepository, queue *email.Queue) NKodeAPI {
+	return NKodeAPI{
+		Db:                 db,
+		EmailQueue:         queue,
+		SignupSessionCache: cache.New(sessionExpiration, sessionCleanupInterval),
+	}
+}
+
+func (n *NKodeAPI) CreateNewCustomer(nkodePolicy entities.NKodePolicy, id *entities.CustomerId) (*entities.CustomerId, error) {
+	newCustomer, err := entities.NewCustomer(nkodePolicy)
+	if id != nil {
+		newCustomer.Id = *id
+	}
+	if err != nil {
+		return nil, err
+	}
+	err = n.Db.CreateCustomer(*newCustomer)
+
+	if err != nil {
+		return nil, err
+	}
+	return &newCustomer.Id, nil
+}
+
+func (n *NKodeAPI) GenerateSignupResetInterface(userEmail entities.UserEmail, customerId entities.CustomerId, kp entities.KeypadDimension, reset bool) (*entities.SignupResetInterface, error) {
+	user, err := n.Db.GetUser(userEmail, customerId)
+	if err != nil {
+		return nil, err
+	}
+	if user != nil && !reset {
+		log.Printf("user %s already exists", string(userEmail))
+		return nil, config.ErrUserAlreadyExists
+	}
+	svgIdxInterface, err := n.Db.RandomSvgIdxInterface(kp)
+	if err != nil {
+		return nil, err
+	}
+	signupSession, err := entities.NewSignupResetSession(userEmail, kp, customerId, svgIdxInterface, reset)
+	if err != nil {
+		return nil, err
+	}
+	if err := n.SignupSessionCache.Add(signupSession.Id.String(), *signupSession, sessionExpiration); err != nil {
+		return nil, err
+	}
+	svgInterface, err := n.Db.GetSvgStringInterface(signupSession.LoginUserInterface.SvgId)
+
+	if err != nil {
+		return nil, err
+	}
+	resp := entities.SignupResetInterface{
+		UserIdxInterface: signupSession.SetIdxInterface,
+		SvgInterface:     svgInterface,
+		SessionId:        uuid.UUID(signupSession.Id).String(),
+		Colors:           signupSession.Colors,
+	}
+	return &resp, nil
+}
+
+func (n *NKodeAPI) SetNKode(customerId entities.CustomerId, sessionId entities.SessionId, keySelection entities.KeySelection) (entities.IdxInterface, error) {
+	_, err := n.Db.GetCustomer(customerId)
+
+	if err != nil {
+		return nil, err
+	}
+	session, exists := n.SignupSessionCache.Get(sessionId.String())
+	if !exists {
+		log.Printf("session id does not exist %s", sessionId)
+		return nil, config.ErrSignupSessionDNE
+	}
+	userSession, ok := session.(entities.UserSignSession)
+	if !ok {
+		// handle the case where the type assertion fails
+		return nil, config.ErrSignupSessionDNE
+	}
+	confirmInterface, err := userSession.SetUserNKode(keySelection)
+	if err != nil {
+		return nil, err
+	}
+	n.SignupSessionCache.Set(sessionId.String(), userSession, sessionExpiration)
+	return confirmInterface, nil
+}
+
+func (n *NKodeAPI) ConfirmNKode(customerId entities.CustomerId, sessionId entities.SessionId, keySelection entities.KeySelection) error {
+	session, exists := n.SignupSessionCache.Get(sessionId.String())
+	if !exists {
+		log.Printf("session id does not exist %s", sessionId)
+		return config.ErrSignupSessionDNE
+	}
+	userSession, ok := session.(entities.UserSignSession)
+	if !ok {
+		// handle the case where the type assertion fails
+		return config.ErrSignupSessionDNE
+	}
+	customer, err := n.Db.GetCustomer(customerId)
+	if err != nil {
+		return err
+	}
+	passcode, err := userSession.DeducePasscode(keySelection)
+	if err != nil {
+		return err
+	}
+	if err = customer.IsValidNKode(userSession.Kp, passcode); err != nil {
+		return err
+	}
+	user, err := entities.NewUser(*customer, string(userSession.UserEmail), passcode, userSession.LoginUserInterface, userSession.Kp)
+	if err != nil {
+		return err
+	}
+	if userSession.Reset {
+		err = n.Db.UpdateUserNKode(*user)
+	} else {
+		err = n.Db.WriteNewUser(*user)
+	}
+	n.SignupSessionCache.Delete(userSession.Id.String())
+	return err
+}
+
+func (n *NKodeAPI) GetLoginInterface(userEmail entities.UserEmail, customerId entities.CustomerId) (*entities.LoginInterface, error) {
+	user, err := n.Db.GetUser(userEmail, customerId)
+	if err != nil {
+		return nil, err
+	}
+	if user == nil {
+		log.Printf("user %s for customer %s dne", userEmail, customerId)
+		return nil, config.ErrUserForCustomerDNE
+	}
+	svgInterface, err := n.Db.GetSvgStringInterface(user.Interface.SvgId)
+	if err != nil {
+		return nil, err
+	}
+	resp := entities.LoginInterface{
+		UserIdxInterface: user.Interface.IdxInterface,
+		SvgInterface:     svgInterface,
+		NumbOfKeys:       user.Kp.NumbOfKeys,
+		AttrsPerKey:      user.Kp.AttrsPerKey,
+		Colors:           entities.SetColors,
+	}
+	return &resp, nil
+}
+
+func (n *NKodeAPI) Login(customerId entities.CustomerId, userEmail entities.UserEmail, keySelection entities.KeySelection) (*security.AuthenticationTokens, error) {
+	customer, err := n.Db.GetCustomer(customerId)
+	if err != nil {
+		return nil, err
+	}
+	user, err := n.Db.GetUser(userEmail, customerId)
+	if err != nil {
+		return nil, err
+	}
+	if user == nil {
+		log.Printf("user %s for customer %s dne", userEmail, customerId)
+		return nil, config.ErrUserForCustomerDNE
+	}
+	passcode, err := entities.ValidKeyEntry(*user, *customer, keySelection)
+	if err != nil {
+		return nil, err
+	}
+
+	if user.Renew {
+		err = n.Db.RefreshUserPasscode(*user, passcode, customer.Attributes)
+		if err != nil {
+			return nil, err
+		}
+	}
+	jwtToken, err := security.NewAuthenticationTokens(string(user.Email), uuid.UUID(customerId))
+	if err != nil {
+		return nil, err
+	}
+	if err = n.Db.UpdateUserRefreshToken(user.Id, jwtToken.RefreshToken); err != nil {
+		return nil, err
+	}
+	if err = user.Interface.LoginShuffle(); err != nil {
+		return nil, err
+	}
+	if err = n.Db.UpdateUserInterface(user.Id, user.Interface); err != nil {
+		return nil, err
+	}
+	return &jwtToken, nil
+}
+
+func (n *NKodeAPI) RenewAttributes(customerId entities.CustomerId) error {
+	return n.Db.Renew(customerId)
+}
+
+func (n *NKodeAPI) RandomSvgInterface() ([]string, error) {
+	return n.Db.RandomSvgInterface(entities.KeypadMax)
+}
+
+func (n *NKodeAPI) RefreshToken(userEmail entities.UserEmail, customerId entities.CustomerId, refreshToken string) (string, error) {
+	user, err := n.Db.GetUser(userEmail, customerId)
+	if err != nil {
+		return "", err
+	}
+	if user == nil {
+		log.Printf("user %s for customer %s dne", userEmail, customerId)
+		return "", config.ErrUserForCustomerDNE
+	}
+	if user.RefreshToken != refreshToken {
+		return "", config.ErrRefreshTokenInvalid
+	}
+	refreshClaims, err := security.ParseRegisteredClaimToken(refreshToken)
+	if err != nil {
+		return "", err
+	}
+	if err = security.ClaimExpired(*refreshClaims); err != nil {
+		return "", err
+	}
+	newAccessClaims := security.NewAccessClaim(string(userEmail), uuid.UUID(customerId))
+	return security.EncodeAndSignClaims(newAccessClaims)
+}
+
+func (n *NKodeAPI) ResetNKode(userEmail entities.UserEmail, customerId entities.CustomerId) error {
+	user, err := n.Db.GetUser(userEmail, customerId)
+	if err != nil {
+		return fmt.Errorf("error getting user in rest nkode %v", err)
+	}
+
+	if user == nil {
+		return nil
+	}
+
+	nkodeResetJwt, err := security.ResetNKodeToken(string(userEmail), uuid.UUID(customerId))
+	if err != nil {
+		return err
+	}
+	frontendHost := os.Getenv("FRONTEND_HOST")
+	if frontendHost == "" {
+		frontendHost = config.FrontendHost
+	}
+	htmlBody := fmt.Sprintf("<h1>Hello!</h1><p>Click the link to reset your nKode.</p><a href=\"%s?token=%s\">Reset nKode</a>", frontendHost, nkodeResetJwt)
+	email := email.Email{
+		Sender:    "no-reply@nkode.tech",
+		Recipient: string(userEmail),
+		Subject:   "nKode Reset",
+		Content:   htmlBody,
+	}
+	n.EmailQueue.AddEmail(email)
+	return nil
+}
--- a/pkg/nkode-core/api/nkode_api_test.go
+++ b/pkg/nkode-core/api/nkode_api_test.go
@@ -0,0 +1,124 @@
+package api
+
+import (
+	"context"
+	"github.com/stretchr/testify/assert"
+	"go-nkode/pkg/nkode-core/email"
+	"go-nkode/pkg/nkode-core/entities"
+	repository2 "go-nkode/pkg/nkode-core/repository"
+	"go-nkode/pkg/nkode-core/security"
+	sqlite_queue "go-nkode/pkg/nkode-core/sqlc"
+	"log"
+	"os"
+	"testing"
+)
+
+func TestNKodeAPI(t *testing.T) {
+	//db1 := NewInMemoryDb()
+	//testNKodeAPI(t, &db1)
+
+	dbPath := os.Getenv("TEST_DB")
+	ctx := context.Background()
+	sqliteDb, err := sqlite_queue.OpenSqliteDb(dbPath)
+	assert.NoError(t, err)
+
+	queue, err := sqlite_queue.NewQueue(sqliteDb, ctx)
+	assert.NoError(t, err)
+	queue.Start()
+	defer func(queue *sqlite_queue.Queue) {
+		if err := queue.Stop(); err != nil {
+			log.Fatal(err)
+		}
+	}(queue)
+	sqlitedb := repository2.NewSqliteRepository(queue, ctx)
+	testNKodeAPI(t, &sqlitedb)
+
+	//if _, err := os.Stat(dbPath); err == nil {
+	//	err = os.Remove(dbPath)
+	//	assert.NoError(t, err)
+	//} else {
+	//	assert.NoError(t, err)
+	//}
+}
+
+func testNKodeAPI(t *testing.T, db repository2.CustomerUserRepository) {
+	bufferSize := 100
+	emailsPerSec := 14
+	testClient := email.TestEmailClient{}
+	queue := email.NewEmailQueue(bufferSize, emailsPerSec, &testClient)
+	queue.Start()
+	defer queue.Stop()
+	attrsPerKey := 5
+	numbOfKeys := 4
+	for idx := 0; idx < 1; idx++ {
+		userEmail := entities.UserEmail("test_username" + security.GenerateRandomString(12) + "@example.com")
+		passcodeLen := 4
+		nkodePolicy := entities.NewDefaultNKodePolicy()
+		keypadSize := entities.KeypadDimension{AttrsPerKey: attrsPerKey, NumbOfKeys: numbOfKeys}
+		nkodeApi := NewNKodeAPI(db, queue)
+		customerId, err := nkodeApi.CreateNewCustomer(nkodePolicy, nil)
+		assert.NoError(t, err)
+		signupResponse, err := nkodeApi.GenerateSignupResetInterface(userEmail, *customerId, keypadSize, false)
+		assert.NoError(t, err)
+		setInterface := signupResponse.UserIdxInterface
+		sessionIdStr := signupResponse.SessionId
+		sessionId, err := entities.SessionIdFromString(sessionIdStr)
+		assert.NoError(t, err)
+		keypadSize = entities.KeypadDimension{AttrsPerKey: numbOfKeys, NumbOfKeys: numbOfKeys}
+		userPasscode := setInterface[:passcodeLen]
+		setKeySelect, err := entities.SelectKeyByAttrIdx(setInterface, userPasscode, keypadSize)
+		assert.NoError(t, err)
+		confirmInterface, err := nkodeApi.SetNKode(*customerId, sessionId, setKeySelect)
+		assert.NoError(t, err)
+		confirmKeySelect, err := entities.SelectKeyByAttrIdx(confirmInterface, userPasscode, keypadSize)
+		err = nkodeApi.ConfirmNKode(*customerId, sessionId, confirmKeySelect)
+		assert.NoError(t, err)
+
+		keypadSize = entities.KeypadDimension{AttrsPerKey: attrsPerKey, NumbOfKeys: numbOfKeys}
+		loginInterface, err := nkodeApi.GetLoginInterface(userEmail, *customerId)
+		assert.NoError(t, err)
+		loginKeySelection, err := entities.SelectKeyByAttrIdx(loginInterface.UserIdxInterface, userPasscode, keypadSize)
+		assert.NoError(t, err)
+		_, err = nkodeApi.Login(*customerId, userEmail, loginKeySelection)
+		assert.NoError(t, err)
+
+		err = nkodeApi.RenewAttributes(*customerId)
+		assert.NoError(t, err)
+
+		loginInterface, err = nkodeApi.GetLoginInterface(userEmail, *customerId)
+		assert.NoError(t, err)
+		loginKeySelection, err = entities.SelectKeyByAttrIdx(loginInterface.UserIdxInterface, userPasscode, keypadSize)
+		assert.NoError(t, err)
+		_, err = nkodeApi.Login(*customerId, userEmail, loginKeySelection)
+		assert.NoError(t, err)
+
+		/// Reset nKode
+		attrsPerKey = 6
+		keypadSize = entities.KeypadDimension{AttrsPerKey: attrsPerKey, NumbOfKeys: numbOfKeys}
+		resetResponse, err := nkodeApi.GenerateSignupResetInterface(userEmail, *customerId, keypadSize, true)
+		assert.NoError(t, err)
+		setInterface = resetResponse.UserIdxInterface
+		sessionIdStr = resetResponse.SessionId
+		sessionId, err = entities.SessionIdFromString(sessionIdStr)
+		assert.NoError(t, err)
+		keypadSize = entities.KeypadDimension{AttrsPerKey: numbOfKeys, NumbOfKeys: numbOfKeys}
+		userPasscode = setInterface[:passcodeLen]
+		setKeySelect, err = entities.SelectKeyByAttrIdx(setInterface, userPasscode, keypadSize)
+		assert.NoError(t, err)
+		confirmInterface, err = nkodeApi.SetNKode(*customerId, sessionId, setKeySelect)
+		assert.NoError(t, err)
+		confirmKeySelect, err = entities.SelectKeyByAttrIdx(confirmInterface, userPasscode, keypadSize)
+		err = nkodeApi.ConfirmNKode(*customerId, sessionId, confirmKeySelect)
+		assert.NoError(t, err)
+
+		keypadSize = entities.KeypadDimension{AttrsPerKey: attrsPerKey, NumbOfKeys: numbOfKeys}
+		loginInterface2, err := nkodeApi.GetLoginInterface(userEmail, *customerId)
+		assert.NoError(t, err)
+		loginKeySelection, err = entities.SelectKeyByAttrIdx(loginInterface2.UserIdxInterface, userPasscode, keypadSize)
+		assert.NoError(t, err)
+		_, err = nkodeApi.Login(*customerId, userEmail, loginKeySelection)
+		assert.NoError(t, err)
+		signupResponse, err = nkodeApi.GenerateSignupResetInterface(userEmail, *customerId, keypadSize, false)
+		assert.Error(t, err)
+	}
+}