You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
120 lines
2.9 KiB
120 lines
2.9 KiB
package model
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"time"
|
|
|
|
"git.aiterp.net/AiteRP/aitestory/server"
|
|
)
|
|
|
|
// Header contains a subset of the page database
|
|
// table needed for the front page. It's a read-only
|
|
// model.
|
|
type Header struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Author string `json:"author"`
|
|
Category string `json:"category"`
|
|
FictionalDate time.Time `json:"fictionalDate"`
|
|
PublishDate time.Time `json:"publishDate"`
|
|
EditDate time.Time `json:"editDate"`
|
|
Dated bool `json:"dated"`
|
|
PrimaryTag *Tag `json:"primaryTag"`
|
|
}
|
|
|
|
// ListHeaders grabs all the general pages from
|
|
// the database to list them
|
|
func ListHeaders() ([]Header, error) {
|
|
const query = `
|
|
SELECT page.id,page.name,author,category,fictional_date,publish_date,edit_date,dated,tag.id,tag.type,tag.name
|
|
FROM page
|
|
LEFT JOIN page_tag ON (page.id = page_tag.page_id AND page_tag.primary = true)
|
|
LEFT JOIN tag ON (tag.id = page_tag.tag_id)
|
|
WHERE page.specific=false AND page.published=true AND page.unlisted=false;
|
|
`
|
|
|
|
db := server.Main.DB
|
|
|
|
rows, err := db.Query(query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
results := make([]Header, 0, 64)
|
|
header := Header{}
|
|
for rows.Next() {
|
|
err := parseHeader(&header, rows)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
results = append(results, header)
|
|
}
|
|
|
|
return results, nil
|
|
}
|
|
|
|
// ListHeadersByTag lists all headers that has the tag
|
|
func ListHeadersByTag(tag *Tag) ([]Header, error) {
|
|
const query = `
|
|
SELECT page.id,page.name,page.author,page.category,page.fictional_date,page.publish_date,page.edit_date,page.dated,tag.id,tag.type,tag.name
|
|
FROM page_tag
|
|
RIGHT JOIN page ON page.id = page_tag.page_id
|
|
LEFT JOIN (page_tag AS pt2) ON (page.id = pt2.page_id AND pt2.primary = true)
|
|
LEFT JOIN (tag AS tag) ON (tag.id = pt2.tag_id)
|
|
WHERE page_tag.tag_id=?
|
|
`
|
|
|
|
if tag == nil {
|
|
return nil, errors.New("no tag")
|
|
}
|
|
|
|
db := server.Main.DB
|
|
|
|
rows, err := db.Query(query, tag.ID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
results := make([]Header, 0, 64)
|
|
header := Header{}
|
|
for rows.Next() {
|
|
err := parseHeader(&header, rows)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
results = append(results, header)
|
|
}
|
|
|
|
return results, nil
|
|
}
|
|
|
|
func parseHeader(header *Header, rows *sql.Rows) error {
|
|
var tagID, tagName, tagType string
|
|
var fictionalDate, publishDate, editDate string
|
|
var err error
|
|
|
|
rows.Scan(&header.ID, &header.Name, &header.Author, &header.Category, &fictionalDate, &publishDate, &editDate, &header.Dated, &tagID, &tagType, &tagName)
|
|
if tagID != "" {
|
|
header.PrimaryTag = &Tag{tagID, tagName, tagType}
|
|
}
|
|
|
|
header.FictionalDate, err = time.Parse("2006-01-02 15:04:05", fictionalDate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
header.PublishDate, err = time.Parse("2006-01-02 15:04:05", publishDate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
header.EditDate, err = time.Parse("2006-01-02 15:04:05", editDate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|