From 5a9dced2497185dd6e13556ce86d7b54ce317d4b Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Thu, 26 Jan 2023 12:34:53 +0100
Subject: [PATCH] database: store user creation date

This is useful to figure out which users were created recently.
---
 database/postgres.go |  4 +++-
 database/sqlite.go   | 15 ++++++++++++---
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/database/postgres.go b/database/postgres.go
index ce940bb..2387229 100644
--- a/database/postgres.go
+++ b/database/postgres.go
@@ -31,7 +31,8 @@ CREATE TABLE "User" (
 	password VARCHAR(255),
 	admin BOOLEAN NOT NULL DEFAULT FALSE,
 	nick VARCHAR(255),
-	realname VARCHAR(255)
+	realname VARCHAR(255),
+	created_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now()
 );
 
 CREATE TYPE sasl_mechanism AS ENUM ('PLAIN', 'EXTERNAL');
@@ -167,6 +168,7 @@ var postgresMigrations = []string{
 	`,
 	`ALTER TABLE "Network" ADD COLUMN auto_away BOOLEAN NOT NULL DEFAULT TRUE`,
 	`ALTER TABLE "Network" ADD COLUMN certfp TEXT`,
+	`ALTER TABLE "User" ADD COLUMN created_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now()`,
 }
 
 type PostgresDB struct {
diff --git a/database/sqlite.go b/database/sqlite.go
index 6974ba4..a2dbbc7 100644
--- a/database/sqlite.go
+++ b/database/sqlite.go
@@ -20,6 +20,7 @@ const SqliteEnabled = true
 const sqliteQueryTimeout = 5 * time.Second
 
 const sqliteTimeLayout = "2006-01-02T15:04:05.000Z"
+const sqliteTimeFormat = "%Y-%m-%dT%H:%M:%fZ"
 
 type sqliteTime struct {
 	time.Time
@@ -60,7 +61,8 @@ CREATE TABLE User (
 	password TEXT,
 	admin INTEGER NOT NULL DEFAULT 0,
 	realname TEXT,
-	nick TEXT
+	nick TEXT,
+	created_at TEXT NOT NULL
 );
 
 CREATE TABLE Network (
@@ -281,6 +283,12 @@ var sqliteMigrations = []string{
 	"ALTER TABLE User ADD COLUMN nick TEXT;",
 	"ALTER TABLE Network ADD COLUMN auto_away INTEGER NOT NULL DEFAULT 1;",
 	"ALTER TABLE Network ADD COLUMN certfp TEXT;",
+	// SQLite doesn't support non-constant default values, so use an empty
+	// string as default and update all columns in a separate statement
+	`
+		ALTER TABLE User ADD COLUMN created_at TEXT NOT NULL DEFAULT '';
+		UPDATE User SET created_at = strftime('` + sqliteTimeFormat + `', 'now');
+	`,
 }
 
 type SqliteDB struct {
@@ -434,6 +442,7 @@ func (db *SqliteDB) StoreUser(ctx context.Context, user *User) error {
 		sql.Named("admin", user.Admin),
 		sql.Named("nick", toNullString(user.Nick)),
 		sql.Named("realname", toNullString(user.Realname)),
+		sql.Named("now", sqliteTime{time.Now()}),
 	}
 
 	var err error
@@ -446,8 +455,8 @@ func (db *SqliteDB) StoreUser(ctx context.Context, user *User) error {
 		var res sql.Result
 		res, err = db.db.ExecContext(ctx, `
 			INSERT INTO
-			User(username, password, admin, nick, realname)
-			VALUES (:username, :password, :admin, :nick, :realname)`,
+			User(username, password, admin, nick, realname, created_at)
+			VALUES (:username, :password, :admin, :nick, :realname, :now)`,
 			args...)
 		if err != nil {
 			return err