Something went wrong. Try again.
forked from tangled.org/core
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288package db
import ( "strings" "time")
type Email struct { ID int64 Did string Address string Verified bool Primary bool VerificationCode string LastSent *time.Time CreatedAt time.Time}
func GetPrimaryEmail(e Execer, did string) (Email, error) { query := ` select id, did, email, verified, is_primary, verification_code, last_sent, created from emails where did = ? and is_primary = true ` var email Email var createdStr string var lastSent string err := e.QueryRow(query, did).Scan(&email.ID, &email.Did, &email.Address, &email.Verified, &email.Primary, &email.VerificationCode, &lastSent, &createdStr) if err != nil { return Email{}, err } email.CreatedAt, err = time.Parse(time.RFC3339, createdStr) if err != nil { return Email{}, err } parsedTime, err := time.Parse(time.RFC3339, lastSent) if err != nil { return Email{}, err } email.LastSent = &parsedTime return email, nil}
func GetEmail(e Execer, did string, em string) (Email, error) { query := ` select id, did, email, verified, is_primary, verification_code, last_sent, created from emails where did = ? and email = ? ` var email Email var createdStr string var lastSent string err := e.QueryRow(query, did, em).Scan(&email.ID, &email.Did, &email.Address, &email.Verified, &email.Primary, &email.VerificationCode, &lastSent, &createdStr) if err != nil { return Email{}, err } email.CreatedAt, err = time.Parse(time.RFC3339, createdStr) if err != nil { return Email{}, err } parsedTime, err := time.Parse(time.RFC3339, lastSent) if err != nil { return Email{}, err } email.LastSent = &parsedTime return email, nil}
func GetDidForEmail(e Execer, em string) (string, error) { query := ` select did from emails where email = ? ` var did string err := e.QueryRow(query, em).Scan(&did) if err != nil { return "", err } return did, nil}
func GetEmailToDid(e Execer, ems []string, isVerifiedFilter bool) (map[string]string, error) { if len(ems) == 0 { return make(map[string]string), nil }
verifiedFilter := 0 if isVerifiedFilter { verifiedFilter = 1 }
// Create placeholders for the IN clause placeholders := make([]string, len(ems)) args := make([]any, len(ems)+1)
args[0] = verifiedFilter for i, em := range ems { placeholders[i] = "?" args[i+1] = em }
query := ` select email, did from emails where verified = ? and email in (` + strings.Join(placeholders, ",") + `) `
rows, err := e.Query(query, args...) if err != nil { return nil, err } defer rows.Close()
assoc := make(map[string]string)
for rows.Next() { var email, did string if err := rows.Scan(&email, &did); err != nil { return nil, err } assoc[email] = did }
if err := rows.Err(); err != nil { return nil, err }
return assoc, nil}
func GetVerificationCodeForEmail(e Execer, did string, email string) (string, error) { query := ` select verification_code from emails where did = ? and email = ? ` var code string err := e.QueryRow(query, did, email).Scan(&code) if err != nil { return "", err } return code, nil}
func CheckEmailExists(e Execer, did string, email string) (bool, error) { query := ` select count(*) from emails where did = ? and email = ? ` var count int err := e.QueryRow(query, did, email).Scan(&count) if err != nil { return false, err } return count > 0, nil}
func CheckValidVerificationCode(e Execer, did string, email string, code string) (bool, error) { query := ` select count(*) from emails where did = ? and email = ? and verification_code = ? ` var count int err := e.QueryRow(query, did, email, code).Scan(&count) if err != nil { return false, err } return count > 0, nil}
func AddEmail(e Execer, email Email) error { // Check if this is the first email for this DID countQuery := ` select count(*) from emails where did = ? ` var count int err := e.QueryRow(countQuery, email.Did).Scan(&count) if err != nil { return err }
// If this is the first email, mark it as primary if count == 0 { email.Primary = true }
query := ` insert into emails (did, email, verified, is_primary, verification_code) values (?, ?, ?, ?, ?) ` _, err = e.Exec(query, email.Did, email.Address, email.Verified, email.Primary, email.VerificationCode) return err}
func DeleteEmail(e Execer, did string, email string) error { query := ` delete from emails where did = ? and email = ? ` _, err := e.Exec(query, did, email) return err}
func MarkEmailVerified(e Execer, did string, email string) error { query := ` update emails set verified = true where did = ? and email = ? ` _, err := e.Exec(query, did, email) return err}
func MakeEmailPrimary(e Execer, did string, email string) error { // First, unset all primary emails for this DID query1 := ` update emails set is_primary = false where did = ? ` _, err := e.Exec(query1, did) if err != nil { return err }
// Then, set the specified email as primary query2 := ` update emails set is_primary = true where did = ? and email = ? ` _, err = e.Exec(query2, did, email) return err}
func GetAllEmails(e Execer, did string) ([]Email, error) { query := ` select did, email, verified, is_primary, verification_code, last_sent, created from emails where did = ? ` rows, err := e.Query(query, did) if err != nil { return nil, err } defer rows.Close()
var emails []Email for rows.Next() { var email Email var createdStr string var lastSent string err := rows.Scan(&email.Did, &email.Address, &email.Verified, &email.Primary, &email.VerificationCode, &lastSent, &createdStr) if err != nil { return nil, err } email.CreatedAt, err = time.Parse(time.RFC3339, createdStr) if err != nil { return nil, err } parsedTime, err := time.Parse(time.RFC3339, lastSent) if err != nil { return nil, err } email.LastSent = &parsedTime emails = append(emails, email) } return emails, nil}
func UpdateVerificationCode(e Execer, did string, email string, code string) error { query := ` update emails set verification_code = ?, last_sent = strftime('%Y-%m-%dT%H:%M:%SZ', 'now') where did = ? and email = ? ` _, err := e.Exec(query, code, did, email) return err}