56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"sproutgate-backend/internal/models"
|
|
)
|
|
|
|
func (s *Store) SaveReset(record models.ResetPassword) error {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
path := s.resetFilePath(record.Account)
|
|
return writeJSONFile(path, record)
|
|
}
|
|
|
|
func (s *Store) GetReset(account string) (models.ResetPassword, bool, error) {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
path := s.resetFilePath(account)
|
|
if _, err := os.Stat(path); errors.Is(err, os.ErrNotExist) {
|
|
return models.ResetPassword{}, false, nil
|
|
}
|
|
var record models.ResetPassword
|
|
if err := readJSONFile(path, &record); err != nil {
|
|
return models.ResetPassword{}, false, err
|
|
}
|
|
return record, true, nil
|
|
}
|
|
|
|
func (s *Store) DeleteReset(account string) error {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
path := s.resetFilePath(account)
|
|
if _, err := os.Stat(path); errors.Is(err, os.ErrNotExist) {
|
|
return nil
|
|
}
|
|
return os.Remove(path)
|
|
}
|
|
|
|
func (s *Store) resetFilePath(account string) string {
|
|
return filepath.Join(s.resetDir, resetFileName(account))
|
|
}
|
|
|
|
func resetFileName(account string) string {
|
|
safe := strings.TrimSpace(account)
|
|
if safe == "" {
|
|
safe = "unknown"
|
|
}
|
|
encoded := base64.RawURLEncoding.EncodeToString([]byte(safe))
|
|
return encoded + ".json"
|
|
}
|