This commit is contained in:
Radon 2025-01-22 20:04:47 -06:00
parent 8185bb0559
commit 7e617eb1af
11 changed files with 1006 additions and 897 deletions

View File

@ -5,10 +5,9 @@
},
"paths": {
"databasePath": "/home/radon/Documents/chattest.db",
"indexJsPath": "./content/root.js",
"indexCssPath": "./content/root.css",
"indexHtmlPath": "./content/root.html",
"messagesHtmlPath": "./content/messages.html"
"indexJsPath": "./public/index.js",
"indexCssPath": "./public/style.css",
"indexHtmlPath": "./public/index.html"
},
"options": {
"messageMaxAge": 259200,

View File

@ -1,7 +0,0 @@
<html>
<body>
<div class="container">
{{body}}
</div>
</body>
</html>

326
db/db.go Normal file
View File

@ -0,0 +1,326 @@
package db
import (
"database/sql"
"fmt"
_ "github.com/mattn/go-sqlite3"
"strconv"
"time"
)
type User struct {
Id string
Username string
IpAddress string
Timezone string
}
type Message struct {
Id string
SenderIp string
SenderUsername string
Content string
Timestamp string
Edited bool
}
type Database struct {
db *sql.DB
}
func OpenDatabase(filepath string) *Database {
if db, err := sql.Open("sqlite3", filepath); err != nil {
return nil
} else {
return &Database{
db: db,
}
}
}
func (db *Database) Close() {
db.db.Close()
}
func (db *Database) DbCreateTableMessages() {
stmt := `CREATE TABLE IF NOT EXISTS messages (
id INTEGER PRIMARY KEY AUTOINCREMENT,
ip_address TEXT NOT NULL,
content TEXT NOT NULL,
edited INTEGER DEFAULT 0,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
)`
db.db.Exec(stmt)
}
func (db *Database) DbCreateTableUsers() {
stmt := `CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
ip_address TEXT NOT NULL,
username TEXT NOT NULL UNIQUE,
timezone TEXT DEFAULT 'America/New_York',
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
)`
db.db.Exec(stmt)
}
func (db *Database) UserTimezoneSet(ip_address, timezone string) {
_, err := db.db.Exec("UPDATE users SET timezone = ? WHERE ip_address = ?", timezone, ip_address)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UserAdd(ip_address, username string) {
_, err := db.db.Exec("INSERT INTO users (username, ip_address) VALUES (?, ?)", username, ip_address)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) MessageAdd(ip_address string, content string) {
_, err := db.db.Exec("INSERT INTO messages (ip_address, content) VALUES (?, ?)", ip_address, content)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UserNameGet(ip_address string) string {
rows, err := db.db.Query("SELECT username FROM users WHERE ip_address = ?", ip_address)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var username string
rows.Next()
rows.Scan(&username)
return username
}
func (db *Database) UserIpGet(username string) string {
rows, err := db.db.Query("SELECT ip_address FROM users WHERE username = ?", username)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var ip_address string
rows.Next()
rows.Scan(&ip_address)
return ip_address
}
func (db *Database) UserGetTimezone(ip_address string) string {
rows, err := db.db.Query("SELECT timezone FROM users WHERE ip_address = ?", ip_address)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var timezone string
rows.Next()
rows.Scan(&timezone)
return timezone
}
func (db *Database) UsersGet() []User {
rows, err := db.db.Query("SELECT * FROM users")
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var users []User
for rows.Next() {
var id string
var ip_address string
var username string
var created_at string
var timezone string
rows.Scan(&id, &ip_address, &username, &created_at, &timezone)
user := User{
Id: id,
Username: username,
IpAddress: ip_address,
Timezone: timezone,
}
users = append(users, user)
}
return users
}
func (db *Database) MessagesGet() []Message {
rows, err := db.db.Query(`
SELECT messages.id, messages.ip_address, messages.content,
strftime('%Y-%m-%d %H:%M:%S', messages.created_at) as created_at,
users.username, messages.edited
FROM messages
LEFT JOIN users ON messages.ip_address = users.ip_address;
`)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var messages []Message
for rows.Next() {
var id string
var content string
var ip_address string
var created_at string
var username string
var edited int
rows.Scan(&id, &ip_address, &content, &created_at, &username, &edited)
editedBool := false
if edited == 1 {
editedBool = true
}
message := Message{
Id: id,
Content: content,
SenderIp: ip_address,
SenderUsername: username,
Edited: editedBool,
Timestamp: created_at,
}
messages = append(messages, message)
}
return messages
}
func (db *Database) UserNameExists(username string) bool {
rows, err := db.db.Query("SELECT * FROM users WHERE username = ?", username)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
return rows.Next()
}
func (db *Database) UserExists(ip string) bool {
rows, err := db.db.Query("SELECT * FROM users WHERE ip_address = ?", ip)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
return rows.Next()
}
func (db *Database) UserNameChange(ip, newUsername string) {
_, err := db.db.Exec("UPDATE users SET username = ? WHERE ip_address = ?", newUsername, ip)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UserMessagesDelete(ip string) {
_, err := db.db.Exec("DELETE FROM messages WHERE ip_address = ?", ip)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UserMessagesGet(ip string) []Message {
rows, err := db.db.Query(`
SELECT messages.*, users.username
FROM messages
LEFT JOIN users ON messages.ip_address = users.ip_address
WHERE messages.ip_address = ?
ORDER BY messages.created_at DESC;
`, ip)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var messages []Message
for rows.Next() {
var id string
var content string
var ip_address string
var created_at string
var username string
var edited int
rows.Scan(&id, &ip_address, &content, &created_at, &username, &edited)
t, _ := time.Parse(created_at, created_at)
editedBool := false
if edited == 1 {
editedBool = true
}
message := Message{
Id: id,
Content: content,
SenderIp: ip_address,
SenderUsername: username,
Edited: editedBool,
Timestamp: t.Format(created_at),
}
messages = append(messages, message)
}
return messages
}
func (db *Database) MessageDeleteId(id string) {
_, err := db.db.Exec("DELETE FROM messages WHERE id = ?", id)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) MessageDeleteIfOwner(id string, ip string) (int, error) {
res, err := db.db.Exec("DELETE FROM messages WHERE id = ? AND ip_address = ?", id, ip)
if err != nil {
return 0, err
}
affected, err := res.RowsAffected()
if err != nil {
return 0, err
}
return int(affected), nil
}
func (db *Database) MessageEditIfOwner(id string, content string, ip string) (int, error) {
res, err := db.db.Exec("UPDATE messages SET content = ?, edited = 1 WHERE id = ? AND ip_address = ?", content, id, ip)
if err != nil {
return 0, err
}
affected, err := res.RowsAffected()
if err != nil {
return 0, err
}
return int(affected), nil
}
func (db *Database) DeleteOldMessages(ageMinutes int) {
if ageMinutes <= 0 {
return
}
age := strconv.Itoa(ageMinutes)
_, err := db.db.Exec("DELETE FROM messages WHERE created_at < datetime('now', ? || ' minutes')", "-"+age)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UserDeleteIp(ip string) {
_, err := db.db.Exec("DELETE FROM users WHERE ip_address = ?", ip)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UsersDelete() {
_, err := db.db.Exec("DELETE FROM users")
if err != nil {
fmt.Println(err)
}
}
func (db *Database) MessagesDelete() {
_, err := db.db.Exec("DELETE FROM messages")
if err != nil {
fmt.Println(err)
}
}

779
main.go
View File

@ -1,789 +1,22 @@
package main
import (
"compress/gzip"
"database/sql"
"encoding/json"
"chat/srv"
"fmt"
_ "github.com/mattn/go-sqlite3"
"io"
"net"
"net/http"
"os"
"path/filepath"
"strconv"
"strings"
"sync"
"time"
)
type Database struct {
db *sql.DB
}
var config srv.Config
func TimezoneToLocation(timezone string) *time.Location {
defaultLocation := time.FixedZone("UTC", 0)
location, err := time.LoadLocation(timezone)
if err != nil {
return defaultLocation
} else {
return location
}
}
func TimeStringToTime(timeString string) time.Time {
t, _ := time.Parse("2006-01-02 15:04:05", timeString)
return t
}
func TimeStringToTimeInLocation(timeString string, timezone string) string {
t := TimeStringToTime(timeString)
location := TimezoneToLocation(timezone)
return t.In(location).Format("2006-01-02 15:04:05")
}
func OpenDatabase(filepath string) *Database {
if db, err := sql.Open("sqlite3", filepath); err != nil {
return nil
} else {
return &Database{
db: db,
}
}
}
func (db *Database) Close() {
db.db.Close()
}
func (db *Database) DbCreateTableMessages() {
stmt := `CREATE TABLE IF NOT EXISTS messages (
id INTEGER PRIMARY KEY AUTOINCREMENT,
ip_address TEXT NOT NULL,
content TEXT NOT NULL,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
)`
db.db.Exec(stmt)
}
func (db *Database) DbCreateTableUsers() {
stmt := `CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
ip_address TEXT NOT NULL,
username TEXT NOT NULL UNIQUE,
timezone TEXT DEFAULT 'America/New_York',
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
)`
db.db.Exec(stmt)
}
func (db *Database) UserTimezoneSet(ip_address, timezone string) {
_, err := db.db.Exec("UPDATE users SET timezone = ? WHERE ip_address = ?", timezone, ip_address)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UserAdd(ip_address, username string) {
_, err := db.db.Exec("INSERT INTO users (username, ip_address) VALUES (?, ?)", username, ip_address)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) MessageAdd(ip_address string, content string) {
_, err := db.db.Exec("INSERT INTO messages (ip_address, content) VALUES (?, ?)", ip_address, content)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UserNameGet(ip_address string) string {
rows, err := db.db.Query("SELECT username FROM users WHERE ip_address = ?", ip_address)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var username string
rows.Next()
rows.Scan(&username)
return username
}
func (db *Database) UserIpGet(username string) string {
rows, err := db.db.Query("SELECT ip_address FROM users WHERE username = ?", username)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var ip_address string
rows.Next()
rows.Scan(&ip_address)
return ip_address
}
func (db *Database) UserGetTimezone(ip_address string) string {
rows, err := db.db.Query("SELECT timezone FROM users WHERE ip_address = ?", ip_address)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var timezone string
rows.Next()
rows.Scan(&timezone)
return timezone
}
func (db *Database) UsersGet() []User {
rows, err := db.db.Query("SELECT * FROM users")
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var users []User
for rows.Next() {
var id string
var ip_address string
var username string
var created_at string
var timezone string
rows.Scan(&id, &ip_address, &username, &created_at, &timezone)
user := User{
Id: id,
Username: username,
IpAddress: ip_address,
Timezone: timezone,
}
users = append(users, user)
}
return users
}
func (db *Database) MessagesGet() []Message {
rows, err := db.db.Query(`
SELECT messages.id, messages.ip_address, messages.content,
strftime('%Y-%m-%d %H:%M:%S', messages.created_at) as created_at,
users.username
FROM messages
LEFT JOIN users ON messages.ip_address = users.ip_address;
`)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var messages []Message
for rows.Next() {
var id string
var content string
var ip_address string
var created_at string
var username string
rows.Scan(&id, &ip_address, &content, &created_at, &username)
message := Message{
Id: id,
Content: content,
SenderIp: ip_address,
SenderUsername: username,
Timestamp: created_at,
}
messages = append(messages, message)
}
return messages
}
func (db *Database) UserNameExists(username string) bool {
rows, err := db.db.Query("SELECT * FROM users WHERE username = ?", username)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
return rows.Next()
}
func (db *Database) UserExists(ip string) bool {
rows, err := db.db.Query("SELECT * FROM users WHERE ip_address = ?", ip)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
return rows.Next()
}
func (db *Database) UserNameChange(ip, newUsername string) {
_, err := db.db.Exec("UPDATE users SET username = ? WHERE ip_address = ?", newUsername, ip)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UserMessagesDelete(ip string) {
_, err := db.db.Exec("DELETE FROM messages WHERE ip_address = ?", ip)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UserMessagesGet(ip string) []Message {
rows, err := db.db.Query(`
SELECT messages.*, users.username
FROM messages
LEFT JOIN users ON messages.ip_address = users.ip_address
WHERE messages.ip_address = ?
ORDER BY messages.created_at DESC;
`, ip)
if err != nil {
fmt.Println(err)
}
defer rows.Close()
var messages []Message
for rows.Next() {
var id string
var content string
var ip_address string
var created_at string
var username string
rows.Scan(&id, &ip_address, &content, &created_at, &username)
t, _ := time.Parse(created_at, created_at)
message := Message{
Id: id,
Content: content,
SenderIp: ip_address,
SenderUsername: username,
Timestamp: t.Format(created_at),
}
messages = append(messages, message)
}
return messages
}
func (db *Database) MessageDeleteId(id string) {
_, err := db.db.Exec("DELETE FROM messages WHERE id = ?", id)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) MessageDeleteIfOwner(id string, ip string) (int, error) {
res, err := db.db.Exec("DELETE FROM messages WHERE id = ? AND ip_address = ?", id, ip)
if err != nil {
return 0, err
}
affected, err := res.RowsAffected()
if err != nil {
return 0, err
}
return int(affected), nil
}
func (db *Database) MessageEditIfOwner(id string, content string, ip string) (int, error) {
res, err := db.db.Exec("UPDATE messages SET content = ? WHERE id = ? AND ip_address = ?", content, id, ip)
if err != nil {
return 0, err
}
affected, err := res.RowsAffected()
if err != nil {
return 0, err
}
return int(affected), nil
}
func (db *Database) DeleteOldMessages(ageMinutes int) {
if ageMinutes <= 0 {
return
}
age := strconv.Itoa(ageMinutes)
_, err := db.db.Exec("DELETE FROM messages WHERE created_at < datetime('now', ? || ' minutes')", "-"+age)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UserDeleteIp(ip string) {
_, err := db.db.Exec("DELETE FROM users WHERE ip_address = ?", ip)
if err != nil {
fmt.Println(err)
}
}
func (db *Database) UsersDelete() {
_, err := db.db.Exec("DELETE FROM users")
if err != nil {
fmt.Println(err)
}
}
func (db *Database) MessagesDelete() {
_, err := db.db.Exec("DELETE FROM messages")
if err != nil {
fmt.Println(err)
}
}
type gzipResponseWriter struct {
http.ResponseWriter
io.Writer
}
func (g *gzipResponseWriter) Write(data []byte) (int, error) {
return g.Writer.Write(data)
}
func GzipMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if !strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") {
next.ServeHTTP(w, r)
return
}
gz := gzip.NewWriter(w)
defer gz.Close()
w.Header().Set("Content-Encoding", "gzip")
next.ServeHTTP(&gzipResponseWriter{ResponseWriter: w, Writer: gz}, r)
})
}
func getClientIP(r *http.Request) string {
if fwdIP := r.Header.Get("X-Forwarded-For"); fwdIP != "" {
return strings.Split(fwdIP, ",")[0]
}
clientIP := r.RemoteAddr
if host, _, err := net.SplitHostPort(clientIP); err == nil {
return host
}
return clientIP
}
type User struct {
Id string
Username string
IpAddress string
Timezone string
}
type Message struct {
Id string
Content string
SenderIp string
SenderUsername string
Timestamp string
}
type Server struct {
Connected map[string]time.Time // Map IP -> Last activity time
Database *Database
Config Config
mu sync.Mutex // For thread safety
}
func NewServer(config Config) *Server {
return &Server{
Connected: make(map[string]time.Time),
Database: OpenDatabase(config.Paths.DatabasePath),
Config: config,
mu: sync.Mutex{},
}
}
func (s *Server) AddMessage(userip string, contents string) {
s.Database.MessageAdd(userip, contents)
}
func (s *Server) updateActivity(ip string) {
s.mu.Lock()
defer s.mu.Unlock()
s.Connected[ip] = time.Now()
}
func (s *Server) cleanupActivity() {
s.mu.Lock()
defer s.mu.Unlock()
for ip, lastActivity := range s.Connected {
if time.Since(lastActivity) > 10*time.Second {
delete(s.Connected, ip)
}
}
}
func (s *Server) handlePing(w http.ResponseWriter, r *http.Request) {
clientIP := getClientIP(r)
s.updateActivity(clientIP)
s.cleanupActivity()
w.WriteHeader(http.StatusOK)
}
func validUsername(username string) bool {
for _, c := range username {
if !((c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') || (c >= '0' && c <= '9') || c == '_') {
return false
}
}
return true
}
func (s *Server) handleUsername(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodPut {
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
w.Header().Set("Content-Type", "application/json")
clientIP := getClientIP(r)
var req struct {
Username string `json:"username"`
}
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, `{"error": "Invalid JSON"}`, http.StatusBadRequest)
return
}
s.mu.Lock()
if len(req.Username) > s.Config.Options.NameMaxLength {
http.Error(w, fmt.Sprintf(`{"error": "Username too long (%v out of %v characters maximum)"}`, len(req.Username), s.Config.Options.NameMaxLength), http.StatusRequestEntityTooLarge)
s.mu.Unlock()
return
}
if !validUsername(req.Username) {
http.Error(w, fmt.Sprintf(`{"error": "Username must only contain alphanumeric characters and/or underscores"}`), http.StatusBadRequest)
s.mu.Unlock()
return
}
if s.Database.UserNameExists(req.Username) {
http.Error(w, fmt.Sprintf(`{"error": "Username already exists"}`), http.StatusConflict)
s.mu.Unlock()
return
}
if s.Database.UserExists(clientIP) {
s.Database.UserNameChange(clientIP, req.Username)
} else {
s.Database.UserAdd(clientIP, req.Username)
}
s.mu.Unlock()
json.NewEncoder(w).Encode(map[string]string{"status": "Username registered"})
}
func getMessageTemplate(filepath string, body string) string {
contents, _ := os.ReadFile(filepath)
return strings.Replace(string(contents), "{{body}}", body, 1)
}
func (s *Server) handleMessages(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case http.MethodPatch:
w.Header().Set("Content-Type", "application/json")
var req struct {
MessageId string `json:"messageId"`
MessageContent string `json:"messageContent"`
}
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, `{"error": "Invalid JSON"}`, http.StatusBadRequest)
return
}
clientIP := getClientIP(r)
if affected, err := s.Database.MessageEditIfOwner(req.MessageId, req.MessageContent, clientIP); err != nil {
http.Error(w, `{"error": "Unauthorized"}`, http.StatusNotFound)
return
} else if affected == 0 {
http.Error(w, `{"error": "Message not found"}`, http.StatusNotFound)
return
}
json.NewEncoder(w).Encode(map[string]string{
"status": "Message edited successfully",
})
case http.MethodGet:
w.Header().Set("Content-Type", "text/html")
var body string
messages := s.Database.MessagesGet()
for _, msg := range messages {
clientIP := getClientIP(r)
timeZone := s.Database.UserGetTimezone(clientIP)
timeLocal := TimeStringToTimeInLocation(msg.Timestamp, timeZone)
body += fmt.Sprintf(`<p>%s<br><span class="username">%s</span><br><span class="timestamp">%s</span><br><span class="message">%s</span></p>`,
msg.Id, msg.SenderUsername, timeLocal, msg.Content)
}
w.Write([]byte(getMessageTemplate(s.Config.Paths.MessagesHtmlPath, body)))
case http.MethodPut:
w.Header().Set("Content-Type", "application/json")
// Get client's IP
clientIP := getClientIP(r)
s.mu.Lock()
exists := s.Database.UserExists(clientIP)
username := s.Database.UserNameGet(clientIP)
s.mu.Unlock()
if !exists {
errorFmt := fmt.Sprintf(`{"error": "IP %s not registered with username"}`, clientIP)
http.Error(w, errorFmt, http.StatusUnauthorized)
return
}
var msg struct {
Message string `json:"message"`
}
if err := json.NewDecoder(r.Body).Decode(&msg); err != nil {
http.Error(w, `{"error": "Invalid JSON"}`, http.StatusBadRequest)
return
}
s.Database.MessageAdd(clientIP, msg.Message)
json.NewEncoder(w).Encode(map[string]string{
"status": "Message received",
"from": username,
})
case http.MethodDelete:
w.Header().Set("Content-Type", "application/json")
var req struct {
MessageId string `json:"messageId"`
}
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, `{"error": "Invalid JSON"}`, http.StatusBadRequest)
return
}
clientIP := getClientIP(r)
if affected, err := s.Database.MessageDeleteIfOwner(req.MessageId, clientIP); err != nil {
http.Error(w, `{"error": "Unauthorized"}`, http.StatusNotFound)
return
} else if affected == 0 {
http.Error(w, `{"error": "Message not found"}`, http.StatusNotFound)
return
}
json.NewEncoder(w).Encode(map[string]string{
"status": "Message deleted",
})
default:
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
}
func (s *Server) handleUsernameStatus(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
w.Header().Set("Content-Type", "application/json")
clientIP := getClientIP(r)
s.mu.Lock()
exists := s.Database.UserExists(clientIP)
username := s.Database.UserNameGet(clientIP)
s.mu.Unlock()
json.NewEncoder(w).Encode(map[string]interface{}{
"hasUsername": exists,
"username": username,
})
}
func (s *Server) handleUsers(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
w.Header().Set("Content-Type", "application/json")
clientIP := getClientIP(r)
s.updateActivity(clientIP)
s.cleanupActivity()
s.mu.Lock()
var users []string
for ip := range s.Connected {
// for all connected, get their usernames
users = append(users, s.Database.UserNameGet(ip))
}
s.mu.Unlock()
json.NewEncoder(w).Encode(map[string]interface{}{
"users": users,
})
}
func (s *Server) handleTimezone(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodPost {
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
w.Header().Set("Content-Type", "application/json")
clientIP := getClientIP(r)
var req struct {
Timezone string `json:"timezone"`
}
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, `{"error": "Invalid JSON"}`, http.StatusBadRequest)
return
}
s.mu.Lock()
if !s.Database.UserExists(clientIP) {
http.Error(w, `{"error": "User not registered"}`, http.StatusUnauthorized)
s.mu.Unlock()
return
}
s.Database.UserTimezoneSet(clientIP, req.Timezone)
s.mu.Unlock()
json.NewEncoder(w).Encode(map[string]string{
"status": "success",
})
}
func (s *Server) handleRoot(w http.ResponseWriter, r *http.Request) {
if r.URL.Path != "/" {
http.NotFound(w, r)
return
}
w.Header().Set("Content-Type", "text/html")
file := readFile(s.Config.Paths.IndexHtmlPath)
w.Write(file)
}
func readFile(filepath string) []byte {
contents, _ := os.ReadFile(filepath)
return contents
}
func (s *Server) handleJs(w http.ResponseWriter, r *http.Request) {
_ = r
w.Header().Set("Content-Type", "application/javascript")
file := readFile(s.Config.Paths.IndexJsPath)
w.Write(file)
}
func (s *Server) handleCss(w http.ResponseWriter, r *http.Request) {
_ = r
w.Header().Set("Content-Type", "text/css")
file := readFile(s.Config.Paths.IndexCssPath)
w.Write(file)
}
func (s *Server) handleMessagesLength(w http.ResponseWriter, r *http.Request) {
// should return the number of messages in the database
if r.Method != http.MethodGet {
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
w.Header().Set("Content-Type", "application/json")
messages := s.Database.MessagesGet()
json.NewEncoder(w).Encode(map[string]int{
"length": len(messages),
})
}
func (s *Server) Run() {
s.Database.DbCreateTableMessages()
s.Database.DbCreateTableUsers()
s.Database.DeleteOldMessages(s.Config.Options.MessageMaxAge)
handler := http.NewServeMux()
handler.HandleFunc("/ping", s.handlePing)
handler.HandleFunc("/username", s.handleUsername)
handler.HandleFunc("/messages", s.handleMessages)
handler.HandleFunc("/messages/length", s.handleMessagesLength)
handler.HandleFunc("/username/status", s.handleUsernameStatus)
handler.HandleFunc("/users", s.handleUsers)
handler.HandleFunc("/", s.handleRoot)
handler.HandleFunc("/root.js", s.handleJs)
handler.HandleFunc("/root.css", s.handleCss)
handler.HandleFunc("/timezone", s.handleTimezone)
fmt.Printf("Server starting on %s:%d\n", s.Config.Server.IpAddress, s.Config.Server.Port)
defer s.Stop()
if err := http.ListenAndServe(fmt.Sprintf("%s:%d", s.Config.Server.IpAddress, s.Config.Server.Port), GzipMiddleware(handler)); err != nil {
fmt.Printf("Server error: %v\n", err)
}
}
func (s *Server) Stop() {
s.Database.Close()
}
func main() {
func init() {
if len(os.Args) < 2 {
fmt.Printf("Usage: %s <config.json>\n", os.Args[0])
os.Exit(1)
}
_, err := os.OpenFile(os.Args[1], os.O_RDONLY, 0)
if err != nil {
fmt.Println("Error opening config file: ", err)
os.Exit(1)
}
config := LoadConfig(os.Args[1])
config = srv.LoadConfig(os.Args[1])
fmt.Println("Config loaded")
server := NewServer(config)
server.Run()
}
type Config struct {
Server struct {
IpAddress string `json:"ipAddress"`
Port int `json:"port"`
} `json:"server"`
Paths struct {
DatabasePath string `json:"databasePath"`
IndexJsPath string `json:"indexJsPath"`
IndexCssPath string `json:"indexCssPath"`
IndexHtmlPath string `json:"indexHtmlPath"`
MessagesHtmlPath string `json:"messagesHtmlPath"`
} `json:"paths"`
Options struct {
MessageMaxAge int `json:"messageMaxAge"`
NameMaxLength int `json:"nameMaxLength"`
} `json:"options"`
}
func LoadConfig(filepath string) Config {
contents, _ := os.ReadFile(filepath)
var config Config
err := json.Unmarshal(contents, &config)
config.Paths.IndexHtmlPath = pathMaker(config.Paths.IndexHtmlPath)
config.Paths.IndexJsPath = pathMaker(config.Paths.IndexJsPath)
config.Paths.IndexCssPath = pathMaker(config.Paths.IndexCssPath)
config.Paths.MessagesHtmlPath = pathMaker(config.Paths.MessagesHtmlPath)
config.Paths.DatabasePath = pathMaker(config.Paths.DatabasePath)
if err != nil {
fmt.Println("Error parsing config file: ", err)
os.Exit(1)
}
return config
}
func pathMaker(path string) string {
absPath, _ := filepath.Abs(path)
absPath = filepath.Clean(absPath)
fmt.Println(absPath)
return absPath
func main() {
srv.NewServer(config).Run()
}

View File

@ -29,12 +29,26 @@
</button>
<button class="theme-button" onclick="toggleTheme()">
<svg class="theme-icon" viewBox="0 0 24 24">
<!-- Moon icon for dark mode -->
<path class="moon" d="M12 3c-4.97 0-9 4.03-9 9s4.03 9 9 9 9-4.03 9-9c0-.46-.04-.92-.1-1.36-.98 1.37-2.58 2.26-4.4 2.26-3.03 0-5.5-2.47-5.5-5.5 0-1.82.89-3.42 2.26-4.4-.44-.06-.9-.1-1.36-.1z"/>
<!-- Sun icon for light mode -->
<path class="sun" d="M12 7c-2.76 0-5 2.24-5 5s2.24 5 5 5 5-2.24 5-5-2.24-5-5-5zM2 13h2c.55 0 1-.45 1-1s-.45-1-1-1H2c-.55 0-1 .45-1 1s.45 1 1 1zm18 0h2c.55 0 1-.45 1-1s-.45-1-1-1h-2c-.55 0-1 .45-1 1s.45 1 1 1zM11 2v2c0 .55.45 1 1 1s1-.45 1-1V2c0-.55-.45-1-1-1s-1 .45-1 1zm0 18v2c0 .55.45 1 1 1s1-.45 1-1v-2c0-.55-.45-1-1-1s-1 .45-1 1zM5.99 4.58c-.39-.39-1.03-.39-1.41 0-.39.39-.39 1.03 0 1.41l1.06 1.06c.39.39 1.03.39 1.41 0s.39-1.03 0-1.41L5.99 4.58zm12.37 12.37c-.39-.39-1.03-.39-1.41 0-.39.39-.39 1.03 0 1.41l1.06 1.06c.39.39 1.03.39 1.41 0 .39-.39.39-1.03 0-1.41l-1.06-1.06zm1.06-10.96c.39-.39.39-1.03 0-1.41-.39-.39-1.03-.39-1.41 0l-1.06 1.06c-.39.39-.39 1.03 0 1.41s1.03.39 1.41 0l1.06-1.06zM7.05 18.36c.39-.39.39-1.03 0-1.41-.39-.39-1.03-.39-1.41 0l-1.06 1.06c-.39.39-.39 1.03 0 1.41s1.03.39 1.41 0l1.06-1.06z"/>
</svg>
</button>
<div class="search-trigger">
<div class="search-container" id="searchContainer">
<button class="search-button" id="searchButton">
<svg class="search-icon" viewBox="0 0 24 24">
<path d="M15.5 14h-.79l-.28-.27a6.5 6.5 0 1 0-.7.7l.27.28v.79l5 4.99L20.49 19l-4.99-5zm-6 0A4.5 4.5 0 1 1 14 9.5 4.5 4.5 0 0 1 9.5 14z"/>
</svg>
</button>
<input
type="text"
class="search-input"
id="searchInput"
placeholder="Type to search page..."
>
<span class="search-count" id="searchCount"></span>
</div>
</div>
</div>
<div id="current-user" class="current-user"></div>
@ -58,7 +72,6 @@
<!-- Messages will be loaded here -->
</div>
</div>
<div class="message-section">
<div class="message-container">
<textarea id="message" placeholder="Type a message (use shift+return for a new line)" rows="1"></textarea>
@ -67,22 +80,7 @@
<path d="M7.41 8.59L12 13.17l4.59-4.58L18 10l-6 6-6-6 1.41-1.41z"/>
</svg></button>
</div>
<div class="search-trigger">
<div class="search-container" id="searchContainer">
<button class="search-button" id="searchButton">
<div class="search-icon"></div>
</button>
<input
type="text"
class="search-input"
id="searchInput"
placeholder="Type to search page..."
>
<span class="search-count" id="searchCount"></span>
</div>
</div>
<div id="status"></div>
</div>
</body>
</html>

View File

@ -27,17 +27,42 @@ function toggleTheme() {
localStorage.setItem("theme", newTheme);
}
async function editMessage(messageId, content) {
const newContent = prompt("Edit message:", content);
function replaceDiv(oldDiv, newDiv) {
oldDiv.parentNode.replaceChild(newDiv, oldDiv);
}
if (newContent === null) {
function cancelEditMessage(messageId) {
const editMessageDiv = document.getElementById(
".edit-message@" + messageId,
);
replaceDiv(editMessageDiv, editMessageOriginalDiv);
editMessageOriginalDiv = null;
editing = false;
editMessageNumber = null;
}
async function submitEditMessage(messageId) {
const editMessageDiv = document.getElementById(
".edit-message@" + messageId,
);
const content =
editMessageOriginalDiv.querySelector(".content").textContent;
const newContent =
editMessageDiv.querySelector(".edit-message-textarea").value;
if (content === null) {
cancelEditMessage(messageId);
return;
}
if (newContent === "") {
if (content === "") {
cancelEditMessage(messageId);
deleteMessage(messageId);
return;
}
if (newContent === content) {
cancelEditMessage(messageId);
return;
}
@ -54,6 +79,9 @@ async function editMessage(messageId, content) {
});
if (response.ok) {
editing = false;
editMessageNumber = null;
editMessageOriginalDiv = null;
updateMessagesInPlace();
} else {
console.error("Failed to edit message");
@ -63,6 +91,61 @@ async function editMessage(messageId, content) {
}
}
editMessageOriginalDiv = null;
editMessageNumber = null;
editing = false;
async function editMessage(messageId) {
if (editing) {
cancelEditMessage(editMessageNumber);
}
editing = true;
editMessageNumber = messageId;
const originalMessageDiv = document.getElementById(
".message@" + messageId,
);
const height = originalMessageDiv.scrollHeight;
const queryContent = originalMessageDiv.querySelector(".content");
// we need to replace all the html elements with their text content, this includes links like img, a, iframe
queryContent.querySelectorAll("a").forEach((link) => {
link.textContent = link.href;
});
queryContent.querySelectorAll("img").forEach((img) => {
img.textContent = img.src;
});
queryContent.querySelectorAll("iframe").forEach((iframe) => {
iframe.textContent = iframe.src;
});
const content =
originalMessageDiv.querySelector(".content").textContent;
editMessageOriginalDiv = originalMessageDiv.cloneNode(true);
const editMessageDiv = document.createElement("div");
const editTextArea = document.createElement("textarea");
editTextArea.className = "edit-message-textarea";
editTextArea.id = "edit-message-textarea";
editTextArea.value = content;
editTextArea.style.height = height + "px";
editMessageDiv.className = "edit-message";
editMessageDiv.id = ".edit-message@" + messageId;
editMessageDiv.innerHTML += `<div class="button-container">` +
`<button class="edit-message-button" onclick="submitEditMessage('${messageId}')">Submit</button>` +
`<button class="edit-message-button" onclick="cancelEditMessage('${messageId}')">Cancel</button>` +
`</div>`;
editMessageDiv.appendChild(editTextArea);
replaceDiv(originalMessageDiv, editMessageDiv);
}
async function deleteMessage(messageId) {
if (confirm("Delete this message?")) {
try {
@ -101,7 +184,7 @@ function setScrollLocation(height) {
messagesDiv.scrollTop = height;
}
document.addEventListener("click", function (event) {
document.addEventListener("click", function(event) {
const settingsPanel = document.getElementById("settings-panel");
const settingsButton = document.querySelector(".settings-button");
const usersPanel = document.getElementById("users-panel");
@ -122,7 +205,7 @@ document.addEventListener("click", function (event) {
}
});
document.addEventListener("keypress", function (event) {
document.addEventListener("keypress", function(event) {
if (event.key === "Enter" && !event.shiftKey) {
const settingsPanel = document.getElementById("settings-panel");
const inputPanel = document.getElementById("message");
@ -136,13 +219,13 @@ document.addEventListener("keypress", function (event) {
}
});
document.addEventListener("input", function (_) {
document.addEventListener("input", function(_) {
const msg = document.getElementById("message");
msg.style.height = "auto";
msg.style.height = (msg.scrollHeight) + "px";
});
document.addEventListener("blur", function (_) {
document.addEventListener("blur", function(_) {
const msg = document.getElementById("message");
msg.style.height = "auto";
}, true);
@ -178,6 +261,9 @@ async function getMessageCount() {
let lastMessageCount = 0;
async function loadMessages(forceUpdate = false, scrollLocation) {
if (editing) {
return;
}
try {
const newMessageCount = await getMessageCount();
@ -232,6 +318,7 @@ async function loadMessages(forceUpdate = false, scrollLocation) {
"<br>",
);
messageDiv.id = ".message@" + messageId;
const usernameDiv = document.createElement(
"div",
);
@ -255,23 +342,19 @@ async function loadMessages(forceUpdate = false, scrollLocation) {
doc.querySelector("span").textContent;
const isMultiline = contentString.match(
"\\n",
) && true || false;
"\\n",
) && true || false;
if (
compareUsername ===
document.getElementById(
"current-user",
).textContent
document.getElementById(
"current-user",
).textContent
) {
deleteHtml =
`<button class="delete-button" title="Delete message" onclick="deleteMessage('${messageId}')" style="display: inline;">🗑️</button>`;
editHtml =
`<button class="edit-button" title="Edit message" onclick="editMessage('${messageId}', '${contentString}')" style="display: inline;">📝</button>`;
}
// TODO: Add edit for multiline messages
if (isMultiline) {
editHtml = "";
`<button class="edit-button" title="Edit message" onclick="editMessage('${messageId}')" style="display: inline;">📝</button>`;
}
messageDiv.innerHTML = `
<div class="message-header">
@ -296,7 +379,7 @@ async function loadMessages(forceUpdate = false, scrollLocation) {
function contentEmbedding(content) {
return content.replace(
/(?![^<]*>)(https?:\/\/[^\s<]+)/g,
function (url) {
function(url) {
const videoId = getYouTubeID(
url,
);
@ -306,15 +389,15 @@ function contentEmbedding(content) {
height="315"
src="https://www.youtube.com/embed/${videoId}"
frameborder="0"
onerror="console.log('Video failed to load:', this.src); this.style.display='none'"
onerror="console.log('Video failed to load:', this.src); this.value=this.src"
allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture"
allowfullscreen></iframe></div>`;
} else if (isImageUrl(url)) {
return `<div class="image-embed"><img
src="${url}"
alt="Embedded image"
alt="${url}"
loading="lazy"
onerror="console.log('Image failed to load:', this.src); this.style.display='none'"></div>`;
onerror="console.log('Image failed to load:', this.src); this.value=this.src"></div>`;
}
return `<a href="${url}" target="_blank" rel="noopener noreferrer">${url}</a>`;
},
@ -356,7 +439,7 @@ async function checkUsername() {
username.focus();
username.selectionStart =
username.selectionEnd =
username.value.length;
username.value.length;
}
} catch (error) {
console.error("Error checking username status:", error);
@ -523,7 +606,7 @@ function initializeTheme() {
}
}
document.addEventListener("keyup", function (event) {
document.addEventListener("keyup", function(event) {
const inputPanel = document.getElementById("message");
if (inputPanel.contains(event.target) && event.key === "ArrowUp") {
if (inputPanel.value === "") {
@ -681,9 +764,8 @@ function initializeSearchBox() {
behavior: "smooth",
block: "center",
});
searchCount.textContent = `${
currentMatchIndex + 1
}/${matches.length}`;
searchCount.textContent = `${currentMatchIndex + 1
}/${matches.length}`;
}
searchButton.addEventListener("click", () => {
@ -710,7 +792,7 @@ function initializeSearchBox() {
});
// Debug function to check search coverage (call from console: checkSearchCoverage())
window.checkSearchCoverage = function () {
window.checkSearchCoverage = function() {
const textNodes = findTextNodes(document.body);
console.log(
"Total searchable text nodes found:",

View File

@ -42,47 +42,60 @@
--input-button-active-fg: #dce0e8;
}
.edit-message {
display: flex;
flex-direction: column;
margin-top: 10px;
}
.edit-message textarea {
order: 1; /* Make textarea come first */
padding: 10px;
border: none;
border-radius: 4px;
background-color: var(--input-bg-color);
color: var(--input-text-color);
resize: none;
line-height: 1.4;
margin-bottom: 10px;
}
.edit-message .button-container {
order: 2; /* Make buttons container come second */
display: flex;
gap: 10px;
}
.edit-message button {
padding: 10px 20px;
border: none;
border-radius: 4px;
background-color: var(--input-button-inactive-bg);
color: var(--input-button-inactive-fg);
cursor: pointer;
}
.search-trigger {
position: fixed;
bottom: 20px;
right: 20px;
z-index: 1000;
}
.search-container {
position: flex;
display: flex;
position: absolute;
display: inline;
align-items: center;
background: var(--pum-button-active-bg);
background: var(--pum-button-inactive-bg);
border-radius: 24px;
transition: width 0.2s ease;
overflow: hidden;
width: 48px;
height: 48px;
width: 24px;
height: 24px;
}
.search-container.expanded {
width: 300px;
}
.search-button {
background: none;
border: none;
padding: 12px;
cursor: pointer;
display: flex;
align-items: center;
justify-content: center;
border-radius: 50%;
background: var(--pum-button-active-bg);
color: var(--pum-button-inactive-fg);
}
.search-button:hover {
background: var(--pum-button-active-bg);
color: var(--pum-button-active-fg);
}
.search-input {
display: none;
flex: 2;
@ -107,25 +120,6 @@
color: black;
}
/* Search icon using CSS */
.search-icon {
width: 15px;
height: 15px;
border: 2px solid var(--pum-button-inactive-fg);
border-radius: 50%;
position: relative;
}
.search-icon::after {
content: '';
position: absolute;
width: 2px;
height: 10px;
background: var(--pum-button-inactive-fg);
bottom: -8px;
right: -3px;
transform: rotate(-45deg);
}
.search-count {
display: none;
@ -166,20 +160,25 @@ body {
position: fixed;
top: 20px;
left: 50%;
transform: translateX(-50%);
transform: translateX(-65%);
display: flex;
gap: 20px; /* Space between buttons */
align-items: center;
justify-content: center;
}
.header-controls .search-trigger {
position: relative;
right: 0%;
}
/* Show/hide appropriate icon based on theme */
[data-theme="light"] .theme-icon .sun { display: none; }
[data-theme="dark"] .theme-icon .moon { display: none; }
.settings-button, .users-button, .theme-button {
.settings-button, .users-button, .theme-button, .search-button {
top: 20px;
background: none;
border: none;
@ -194,18 +193,6 @@ body {
background-color: var(--pum-button-inactive-bg);
}
.settings-button {
right: calc(50%);
}
.users-button {
right: calc(60%);
}
.theme-button {
right: calc(30%);
}
.message-header {
display: flex;
justify-content: space-between;
@ -251,19 +238,20 @@ body {
background: var(--timestamp-color);
}
.settings-icon, .users-icon, .theme-icon{
.settings-icon, .users-icon, .theme-icon, .search-icon {
width: 24px;
height: 24px;
fill: var(--pum-button-inactive-fg);
}
.settings-button:hover, .users-button:hover, .theme-button:hover {
.settings-button:hover, .users-button:hover, .theme-button:hover, .search-button:hover {
background-color: var(--pum-button-active-bg);
}
.settings-button:hover .settings-icon,
.users-button:hover .users-icon,
.theme-button:hover .theme-icon {
.theme-button:hover .theme-icon,
.search-button:hover .search-icon {
fill: var(--pum-button-active-fg);
}
@ -488,7 +476,7 @@ button.scroll:hover {
gap: 15px;
width: 100%;
justify-content: center;
}
}
.settings-button, .users-button, .theme-button {
width: 36px;

View File

@ -2,17 +2,15 @@
## Frontend
### High Priority
- Lazy load with pagination (frontend and backend)
- Edit support for multiline message
- Clicking edit message should reposition you at the message with it centered or where the textarea edit box is at the top of the screen at least
### Mid Priority
- Other embeds (Twitter posts, spotify tracks, soundcloud, github repos, instagram posts, other video platforms)
### Low Priority
- Reposition the search button
- Fix mobile views instead of hiding elements that you don't want to position properly
- Fix mobile views instead of hiding elements that you don't want to position properly (search, radchat title, username)
## Backend
### High Priority
- Lazy load with pagination (frontend and backend)
### Mid Priority
- Nothing yet
### Low Priority
- Tidy up
- Perhaps a boolean value in the database to be able to display whether or not a message has been edited or not
- Nothing yet

285
srv/handle.go Normal file
View File

@ -0,0 +1,285 @@
package srv
import (
tu "chat/tu"
"encoding/json"
"fmt"
"net/http"
)
func (s *Server) handlePing(w http.ResponseWriter, r *http.Request) {
clientIP := getClientIP(r)
s.updateActivity(clientIP)
s.cleanupActivity()
w.WriteHeader(http.StatusOK)
}
func (s *Server) handleUsername(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodPut {
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
w.Header().Set("Content-Type", "application/json")
clientIP := getClientIP(r)
var req struct {
Username string `json:"username"`
}
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, `{"error": "Invalid JSON"}`, http.StatusBadRequest)
return
}
s.mu.Lock()
if len(req.Username) > s.Config.Options.NameMaxLength {
http.Error(w, fmt.Sprintf(`{"error": "Username too long (%v out of %v characters maximum)"}`, len(req.Username), s.Config.Options.NameMaxLength), http.StatusRequestEntityTooLarge)
s.mu.Unlock()
return
}
if !validUsername(req.Username) {
http.Error(w, fmt.Sprintf(`{"error": "Username must only contain alphanumeric characters and/or underscores"}`), http.StatusBadRequest)
s.mu.Unlock()
return
}
if s.Database.UserNameExists(req.Username) {
http.Error(w, fmt.Sprintf(`{"error": "Username already exists"}`), http.StatusConflict)
s.mu.Unlock()
return
}
if s.Database.UserExists(clientIP) {
s.Database.UserNameChange(clientIP, req.Username)
} else {
s.Database.UserAdd(clientIP, req.Username)
}
s.mu.Unlock()
json.NewEncoder(w).Encode(map[string]string{"status": "Username registered"})
}
func (s *Server) handleMessages(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case http.MethodPatch:
w.Header().Set("Content-Type", "application/json")
var req struct {
MessageId string `json:"messageId"`
MessageContent string `json:"messageContent"`
}
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, `{"error": "Invalid JSON"}`, http.StatusBadRequest)
return
}
clientIP := getClientIP(r)
if affected, err := s.Database.MessageEditIfOwner(req.MessageId, req.MessageContent, clientIP); err != nil {
http.Error(w, `{"error": "Unauthorized"}`, http.StatusNotFound)
return
} else if affected == 0 {
http.Error(w, `{"error": "Message not found"}`, http.StatusNotFound)
return
}
json.NewEncoder(w).Encode(map[string]string{
"status": "Message edited successfully",
})
case http.MethodGet:
w.Header().Set("Content-Type", "text/html")
var body string
messages := s.Database.MessagesGet()
for _, msg := range messages {
clientIP := getClientIP(r)
timeZone := s.Database.UserGetTimezone(clientIP)
timeLocal := tu.TimeStringToTimeInLocation(msg.Timestamp, timeZone)
edited := ""
if msg.Edited {
edited = "(edited)"
}
body += fmt.Sprintf(`<p>%s<br><span class="username">%s</span><br><span class="timestamp">%s %s</span><br><span class="message">%s</span><br></p>`,
msg.Id, msg.SenderUsername, timeLocal, edited, msg.Content)
}
w.Write([]byte(getMessageTemplate(body)))
case http.MethodPut:
w.Header().Set("Content-Type", "application/json")
// Get client's IP
clientIP := getClientIP(r)
s.mu.Lock()
exists := s.Database.UserExists(clientIP)
username := s.Database.UserNameGet(clientIP)
s.mu.Unlock()
if !exists {
errorFmt := fmt.Sprintf(`{"error": "IP %s not registered with username"}`, clientIP)
http.Error(w, errorFmt, http.StatusUnauthorized)
return
}
var msg struct {
Message string `json:"message"`
}
if err := json.NewDecoder(r.Body).Decode(&msg); err != nil {
http.Error(w, `{"error": "Invalid JSON"}`, http.StatusBadRequest)
return
}
s.Database.MessageAdd(clientIP, msg.Message)
json.NewEncoder(w).Encode(map[string]string{
"status": "Message received",
"from": username,
})
case http.MethodDelete:
w.Header().Set("Content-Type", "application/json")
var req struct {
MessageId string `json:"messageId"`
}
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, `{"error": "Invalid JSON"}`, http.StatusBadRequest)
return
}
clientIP := getClientIP(r)
if affected, err := s.Database.MessageDeleteIfOwner(req.MessageId, clientIP); err != nil {
http.Error(w, `{"error": "Unauthorized"}`, http.StatusNotFound)
return
} else if affected == 0 {
http.Error(w, `{"error": "Message not found"}`, http.StatusNotFound)
return
}
json.NewEncoder(w).Encode(map[string]string{
"status": "Message deleted",
})
default:
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
}
func (s *Server) handleUsernameStatus(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
w.Header().Set("Content-Type", "application/json")
clientIP := getClientIP(r)
s.mu.Lock()
exists := s.Database.UserExists(clientIP)
username := s.Database.UserNameGet(clientIP)
s.mu.Unlock()
json.NewEncoder(w).Encode(map[string]interface{}{
"hasUsername": exists,
"username": username,
})
}
func (s *Server) handleUsers(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
w.Header().Set("Content-Type", "application/json")
clientIP := getClientIP(r)
s.updateActivity(clientIP)
s.cleanupActivity()
s.mu.Lock()
var users []string
for ip := range s.Connected {
// for all connected, get their usernames
users = append(users, s.Database.UserNameGet(ip))
}
s.mu.Unlock()
json.NewEncoder(w).Encode(map[string]interface{}{
"users": users,
})
}
func (s *Server) handleTimezone(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodPost {
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
w.Header().Set("Content-Type", "application/json")
clientIP := getClientIP(r)
var req struct {
Timezone string `json:"timezone"`
}
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, `{"error": "Invalid JSON"}`, http.StatusBadRequest)
return
}
s.mu.Lock()
if !s.Database.UserExists(clientIP) {
http.Error(w, `{"error": "User not registered"}`, http.StatusUnauthorized)
s.mu.Unlock()
return
}
s.Database.UserTimezoneSet(clientIP, req.Timezone)
s.mu.Unlock()
json.NewEncoder(w).Encode(map[string]string{
"status": "success",
})
}
func (s *Server) handleRoot(w http.ResponseWriter, r *http.Request) {
if r.URL.Path != "/" {
http.NotFound(w, r)
return
}
w.Header().Set("Content-Type", "text/html")
file := readFile(s.Config.Paths.IndexHtmlPath)
w.Write(file)
}
func (s *Server) handleJs(w http.ResponseWriter, r *http.Request) {
_ = r
w.Header().Set("Content-Type", "application/javascript")
file := readFile(s.Config.Paths.IndexJsPath)
w.Write(file)
}
func (s *Server) handleCss(w http.ResponseWriter, r *http.Request) {
_ = r
w.Header().Set("Content-Type", "text/css")
file := readFile(s.Config.Paths.IndexCssPath)
w.Write(file)
}
func (s *Server) handleMessagesLength(w http.ResponseWriter, r *http.Request) {
// should return the number of messages in the database
if r.Method != http.MethodGet {
http.Error(w, `{"error": "Method not allowed"}`, http.StatusMethodNotAllowed)
return
}
w.Header().Set("Content-Type", "application/json")
messages := s.Database.MessagesGet()
json.NewEncoder(w).Encode(map[string]int{
"length": len(messages),
})
}

181
srv/srv.go Normal file
View File

@ -0,0 +1,181 @@
package srv
import (
db "chat/db"
"compress/gzip"
"encoding/json"
"fmt"
"io"
"net"
"net/http"
"os"
"path/filepath"
"strings"
"sync"
"time"
)
type Config struct {
Server struct {
IpAddress string `json:"ipAddress"`
Port int `json:"port"`
} `json:"server"`
Paths struct {
DatabasePath string `json:"databasePath"`
IndexJsPath string `json:"indexJsPath"`
IndexCssPath string `json:"indexCssPath"`
IndexHtmlPath string `json:"indexHtmlPath"`
MessagesHtmlPath string `json:"messagesHtmlPath"`
} `json:"paths"`
Options struct {
MessageMaxAge int `json:"messageMaxAge"`
NameMaxLength int `json:"nameMaxLength"`
} `json:"options"`
}
func LoadConfig(filepath string) Config {
contents, _ := os.ReadFile(filepath)
var config Config
err := json.Unmarshal(contents, &config)
config.Paths.IndexHtmlPath = pathMaker(config.Paths.IndexHtmlPath)
config.Paths.IndexJsPath = pathMaker(config.Paths.IndexJsPath)
config.Paths.IndexCssPath = pathMaker(config.Paths.IndexCssPath)
config.Paths.MessagesHtmlPath = pathMaker(config.Paths.MessagesHtmlPath)
config.Paths.DatabasePath = pathMaker(config.Paths.DatabasePath)
if err != nil {
fmt.Println("Error parsing config file: ", err)
os.Exit(1)
}
return config
}
func pathMaker(path string) string {
absPath, _ := filepath.Abs(path)
absPath = filepath.Clean(absPath)
fmt.Println(absPath)
return absPath
}
type gzipResponseWriter struct {
http.ResponseWriter
io.Writer
}
func (g *gzipResponseWriter) Write(data []byte) (int, error) {
return g.Writer.Write(data)
}
func GzipMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if !strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") {
next.ServeHTTP(w, r)
return
}
gz := gzip.NewWriter(w)
defer gz.Close()
w.Header().Set("Content-Encoding", "gzip")
next.ServeHTTP(&gzipResponseWriter{ResponseWriter: w, Writer: gz}, r)
})
}
func getMessageTemplate(body string) string {
template := `<html>
<body>
<div class="container">
{{body}}
</div>
</body>
</html>`
return strings.Replace(template, "{{body}}", body, 1)
}
func getClientIP(r *http.Request) string {
if fwdIP := r.Header.Get("X-Forwarded-For"); fwdIP != "" {
return strings.Split(fwdIP, ",")[0]
}
clientIP := r.RemoteAddr
if host, _, err := net.SplitHostPort(clientIP); err == nil {
return host
}
return clientIP
}
func readFile(filepath string) []byte {
contents, _ := os.ReadFile(filepath)
return contents
}
func validUsername(username string) bool {
for _, c := range username {
if !((c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') || (c >= '0' && c <= '9') || c == '_') {
return false
}
}
return true
}
type Server struct {
Connected map[string]time.Time // Map IP -> Last activity time
Database *db.Database
Config Config
mu sync.Mutex // For thread safety
}
func NewServer(config Config) *Server {
return &Server{
Connected: make(map[string]time.Time),
Database: db.OpenDatabase(config.Paths.DatabasePath),
Config: config,
mu: sync.Mutex{},
}
}
func (s *Server) AddMessage(userip string, contents string) {
s.Database.MessageAdd(userip, contents)
}
func (s *Server) updateActivity(ip string) {
s.mu.Lock()
defer s.mu.Unlock()
s.Connected[ip] = time.Now()
}
func (s *Server) cleanupActivity() {
s.mu.Lock()
defer s.mu.Unlock()
for ip, lastActivity := range s.Connected {
if time.Since(lastActivity) > 10*time.Second {
delete(s.Connected, ip)
}
}
}
func (s *Server) Run() {
s.Database.DbCreateTableMessages()
s.Database.DbCreateTableUsers()
s.Database.DeleteOldMessages(s.Config.Options.MessageMaxAge)
handler := http.NewServeMux()
handler.HandleFunc("/", s.handleRoot)
handler.HandleFunc("/root.js", s.handleJs)
handler.HandleFunc("/root.css", s.handleCss)
handler.HandleFunc("/ping", s.handlePing)
handler.HandleFunc("/timezone", s.handleTimezone)
handler.HandleFunc("/username/status", s.handleUsernameStatus)
handler.HandleFunc("/messages/length", s.handleMessagesLength)
handler.HandleFunc("/users", s.handleUsers)
handler.HandleFunc("/username", s.handleUsername)
handler.HandleFunc("/messages", s.handleMessages)
fmt.Printf("Server starting on %s:%d\n", s.Config.Server.IpAddress, s.Config.Server.Port)
defer s.Stop()
if err := http.ListenAndServe(fmt.Sprintf("%s:%d", s.Config.Server.IpAddress, s.Config.Server.Port), GzipMiddleware(handler)); err != nil {
fmt.Printf("Server error: %v\n", err)
}
}
func (s *Server) Stop() {
s.Database.Close()
}

26
tu/tu.go Normal file
View File

@ -0,0 +1,26 @@
package tu
import (
"time"
)
func TimezoneToLocation(timezone string) *time.Location {
defaultLocation := time.FixedZone("UTC", 0)
location, err := time.LoadLocation(timezone)
if err != nil {
return defaultLocation
} else {
return location
}
}
func TimeStringToTime(timeString string) time.Time {
t, _ := time.Parse("2006-01-02 15:04:05", timeString)
return t
}
func TimeStringToTimeInLocation(timeString string, timezone string) string {
t := TimeStringToTime(timeString)
location := TimezoneToLocation(timezone)
return t.In(location).Format("2006-01-02 15:04:05")
}