4d2066a76d
* Start cleaning up linter errors. For #357 * Fix unmarshalling NullTime values * More linter fixes * Remove commented code * Move defer up * Consolidate error check lines * Move error check to make sure row iteration was successful * Cleaner error check + do not recreate pipe if it exists * Consolidate hashing to generate client id
105 lines
2.1 KiB
Go
105 lines
2.1 KiB
Go
package chat
|
|
|
|
import (
|
|
"database/sql"
|
|
"time"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"github.com/owncast/owncast/core/data"
|
|
"github.com/owncast/owncast/models"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var _db *sql.DB
|
|
|
|
func setupPersistence() {
|
|
_db = data.GetDatabase()
|
|
createTable()
|
|
}
|
|
|
|
func createTable() {
|
|
createTableSQL := `CREATE TABLE IF NOT EXISTS messages (
|
|
"id" string NOT NULL PRIMARY KEY,
|
|
"author" TEXT,
|
|
"body" TEXT,
|
|
"messageType" TEXT,
|
|
"visible" INTEGER,
|
|
"timestamp" DATE
|
|
);`
|
|
|
|
stmt, err := _db.Prepare(createTableSQL)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer stmt.Close()
|
|
_, err = stmt.Exec()
|
|
if err != nil {
|
|
log.Warnln(err)
|
|
}
|
|
}
|
|
|
|
func addMessage(message models.ChatMessage) {
|
|
tx, err := _db.Begin()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
stmt, err := tx.Prepare("INSERT INTO messages(id, author, body, messageType, visible, timestamp) values(?, ?, ?, ?, ?, ?)")
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer stmt.Close()
|
|
|
|
_, err = stmt.Exec(message.ID, message.Author, message.Body, message.MessageType, 1, message.Timestamp)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
err = tx.Commit()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func getChatHistory() []models.ChatMessage {
|
|
history := make([]models.ChatMessage, 0)
|
|
|
|
// Get all messages sent within the past day
|
|
rows, err := _db.Query("SELECT * FROM messages WHERE visible = 1 AND messageType != 'SYSTEM' AND datetime(timestamp) >=datetime('now', '-1 Day')")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
for rows.Next() {
|
|
var id string
|
|
var author string
|
|
var body string
|
|
var messageType string
|
|
var visible int
|
|
var timestamp time.Time
|
|
|
|
err = rows.Scan(&id, &author, &body, &messageType, &visible, ×tamp)
|
|
if err != nil {
|
|
log.Debugln(err)
|
|
log.Error("There is a problem with the chat database. Please delete chat.db and restart Owncast.")
|
|
break
|
|
}
|
|
|
|
message := models.ChatMessage{}
|
|
message.ID = id
|
|
message.Author = author
|
|
message.Body = body
|
|
message.MessageType = messageType
|
|
message.Visible = visible == 1
|
|
message.Timestamp = timestamp
|
|
|
|
history = append(history, message)
|
|
}
|
|
|
|
if err := rows.Err(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
return history
|
|
}
|