commit 9f6c088908763d7218eb45431e3bd98736807018 · bretton.dev/coves

+15 -5

internal/core/comments/comment_service.go

···

       6
        
       	"errors"

     

       7
        
       	"fmt"

     

       8
        
       	"log"

     

       0
        
       
     

       9
        
       	"strings"

     

       10
        
       	"time"

     

       11
        
       

     
···

       475
        
       		// Avatar is stored as blob in community's repository

     

       476
        
       		// Format: https://{pds}/xrpc/com.atproto.sync.getBlob?did={community_did}&cid={avatar_cid}

     

       477
        
       		if community.AvatarCID != "" && community.PDSURL != "" {

     

       478
       -
       			avatarURLString := fmt.Sprintf("%s/xrpc/com.atproto.sync.getBlob?did=%s&cid=%s",

     

       479
       -
       				strings.TrimSuffix(community.PDSURL, "/"),

     

       480
       -
       				community.DID,

     

       481
       -
       				community.AvatarCID)

     

       482
       -
       			avatarURL = &avatarURLString

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       483
        
       		}

     

       484
        
       	} else {

     

       485
        
       		// Log warning but don't fail the entire request

···

       6
        
       	"errors"

     

       7
        
       	"fmt"

     

       8
        
       	"log"

     

       9
       +
       	"net/url"

     

       10
        
       	"strings"

     

       11
        
       	"time"

     

       12
        
       

     
···

       476
        
       		// Avatar is stored as blob in community's repository

     

       477
        
       		// Format: https://{pds}/xrpc/com.atproto.sync.getBlob?did={community_did}&cid={avatar_cid}

     

       478
        
       		if community.AvatarCID != "" && community.PDSURL != "" {

     

       479
       +
       			// Validate HTTPS for security (prevent mixed content warnings, MitM attacks)

     

       480
       +
       			if !strings.HasPrefix(community.PDSURL, "https://") {

     

       481
       +
       				log.Printf("Warning: Skipping non-HTTPS PDS URL for community %s", community.DID)

     

       482
       +
       			} else if !strings.HasPrefix(community.AvatarCID, "baf") {

     

       483
       +
       				// Validate CID format (IPFS CIDs start with "baf" for CIDv1 base32)

     

       484
       +
       				log.Printf("Warning: Invalid CID format for community %s", community.DID)

     

       485
       +
       			} else {

     

       486
       +
       				// Use proper URL escaping to prevent injection attacks

     

       487
       +
       				avatarURLString := fmt.Sprintf("%s/xrpc/com.atproto.sync.getBlob?did=%s&cid=%s",

     

       488
       +
       					strings.TrimSuffix(community.PDSURL, "/"),

     

       489
       +
       					url.QueryEscape(community.DID),

     

       490
       +
       					url.QueryEscape(community.AvatarCID))

     

       491
       +
       				avatarURL = &avatarURLString

     

       492
       +
       			}

     

       493
        
       		}

     

       494
        
       	} else {

     

       495
        
       		// Log warning but don't fail the entire request

+20

internal/core/users/interfaces.go

···

       8
        
       	GetByDID(ctx context.Context, did string) (*User, error)

     

       9
        
       	GetByHandle(ctx context.Context, handle string) (*User, error)

     

       10
        
       	UpdateHandle(ctx context.Context, did, newHandle string) (*User, error)

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       11
        
       	GetByDIDs(ctx context.Context, dids []string) (map[string]*User, error)

     

       12
        
       }

     

       13

···

       8
        
       	GetByDID(ctx context.Context, did string) (*User, error)

     

       9
        
       	GetByHandle(ctx context.Context, handle string) (*User, error)

     

       10
        
       	UpdateHandle(ctx context.Context, did, newHandle string) (*User, error)

     

       11
       +
       

     

       12
       +
       	// GetByDIDs retrieves multiple users by their DIDs in a single batch query.

     

       13
       +
       	// Returns a map of DID → User for efficient lookups.

     

       14
       +
       	// Missing users are not included in the result map (no error for missing users).

     

       15
       +
       	// Returns error only on database failures or validation errors (invalid DIDs, batch too large).

     

       16
       +
       	//

     

       17
       +
       	// Parameters:

     

       18
       +
       	//   - ctx: Context for cancellation and timeout

     

       19
       +
       	//   - dids: Array of DIDs to retrieve (must start with "did:", max 1000 items)

     

       20
       +
       	//

     

       21
       +
       	// Returns:

     

       22
       +
       	//   - map[string]*User: Map of DID → User for found users

     

       23
       +
       	//   - error: Validation or database errors (not errors for missing users)

     

       24
       +
       	//

     

       25
       +
       	// Example:

     

       26
       +
       	//   userMap, err := repo.GetByDIDs(ctx, []string{"did:plc:abc", "did:plc:xyz"})

     

       27
       +
       	//   if err != nil { return err }

     

       28
       +
       	//   if user, found := userMap["did:plc:abc"]; found {

     

       29
       +
       	//       // Use user

     

       30
       +
       	//   }

     

       31
        
       	GetByDIDs(ctx context.Context, dids []string) (map[string]*User, error)

     

       32
        
       }

     

       33

+15

internal/db/postgres/user_repo.go

···

       106
        
       	return user, nil

     

       107
        
       }

     

       108
        
       

     

       0
        
       
     

       0
        
       
     

       109
        
       // GetByDIDs retrieves multiple users by their DIDs in a single query

     

       110
        
       // Returns a map of DID -> User for efficient lookups

     

       111
        
       // Missing users are not included in the result map (no error for missing users)

     

       112
        
       func (r *postgresUserRepo) GetByDIDs(ctx context.Context, dids []string) (map[string]*users.User, error) {

     

       113
        
       	if len(dids) == 0 {

     

       114
        
       		return make(map[string]*users.User), nil

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       115
        
       	}

     

       116
        
       

     

       117
        
       	// Build parameterized query with IN clause

···

       106
        
       	return user, nil

     

       107
        
       }

     

       108
        
       

     

       109
       +
       const MaxBatchSize = 1000

     

       110
       +
       

     

       111
        
       // GetByDIDs retrieves multiple users by their DIDs in a single query

     

       112
        
       // Returns a map of DID -> User for efficient lookups

     

       113
        
       // Missing users are not included in the result map (no error for missing users)

     

       114
        
       func (r *postgresUserRepo) GetByDIDs(ctx context.Context, dids []string) (map[string]*users.User, error) {

     

       115
        
       	if len(dids) == 0 {

     

       116
        
       		return make(map[string]*users.User), nil

     

       117
       +
       	}

     

       118
       +
       

     

       119
       +
       	// Validate batch size to prevent excessive memory usage and query timeouts

     

       120
       +
       	if len(dids) > MaxBatchSize {

     

       121
       +
       		return nil, fmt.Errorf("batch size %d exceeds maximum %d", len(dids), MaxBatchSize)

     

       122
       +
       	}

     

       123
       +
       

     

       124
       +
       	// Validate DID format to prevent SQL injection and malformed queries

     

       125
       +
       	// All atProto DIDs must start with "did:" prefix

     

       126
       +
       	for _, did := range dids {

     

       127
       +
       		if !strings.HasPrefix(did, "did:") {

     

       128
       +
       			return nil, fmt.Errorf("invalid DID format: %s", did)

     

       129
       +
       		}

     

       130
        
       	}

     

       131
        
       

     

       132
        
       	// Build parameterized query with IN clause