Add: set user acitve

pkg/database/method_user.go

@@ -22,7 +22,7 @@ func (database *Database) LoginAsAnonymous() (*User, error) {
 	return user, nil
 }
 
-func (database *Database) Register(username string, password string, usertype int64) (error) {
+func (database *Database) Register(username string, password string, usertype int64) error {
 	countAdmin, err := database.CountAdmin()
 	if err != nil {
 		return err
@@ -64,3 +64,31 @@ func (database *Database) CountAdmin() (int64, error) {
 	}
 	return count, nil
 }
+
+func (database *Database) GetUsers() ([]*User, error) {
+	users := make([]*User, 0)
+
+	rows, err := database.stmt.getUsers.Query()
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	for rows.Next() {
+		user := &User{}
+		err = rows.Scan(&user.ID, &user.Username, &user.Role, &user.Active, &user.AvatarId)
+		if err != nil {
+			return nil, err
+		}
+		users = append(users, user)
+	}
+	return users, nil
+}
+
+func (database *Database) UpdateUserActive(id int64, active bool) error {
+	_, err := database.stmt.updateUserActive.Exec(active, id)
+	if err != nil {
+		return err
+	}
+	return nil
+}

pkg/database/sql_stmt.go

@@ -180,8 +180,12 @@ var countAdminQuery = `SELECT count(*) FROM users WHERE role= 1;`
 
 var getUserQuery = `SELECT id, username, role, avatar_id FROM users WHERE username = ? AND password = ? LIMIT 1;`
 
+var getUsersQuery = `SELECT id, username, role, active, avatar_id FROM users;`
+
 var getUserByIdQuery = `SELECT id, username, role, avatar_id FROM users WHERE id = ? LIMIT 1;`
 
+var updateUserActiveQuery = `UPDATE users SET active = ? WHERE id = ?;`
+
 var getAnonymousUserQuery = `SELECT id, username, role, avatar_id FROM users WHERE role = 0 LIMIT 1;`
 
 var insertTagQuery = `INSERT INTO tags (name, description, created_by_user_id) VALUES (?, ?, ?);`
@@ -276,7 +280,9 @@ type Stmt struct {
 	countUser             *sql.Stmt
 	countAdmin            *sql.Stmt
 	getUser               *sql.Stmt
+	getUsers              *sql.Stmt
 	getUserById           *sql.Stmt
+	updateUserActive      *sql.Stmt
 	getAnonymousUser      *sql.Stmt
 	insertTag             *sql.Stmt
 	getTag                *sql.Stmt
@@ -529,12 +535,24 @@ func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
 		return nil, err
 	}
 
+	// init getUsers
+	stmt.getUsers, err = sqlConn.Prepare(getUsersQuery)
+	if err != nil {
+		return nil, err
+	}
+
 	// init getUserById
 	stmt.getUserById, err = sqlConn.Prepare(getUserByIdQuery)
 	if err != nil {
 		return nil, err
 	}
 
+	// init updateUserActive
+	stmt.updateUserActive, err = sqlConn.Prepare(updateUserActiveQuery)
+	if err != nil {
+		return nil, err
+	}
+
 	// init getAnonymousUser
 	stmt.getAnonymousUser, err = sqlConn.Prepare(getAnonymousUserQuery)
 	if err != nil {
@@ -548,7 +566,7 @@ func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
 		return nil, err
 	}
 	if userCount == 0 {
-		_, err = stmt.insertUser.Exec("Anonymous user", "", 0, 0)
+		_, err = stmt.insertUser.Exec("Anonymous user", "", 0, 1, 0)
 		if err != nil {
 			return nil, err
 		}

pkg/database/struct.go

@@ -25,6 +25,7 @@ type User struct {
 	Username string `json:"username"`
 	Password string `json:"-"`
 	Role     int64  `json:"role"`
+	Active   bool   `json:"active"`
 	AvatarId int64  `json:"avatar_id"`
 }