package service import ( "context" "fmt" "time" "github.com/mitchell/selfpass/credentials/types" ) func NewCredentials(repo types.CredentialRepo) Credentials { return Credentials{ repo: repo, } } type Credentials struct { repo types.CredentialRepo } func (svc Credentials) GetAllMetadata(ctx context.Context, sourceHost string) (output <-chan types.Metadata, errch chan error) { errch = make(chan error, 1) output = svc.repo.GetAllMetadata(ctx, sourceHost, errch) return output, errch } func (svc Credentials) Get(ctx context.Context, id string) (output types.Credential, err error) { if id == "" { return output, fmt.Errorf("%s must specify an id", types.InvalidArgument) } return svc.repo.Get(ctx, id) } func (svc Credentials) Create(ctx context.Context, ci types.CredentialInput) (output types.Credential, err error) { if err = validateCredentialInput(ci); err != nil { return output, err } var c types.Credential c.ID = generateID(ci) c.CreatedAt = time.Now() c.UpdatedAt = time.Now() c.Primary = ci.Primary c.LoginURL = ci.LoginURL c.SourceHost = ci.SourceHost c.Username = ci.Username c.Email = ci.Email c.Password = ci.Password c.OTPSecret = ci.OTPSecret c.Tag = ci.Tag err = svc.repo.Put(ctx, c) return c, err } func validateCredentialInput(c types.CredentialInput) (err error) { switch { case c.SourceHost == "": return fmt.Errorf("%s must specify source host", types.InvalidArgument) case c.Primary == "": return fmt.Errorf("%s must specify primary user key", types.InvalidArgument) case c.Password == "": return fmt.Errorf("%s must specify password", types.InvalidArgument) } return err } func generateID(ci types.CredentialInput) string { idFormat := types.TypePrefixCred + "-%s-%s" if ci.Tag != "" { idFormat += "-%s" return fmt.Sprintf(idFormat, ci.SourceHost, ci.Primary, ci.Tag) } return fmt.Sprintf(idFormat, ci.SourceHost, ci.Primary) } func (svc Credentials) Update(ctx context.Context, id string, ci types.CredentialInput) (output types.Credential, err error) { if err = validateCredentialInput(ci); err != nil { return output, err } if id == "" { return output, fmt.Errorf("%s must specify an id", types.InvalidArgument) } c, err := svc.repo.Get(ctx, id) if err != nil { return output, err } c.ID = generateID(ci) c.UpdatedAt = time.Now() c.Primary = ci.Primary c.LoginURL = ci.LoginURL c.SourceHost = ci.SourceHost c.Password = ci.Password c.OTPSecret = ci.OTPSecret c.Email = ci.Email c.Username = ci.Username c.Tag = ci.Tag return c, svc.repo.Put(ctx, c) } func (svc Credentials) Delete(ctx context.Context, id string) (err error) { if id == "" { return fmt.Errorf("%s must specify an id", types.InvalidArgument) } return svc.repo.Delete(ctx, id) } func (svc Credentials) DumpDB(ctx context.Context) (bs []byte, err error) { return svc.repo.DumpDB(ctx) }