Add: set user acitve

pkg/api/api.go

@@ -83,6 +83,8 @@ func NewAPI(config Config) (*API, error) {
 	apiMux.HandleFunc("/login", api.HandleLogin)
 	apiMux.HandleFunc("/register", api.HandleRegister)
 	apiMux.HandleFunc("/logout", api.LoginAsAnonymous)
+	apiMux.HandleFunc("/get_users", api.HandleGetUsers)
+	apiMux.HandleFunc("/update_user_active", api.HandleUpdateUserActive)
 	// tag
 	apiMux.HandleFunc("/get_tags", api.HandleGetTags)
 	apiMux.HandleFunc("/get_tag_info", api.HandleGetTagInfo)

pkg/api/handle_review.go

@@ -182,3 +182,28 @@ func (api *API) HandleDeleteReview(w http.ResponseWriter, r *http.Request) {
 
 	api.HandleOK(w, r)
 }
+
+type GetReviewsByUserRequest struct {
+	ID int64 `json:"id"`
+}
+
+func (api *API) HandleGetReviewsByUser(w http.ResponseWriter, r *http.Request) {
+	req := &GetReviewsByUserRequest{}
+	err := json.NewDecoder(r.Body).Decode(req)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	reviews, err := api.Db.GetReviewsByUser(req.ID)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	err = json.NewEncoder(w).Encode(reviews)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+}

pkg/api/handle_user.go

@@ -133,7 +133,7 @@ func (api *API) HandleRegister(w http.ResponseWriter, r *http.Request) {
 		api.HandleError(w, r, err)
 		return
 	}
-	
+
 	api.HandleOK(w, r)
 }
 
@@ -185,27 +185,57 @@ func (api *API) GetUserID(w http.ResponseWriter, r *http.Request) (int64, error)
 	return userId.(int64), nil
 }
 
-type GetReviewsByUserRequest struct {
-	ID int64 `json:"id"`
+type GetUsersResponse struct {
+	Users []*database.User `json:"users"`
 }
 
-func (api *API) HandleGetReviewsByUser(w http.ResponseWriter, r *http.Request) {
-	req := &GetReviewsByUserRequest{}
-	err := json.NewDecoder(r.Body).Decode(req)
+func (api *API) HandleGetUsers(w http.ResponseWriter, r *http.Request) {
+	err := api.CheckAdmin(w, r)
 	if err != nil {
 		api.HandleError(w, r, err)
 		return
 	}
 
-	reviews, err := api.Db.GetReviewsByUser(req.ID)
+	users, err := api.Db.GetUsers()
 	if err != nil {
 		api.HandleError(w, r, err)
 		return
 	}
 
-	err = json.NewEncoder(w).Encode(reviews)
+	ret := &GetUsersResponse{
+		Users: users,
+	}
+
+	err = json.NewEncoder(w).Encode(ret)
 	if err != nil {
 		api.HandleError(w, r, err)
 		return
 	}
 }
+
+type UpdateUserActiveRequest struct {
+	ID     int64 `json:"id"`
+	Active bool  `json:"active"`
+}
+
+func (api *API) HandleUpdateUserActive(w http.ResponseWriter, r *http.Request) {
+	err := api.CheckAdmin(w, r)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	req := &UpdateUserActiveRequest{}
+	err = json.NewDecoder(r.Body).Decode(req)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	err = api.Db.UpdateUserActive(req.ID, req.Active)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+	api.HandleOK(w, r)
+}

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"`
 }