Initial commit

2026-08-05 19:31:35 +03:00
commit a33fa24e99
272 changed files with 40566 additions and 0 deletions
@@ -0,0 +1,124 @@
+package database
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"github.com/TelegramExchange/pkg/transaction"
+	"github.com/TelegramExchange/tgex-backend/tg_parser/internal/domain"
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+func (d *Database) UpdateChannelIfNotAccessible(ctx context.Context, channel domain.Channel) error {
+	query := `UPDATE channel
+SET telegram_id = $2,
+    username = $3,
+    title = $4,
+    access_hash = $5,
+    pts = 0,
+    is_accessible = TRUE,
+    invite_link = $6,
+    updated_at = CURRENT_TIMESTAMP
+WHERE telegram_id = $1
+  AND is_accessible = FALSE;`
+
+	txOrPool := transaction.TryExtractTX(ctx)
+
+	username := strings.TrimSpace(channel.Username)
+	var usernameDTO pgtype.Text
+	if username != "" {
+		usernameDTO = pgtype.Text{String: username, Valid: true}
+	} else {
+		usernameDTO = pgtype.Text{Valid: false}
+	}
+
+	inviteLink := strings.TrimSpace(channel.InviteLink)
+	var inviteLinkDTO pgtype.Text
+	if inviteLink != "" {
+		inviteLinkDTO = pgtype.Text{String: inviteLink, Valid: true}
+	} else {
+		inviteLinkDTO = pgtype.Text{Valid: false}
+	}
+
+	result, err := txOrPool.Exec(ctx, query,
+		channel.TelegramID,
+		pgtype.Int8{Int64: channel.TelegramID, Valid: true},
+		usernameDTO,
+		pgtype.Text{String: channel.Title, Valid: true},
+		pgtype.Int8{Int64: channel.AccessHash, Valid: true},
+		inviteLinkDTO,
+	)
+	if err != nil {
+		return fmt.Errorf("txOrPool.Exec: %w", err)
+	}
+
+	// Log if no rows were updated (channel either doesn't exist or is already accessible)
+	rowsAffected := result.RowsAffected()
+	if rowsAffected == 0 {
+		// Channel doesn't exist or is already accessible - this is fine
+		return nil
+	}
+
+	return nil
+}
+
+func (d *Database) UpdateChannel(ctx context.Context, channel domain.Channel) error {
+	query := `UPDATE channel
+SET telegram_id = $2,
+    username = $3,
+    title = $4,
+    access_hash = $5,
+    pts = $6,
+    is_accessible = $7,
+    invite_link = $8,
+    updated_at = CURRENT_TIMESTAMP
+WHERE id = $1;`
+
+	txOrPool := transaction.TryExtractTX(ctx)
+
+	username := strings.TrimSpace(channel.Username)
+	var usernameDTO pgtype.Text
+	if username != "" {
+		usernameDTO = pgtype.Text{String: username, Valid: true}
+	} else {
+		usernameDTO = pgtype.Text{Valid: false}
+	}
+
+	inviteLink := strings.TrimSpace(channel.InviteLink)
+	var inviteLinkDTO pgtype.Text
+	if inviteLink != "" {
+		inviteLinkDTO = pgtype.Text{String: inviteLink, Valid: true}
+	} else {
+		inviteLinkDTO = pgtype.Text{Valid: false}
+	}
+
+	dto := updateChannelDTO{
+		ID:           pgtype.UUID{Bytes: channel.ID, Valid: true},
+		TelegramID:   pgtype.Int8{Int64: channel.TelegramID, Valid: true},
+		Username:     usernameDTO,
+		Title:        pgtype.Text{String: channel.Title, Valid: true},
+		AccessHash:   pgtype.Int8{Int64: channel.AccessHash, Valid: true},
+		Pts:          pgtype.Int4{Int32: int32(channel.Pts), Valid: true},
+		IsAccessible: pgtype.Bool{Bool: channel.IsAccessible, Valid: true},
+		InviteLink:   inviteLinkDTO,
+	}
+
+	_, err := txOrPool.Exec(ctx, query, dto.ID, dto.TelegramID, dto.Username, dto.Title, dto.AccessHash, dto.Pts, dto.IsAccessible, dto.InviteLink)
+	if err != nil {
+		return fmt.Errorf("txOrPool.Exec: %w", err)
+	}
+
+	return nil
+}
+
+type updateChannelDTO struct {
+	ID           pgtype.UUID
+	TelegramID   pgtype.Int8
+	Username     pgtype.Text
+	Title        pgtype.Text
+	AccessHash   pgtype.Int8
+	Pts          pgtype.Int4
+	IsAccessible pgtype.Bool
+	InviteLink   pgtype.Text
+}