commit 444b92046f596b38775b125f5d523dbda04e9f29 · kot.pink/core

+109
appview/db/profile.go
···

       348
       348
        
       	return tx.Commit()

     

       349
       349
        
       }

     

       350
       350
        
       

     

       351
       351
       +
       func GetProfiles(e Execer, filters ...filter) ([]Profile, error) {

     

       352
       352
       +
       	var conditions []string

     

       353
       353
       +
       	var args []any

     

       354
       354
       +
       	for _, filter := range filters {

     

       355
       355
       +
       		conditions = append(conditions, filter.Condition())

     

       356
       356
       +
       		args = append(args, filter.Arg()...)

     

       357
       357
       +
       	}

     

       358
       358
       +
       

     

       359
       359
       +
       	whereClause := ""

     

       360
       360
       +
       	if conditions != nil {

     

       361
       361
       +
       		whereClause = " where " + strings.Join(conditions, " and ")

     

       362
       362
       +
       	}

     

       363
       363
       +
       

     

       364
       364
       +
       	profilesQuery := fmt.Sprintf(

     

       365
       365
       +
       		`select

     

       366
       366
       +
       			id,

     

       367
       367
       +
       			did,

     

       368
       368
       +
       			description,

     

       369
       369
       +
       			include_bluesky,

     

       370
       370
       +
       			location

     

       371
       371
       +
       		from

     

       372
       372
       +
       			profile

     

       373
       373
       +
       		%s`,

     

       374
       374
       +
       		whereClause,

     

       375
       375
       +
       	)

     

       376
       376
       +
       	rows, err := e.Query(profilesQuery, args...)

     

       377
       377
       +
       	if err != nil {

     

       378
       378
       +
       		return nil, err

     

       379
       379
       +
       	}

     

       380
       380
       +
       

     

       381
       381
       +
       	profileMap := make(map[string]*Profile)

     

       382
       382
       +
       	for rows.Next() {

     

       383
       383
       +
       		var profile Profile

     

       384
       384
       +
       		var includeBluesky int

     

       385
       385
       +
       

     

       386
       386
       +
       		err = rows.Scan(&profile.ID, &profile.Did, &profile.Description, &includeBluesky, &profile.Location)

     

       387
       387
       +
       		if err != nil {

     

       388
       388
       +
       			return nil, err

     

       389
       389
       +
       		}

     

       390
       390
       +
       

     

       391
       391
       +
       		if includeBluesky != 0 {

     

       392
       392
       +
       			profile.IncludeBluesky = true

     

       393
       393
       +
       		}

     

       394
       394
       +
       

     

       395
       395
       +
       		profileMap[profile.Did] = &profile

     

       396
       396
       +
       	}

     

       397
       397
       +
       	if err = rows.Err(); err != nil {

     

       398
       398
       +
       		return nil, err

     

       399
       399
       +
       	}

     

       400
       400
       +
       

     

       401
       401
       +
       	// populate profile links

     

       402
       402
       +
       	inClause := strings.TrimSuffix(strings.Repeat("?, ", len(profileMap)), ", ")

     

       403
       403
       +
       	args = make([]any, len(profileMap))

     

       404
       404
       +
       	i := 0

     

       405
       405
       +
       	for did := range profileMap {

     

       406
       406
       +
       		args[i] = did

     

       407
       407
       +
       		i++

     

       408
       408
       +
       	}

     

       409
       409
       +
       

     

       410
       410
       +
       	linksQuery := fmt.Sprintf("select link, did from profile_links where did in (%s)", inClause)

     

       411
       411
       +
       	rows, err = e.Query(linksQuery, args...)

     

       412
       412
       +
       	if err != nil {

     

       413
       413
       +
       		return nil, err

     

       414
       414
       +
       	}

     

       415
       415
       +
       	idxs := make(map[string]int)

     

       416
       416
       +
       	for did := range profileMap {

     

       417
       417
       +
       		idxs[did] = 0

     

       418
       418
       +
       	}

     

       419
       419
       +
       	for rows.Next() {

     

       420
       420
       +
       		var link, did string

     

       421
       421
       +
       		if err = rows.Scan(&link, &did); err != nil {

     

       422
       422
       +
       			return nil, err

     

       423
       423
       +
       		}

     

       424
       424
       +
       

     

       425
       425
       +
       		idx := idxs[did]

     

       426
       426
       +
       		log.Println("idx", "idx", idx, "link", link)

     

       427
       427
       +
       		profileMap[did].Links[idx] = link

     

       428
       428
       +
       		idxs[did] = idx + 1

     

       429
       429
       +
       	}

     

       430
       430
       +
       

     

       431
       431
       +
       	pinsQuery := fmt.Sprintf("select at_uri, did from profile_pinned_repositories where did in (%s)", inClause)

     

       432
       432
       +
       	rows, err = e.Query(pinsQuery, args...)

     

       433
       433
       +
       	if err != nil {

     

       434
       434
       +
       		return nil, err

     

       435
       435
       +
       	}

     

       436
       436
       +
       	idxs = make(map[string]int)

     

       437
       437
       +
       	for did := range profileMap {

     

       438
       438
       +
       		idxs[did] = 0

     

       439
       439
       +
       	}

     

       440
       440
       +
       	for rows.Next() {

     

       441
       441
       +
       		var link syntax.ATURI

     

       442
       442
       +
       		var did string

     

       443
       443
       +
       		if err = rows.Scan(&link, &did); err != nil {

     

       444
       444
       +
       			return nil, err

     

       445
       445
       +
       		}

     

       446
       446
       +
       

     

       447
       447
       +
       		idx := idxs[did]

     

       448
       448
       +
       		profileMap[did].PinnedRepos[idx] = link

     

       449
       449
       +
       		idxs[did] = idx + 1

     

       450
       450
       +
       	}

     

       451
       451
       +
       

     

       452
       452
       +
       	var profiles []Profile

     

       453
       453
       +
       	for _, p := range profileMap {

     

       454
       454
       +
       		profiles = append(profiles, *p)

     

       455
       455
       +
       	}

     

       456
       456
       +
       

     

       457
       457
       +
       	return profiles, nil

     

       458
       458
       +
       }

     

       459
       459
       +
       

     

       351
       460
        
       func GetProfile(e Execer, did string) (*Profile, error) {

     

       352
       461
        
       	var profile Profile

     

       353
       462
        
       	profile.Did = did