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.
148 lines
4.0 KiB
148 lines
4.0 KiB
package storage
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
const logHeaderInsert = "INSERT INTO `logheader` (id,channel,time,status,status_time,title,tag,description) VALUES (?, ?, ?, ?, ?, ?, ?, ?);"
|
|
const logHeaderUpdate = "UPDATE `logheader` SET channel=?,time=?,status=?,status_time=?,title=?,tag=?,description=? WHERE id=?;"
|
|
const logNicksInsert = "INSERT INTO `lognicks` (log_id,nick) VALUES (?, ?);"
|
|
const logNicksDelete = "DELETE FROM `lognicks` WHERE log_id=? AND nick=?;"
|
|
|
|
// LogHeader is the struct
|
|
type LogHeader struct {
|
|
ID string `json:"id"`
|
|
Channel string `json:"channel"`
|
|
Time time.Time `json:"time"`
|
|
Status string `json:"status"`
|
|
StatusTime time.Time `json:"statusTime"`
|
|
Title string `json:"title"`
|
|
Tag string `json:"tag"`
|
|
Description string `json:"description"`
|
|
Nicks []string `json:"nicks"`
|
|
|
|
nicksAdded []string
|
|
nicksRemoved []string
|
|
}
|
|
|
|
// SetStatus changes the value of header.Status and updates
|
|
// header.StatusTime.
|
|
func (header *LogHeader) SetStatus(value string) {
|
|
header.Status = strings.ToLower(value)
|
|
header.StatusTime = time.Now()
|
|
}
|
|
|
|
// HasNick returns whether a nick exists
|
|
func (header *LogHeader) HasNick(nick string) bool {
|
|
for _, existingNick := range header.Nicks {
|
|
if existingNick == nick {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
// AddNick adds a nick to the Nicks array, recording the
|
|
// change. If the nick exists, it will not be added, though
|
|
// it won't throw an error.
|
|
func (header *LogHeader) AddNick(nick string) {
|
|
if !header.HasNick(nick) {
|
|
header.Nicks = append(header.Nicks, nick)
|
|
header.nicksAdded = append(header.nicksAdded, nick)
|
|
}
|
|
}
|
|
|
|
// RemoveNick removes a nick from the Nicks array, recording
|
|
// the change.
|
|
func (header *LogHeader) RemoveNick(nick string) {
|
|
for i, existingNick := range header.Nicks {
|
|
if existingNick == nick {
|
|
header.Nicks = append(header.Nicks[:i], header.Nicks[i+1:]...)
|
|
header.nicksRemoved = append(header.nicksRemoved, nick)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
// Insert inserts the LogHeader into the database, as well as all nicks
|
|
func (header *LogHeader) Insert(db *sql.DB) error {
|
|
// Insert the header
|
|
result, err := db.Exec(logHeaderInsert, header.ID, header.Channel, header.Time, header.Status, header.StatusTime, header.Title, header.Tag, header.Description)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Check if it was actually inserted
|
|
rows, err := result.RowsAffected()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if rows != 1 {
|
|
return errors.New("storage: No LogHeader inserted")
|
|
}
|
|
|
|
// Insert all lognicks
|
|
for _, nick := range header.Nicks {
|
|
_, err := db.Exec(logNicksInsert, header.ID, nick)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Clear the nicksAdded and nicksRemoved lists
|
|
header.nicksAdded = nil
|
|
header.nicksRemoved = nil
|
|
|
|
// Success
|
|
return nil
|
|
}
|
|
|
|
// Save updates the logheader
|
|
func (header *LogHeader) Save(db *sql.DB) error {
|
|
// Insert the header
|
|
_, err := db.Exec(logHeaderUpdate, header.Channel, header.Time, header.Status, header.StatusTime, header.Title, header.Tag, header.Description, header.ID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Delete all removed lognicks
|
|
for _, nick := range header.nicksRemoved {
|
|
_, err := db.Exec(logNicksDelete, header.ID, nick)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Insert all new lognicks
|
|
for _, nick := range header.nicksAdded {
|
|
_, err := db.Exec(logNicksInsert, header.ID, nick)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Clear the nicksAdded and nicksRemoved lists
|
|
header.nicksAdded = nil
|
|
header.nicksRemoved = nil
|
|
|
|
// Success
|
|
return nil
|
|
}
|
|
|
|
// NewLogHeader creates a new LogHeader with an ID based the time and channel
|
|
// value. It also sets StatusTime to the current time. It leaves
|
|
// Title, Tag and Nicks blank.
|
|
func NewLogHeader(logTime time.Time, channel, status string) *LogHeader {
|
|
header := new(LogHeader)
|
|
header.ID = logTime.Format("2006-01-02_150405999_") + channel[1:]
|
|
header.Channel = channel
|
|
header.Time = logTime
|
|
header.Status = status
|
|
header.StatusTime = time.Now()
|
|
|
|
return header
|
|
}
|