75 lines
1.5 KiB
Go
75 lines
1.5 KiB
Go
package models
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/google/uuid"
|
|
|
|
"github.com/jtom38/newsbot/collector/database"
|
|
)
|
|
|
|
type SourceDto struct {
|
|
ID uuid.UUID `json:"id"`
|
|
Site string `json:"site"`
|
|
Name string `json:"name"`
|
|
Source string `json:"source"`
|
|
Type string `json:"type"`
|
|
Value string `json:"value"`
|
|
Enabled bool `json:"enabled"`
|
|
Url string `json:"url"`
|
|
Tags []string `json:"tags"`
|
|
Deleted bool `json:"deleted"`
|
|
}
|
|
|
|
func ConvertToSourceDto(i database.Source) SourceDto {
|
|
var deleted bool
|
|
if !i.Deleted.Valid {
|
|
deleted = true
|
|
}
|
|
|
|
return SourceDto{
|
|
ID: i.ID,
|
|
Site: i.Site,
|
|
Name: i.Name,
|
|
Source: i.Source,
|
|
Type: i.Type,
|
|
Value: i.Value.String,
|
|
Enabled: i.Enabled,
|
|
Url: i.Url,
|
|
Tags: splitTags(i.Tags),
|
|
Deleted: deleted,
|
|
}
|
|
}
|
|
|
|
type DiscordQueueDto struct {
|
|
ID uuid.UUID `json:"id"`
|
|
Articleid uuid.UUID `json:"articleId"`
|
|
}
|
|
|
|
func ConvertToDiscordQueueDto(i database.Discordqueue) DiscordQueueDto {
|
|
return DiscordQueueDto{
|
|
ID: i.ID,
|
|
Articleid: i.Articleid,
|
|
}
|
|
}
|
|
|
|
type SubscriptionDto struct {
|
|
ID uuid.UUID `json:"id"`
|
|
DiscordWebhookId uuid.UUID `json:"discordwebhookid"`
|
|
SourceId uuid.UUID `json:"sourceid"`
|
|
}
|
|
|
|
func ConvertToSubscriptionDto(i database.Subscription) SubscriptionDto {
|
|
c := SubscriptionDto{
|
|
ID: i.ID,
|
|
DiscordWebhookId: i.Discordwebhookid,
|
|
SourceId: i.Sourceid,
|
|
}
|
|
return c
|
|
}
|
|
|
|
func splitTags(t string) []string {
|
|
items := strings.Split(t, ", ")
|
|
return items
|
|
}
|