package postgres import ( "context" "database/sql" "errors" "fmt" "git.aiterp.net/rpdata/api/database/postgres/psqlcore" "git.aiterp.net/rpdata/api/models" "strconv" "strings" ) var ErrNickConflict = errors.New("nick already in use by another character") type characterRepository struct { insertWithIDs bool db *sql.DB } func (r *characterRepository) character(row psqlcore.DataCharacter) *models.Character { return &models.Character{ ID: strings.Trim(row.ID, " "), Nicks: row.Nicks, Name: row.Name, ShortName: row.ShortName, Author: row.Author, Description: row.Description, } } func (r *characterRepository) characters(rows []psqlcore.DataCharacter) []*models.Character { results := make([]*models.Character, 0, len(rows)) for _, row := range rows { results = append(results, r.character(row)) } return results } func (r *characterRepository) Find(ctx context.Context, id string) (*models.Character, error) { row, err := psqlcore.New(r.db).SelectCharacterByID(ctx, id) if err != nil { return nil, err } return r.character(row), nil } func (r *characterRepository) FindNick(ctx context.Context, nick string) (*models.Character, error) { row, err := psqlcore.New(r.db).SelectCharacterByNick(ctx, nick) if err != nil { return nil, err } return r.character(row), nil } func (r *characterRepository) FindName(ctx context.Context, name string) (*models.Character, error) { row, err := psqlcore.New(r.db).SelectCharacterByName(ctx, name) if err != nil { return nil, err } return r.character(row), nil } func (r *characterRepository) List(ctx context.Context, filter models.CharacterFilter) ([]*models.Character, error) { params := psqlcore.SelectCharactersParams{ LimitSize: 10000, } if filter.IDs != nil { params.FilterID = true params.Ids = filter.IDs } if filter.Nicks != nil { params.FilterNick = true params.Nicks = filter.Nicks } if filter.Names != nil { params.FilterName = true params.Names = filter.Names } if filter.Author != nil { params.FilterAuthor = true params.Author = *filter.Author } if filter.Search != nil { params.FilterSearch = true params.Search = *filter.Search } if filter.Limit > 0 { params.LimitSize = int32(filter.Limit) } rows, err := psqlcore.New(r.db).SelectCharacters(ctx, params) if err != nil { return nil, err } return r.characters(rows), nil } func (r *characterRepository) Insert(ctx context.Context, character models.Character) (*models.Character, error) { tx, err := r.db.BeginTx(ctx, nil) if err != nil { return nil, err } defer func() { _ = tx.Rollback() }() q := psqlcore.New(tx) if !r.insertWithIDs || character.ID == "" { next, err := q.IncrementCounter(ctx, "data_character_id") if err != nil { return nil, err } character.ID = fmt.Sprintf("C%d", next) } else { n, err := strconv.Atoi(character.ID[1:]) if err != nil { return nil, err } err = q.BumpCounter(ctx, psqlcore.BumpCounterParams{ID: "data_character_id", Value: int32(n)}) if err != nil { return nil, err } } rows, err := q.SelectCharacters(ctx, psqlcore.SelectCharactersParams{ FilterNick: true, Nicks: character.Nicks, LimitSize: 1, }) if err != nil { return nil, fmt.Errorf("failed to select: %s", err) } if len(rows) != 0 { return nil, ErrNickConflict } err = q.InsertCharacter(ctx, psqlcore.InsertCharacterParams{ ID: character.ID, Nicks: character.Nicks, Name: character.Name, ShortName: character.ShortName, Author: character.Author, Description: character.Description, }) if err != nil { return nil, err } return &character, tx.Commit() } func (r *characterRepository) Update(ctx context.Context, character models.Character, update models.CharacterUpdate) (*models.Character, error) { character.ApplyUpdate(update) err := psqlcore.New(r.db).UpdateCharacter(ctx, psqlcore.UpdateCharacterParams{ ID: character.ID, Name: character.Name, ShortName: character.ShortName, Description: character.Description, }) if err != nil { return nil, err } return &character, nil } func (r *characterRepository) AddNick(ctx context.Context, character models.Character, nick string) (*models.Character, error) { tx, err := r.db.BeginTx(ctx, nil) if err != nil { return nil, err } defer func() { _ = tx.Rollback() }() q := psqlcore.New(tx) rows, err := q.SelectCharacters(ctx, psqlcore.SelectCharactersParams{ FilterNick: true, Nicks: []string{nick}, LimitSize: 1, }) if err != nil { return nil, err } if len(rows) != 0 { return nil, ErrNickConflict } err = q.AddCharacterNick(ctx, psqlcore.AddCharacterNickParams{ID: character.ID, Nick: nick}) if err != nil { return nil, err } character.Nicks = append(character.Nicks, nick) return &character, nil } func (r *characterRepository) RemoveNick(ctx context.Context, character models.Character, nick string) (*models.Character, error) { err := psqlcore.New(r.db).AddCharacterNick(ctx, psqlcore.AddCharacterNickParams{ID: character.ID, Nick: nick}) if err != nil { return nil, err } for i, nick2 := range character.Nicks { if nick2 == nick { character.Nicks = append(character.Nicks[:i], character.Nicks[i+1:]...) break } } return &character, nil } func (r *characterRepository) Delete(ctx context.Context, character models.Character) error { return psqlcore.New(r.db).DeleteCharacter(ctx, character.ID) }