Changeset 0.21.4 (#90)

server/store/postgres/git_credentials.go

@@ -0,0 +1,76 @@
+package postgres
+
+import (
+	"context"
+	"database/sql"
+
+	"git.gobha.me/xcaliber/chat-switchboard/models"
+)
+
+// GitCredentialStore implements store.GitCredentialStore for PostgreSQL.
+type GitCredentialStore struct{}
+
+func (s *GitCredentialStore) Create(ctx context.Context, cred *models.GitCredential) error {
+	if cred.ID != "" {
+		return DB.QueryRowContext(ctx, `
+			INSERT INTO git_credentials (id, user_id, name, auth_type, encrypted_data, nonce)
+			VALUES ($1, $2, $3, $4, $5, $6)
+			RETURNING created_at`,
+			cred.ID, cred.UserID, cred.Name, cred.AuthType,
+			cred.EncryptedData, cred.Nonce,
+		).Scan(&cred.CreatedAt)
+	}
+	return DB.QueryRowContext(ctx, `
+		INSERT INTO git_credentials (user_id, name, auth_type, encrypted_data, nonce)
+		VALUES ($1, $2, $3, $4, $5)
+		RETURNING id, created_at`,
+		cred.UserID, cred.Name, cred.AuthType,
+		cred.EncryptedData, cred.Nonce,
+	).Scan(&cred.ID, &cred.CreatedAt)
+}
+
+func (s *GitCredentialStore) GetByID(ctx context.Context, id string) (*models.GitCredential, error) {
+	var c models.GitCredential
+	err := DB.QueryRowContext(ctx, `
+		SELECT id, user_id, name, auth_type, encrypted_data, nonce, created_at
+		FROM git_credentials WHERE id = $1`, id,
+	).Scan(&c.ID, &c.UserID, &c.Name, &c.AuthType,
+		&c.EncryptedData, &c.Nonce, &c.CreatedAt)
+	if err != nil {
+		return nil, err
+	}
+	return &c, nil
+}
+
+func (s *GitCredentialStore) ListByUser(ctx context.Context, userID string) ([]models.GitCredential, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT id, user_id, name, auth_type, encrypted_data, nonce, created_at
+		FROM git_credentials WHERE user_id = $1 ORDER BY created_at DESC`, userID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var creds []models.GitCredential
+	for rows.Next() {
+		var c models.GitCredential
+		if err := rows.Scan(&c.ID, &c.UserID, &c.Name, &c.AuthType,
+			&c.EncryptedData, &c.Nonce, &c.CreatedAt); err != nil {
+			return nil, err
+		}
+		creds = append(creds, c)
+	}
+	return creds, rows.Err()
+}
+
+func (s *GitCredentialStore) Delete(ctx context.Context, id, userID string) error {
+	res, err := DB.ExecContext(ctx, `DELETE FROM git_credentials WHERE id = $1 AND user_id = $2`, id, userID)
+	if err != nil {
+		return err
+	}
+	n, _ := res.RowsAffected()
+	if n == 0 {
+		return sql.ErrNoRows
+	}
+	return nil
+}