LogalDeveloper/owncast
0
Fork 0
Code Activity

chore(go): migrate more models to codegen versions. For #3778

Browse Source
This commit is contained in:
Gabe Kangas
2025-02-12 21:18:13 -08:00
parent 8bdb52fd37
commit 4b627f0693
6 changed files with 17 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
persistence/authrepository/authrepository.go
View File

@@ -3,13 +3,13 @@ package authrepository
import (
"database/sql"
"github.com/owncast/owncast/models"
"github.com/owncast/owncast/webserver/handlers/generated"
)
type AuthRepository interface {
CreateBanIPTable(db *sql.DB)
BanIPAddress(address, note string) error
IsIPAddressBanned(address string) (bool, error)
GetIPAddressBans() ([]models.IPAddress, error)
GetIPAddressBans() ([]generated.IPAddress, error)
RemoveIPAddressBan(address string) error
}

14
persistence/authrepository/bans.go
View File

@@ -6,7 +6,7 @@ import (
"log"
"github.com/owncast/owncast/db"
"github.com/owncast/owncast/models"
"github.com/owncast/owncast/webserver/handlers/generated"
)
// CreateBanIPTable will create the IP ban table if needed.
@@ -42,18 +42,18 @@ func (r *SqlAuthRepository) IsIPAddressBanned(address string) (bool, error) {
}
// GetIPAddressBans will return all the banned IP addresses.
func (r *SqlAuthRepository) GetIPAddressBans() ([]models.IPAddress, error) {
func (r *SqlAuthRepository) GetIPAddressBans() ([]generated.IPAddress, error) {
result, err := r.datastore.GetQueries().GetIPAddressBans(context.Background())
if err != nil {
return nil, err
}
response := []models.IPAddress{}
response := []generated.IPAddress{}
for _, ip := range result {
response = append(response, models.IPAddress{
IPAddress: ip.IpAddress,
Notes: ip.Notes.String,
CreatedAt: ip.CreatedAt.Time,
response = append(response, generated.IPAddress{
IpAddress: &ip.IpAddress,
Notes: &ip.Notes.String,
CreatedAt: &ip.CreatedAt.Time,
})
}
return response, err