2017-06-13 07:12:58 +00:00
package main
2017-08-06 15:22:18 +00:00
import (
"log"
"sync"
"errors"
"strings"
"strconv"
"database/sql"
"./query_gen/lib"
"golang.org/x/crypto/bcrypt"
)
2017-06-13 07:12:58 +00:00
Added the Social Groups plugin. This is still under construction.
Made a few improvements to the ForumStore, including bringing it's API closer in line with the other datastores, adding stubs for future subforum functionality, and improving efficiency in a few places.
The auth interface now handles all the authentication stuff.
Renamed the debug config variable to debug_mode.
Added the PluginPerms API.
Internal Errors will now dump the stack trace in the console.
Added support for installable plugins.
Refactored the routing logic so that the router now handles the common PreRoute logic(exc. /static/)
Added the CreateTable method to the query generator. It might need some tweaking to better support other database systems.
Added the same CreateTable method to the query builder.
Began work on PostgreSQL support.
Added the string-string hook type
Added the pre_render hook type.
Added the ParentID and ParentType fields to forums.
Added the get_forum_url_prefix function.
Added a more generic build_slug function.
Added the get_topic_url_prefix function.
Added the override_perms and override_forum_perms functions for bulk setting and unsetting permissions.
Added more ExtData fields in a few structs and removed them on the Perms struct as the PluginPerms API supersedes them there.
Plugins can now see the router instance.
The plugin initialisation handlers can now throw errors.
Plugins are now initialised after all the forum's subsystems are.
Refactored the unit test logic. For instance, we now use the proper .Log method rather than fmt.Println in many cases.
Sorry, we'll have to break Github's generated file detection, as the build instructions aren't working, unless I put them at the top, and they're far, far more important than getting Github to recognise the generated code as generated code.
Fixed an issue with mysql.go's _init_database() overwriting the dbpassword variable. Not a huge issue, but it is a "gotcha" for those not expecting a ':' at the start.
Fixed an issue with forum creation where the forum permissions didn't get cached.
Fixed a bug in plugin_bbcode where negative numbers in rand would crash Gosora.
Made the outputs of plugin_markdown and plugin_bbcode more compliant with the tests.
Revamped the phrase system to make it easier for us to add language pack related features in the future.
Added the WidgetMenu widget type.
Revamped the theme again. I'm experimenting to see which approach I like most.
- Excuse the little W3C rage. Some things about CSS drive me crazy :p
Tests:
Added 22 bbcode_full_parse tests.
Added 19 bbcode_regex_parse tests.
Added 27 markdown_parse tests.
Added four UserStore tests. More to come when the test database functionality is added.
Added 18 name_to_slug tests.
Hooks:
Added the pre_render hook.
Added the pre_render_forum_list hook.
Added the pre_render_view_forum hook.
Added the pre_render_topic_list hook.
Added the pre_render_view_topic hook.
Added the pre_render_profile hook.
Added the pre_render_custom_page hook.
Added the pre_render_overview hook.
Added the pre_render_create_topic hook.
Added the pre_render_account_own_edit_critical hook.
Added the pre_render_account_own_edit_avatar hook.
Added the pre_render_account_own_edit_username hook.
Added the pre_render_account_own_edit_email hook.
Added the pre_render_login hook.
Added the pre_render_register hook.
Added the pre_render_ban hook.
Added the pre_render_panel_dashboard hook.
Added the pre_render_panel_forums hook.
Added the pre_render_panel_delete_forum hook.
Added the pre_render_panel_edit_forum hook.
Added the pre_render_panel_settings hook.
Added the pre_render_panel_setting hook.
Added the pre_render_panel_plugins hook.
Added the pre_render_panel_users hook.
Added the pre_render_panel_edit_user hook.
Added the pre_render_panel_groups hook.
Added the pre_render_panel_edit_group hook.
Added the pre_render_panel_edit_group_perms hook.
Added the pre_render_panel_themes hook.
Added the pre_render_panel_mod_log hook.
Added the pre_render_error hook.
Added the pre_render_security_error hook.
Added the create_group_preappend hook.
Added the intercept_build_widgets hook.
Added the simple_forum_check_pre_perms hook.
Added the forum_check_pre_perms hook.
2017-07-09 12:06:04 +00:00
// TO-DO: Add the watchdog goroutine
2017-06-13 07:12:58 +00:00
var users UserStore
2017-06-25 09:56:39 +00:00
var err_account_exists = errors . New ( "This username is already in use." )
2017-06-13 07:12:58 +00:00
type UserStore interface {
Load ( id int ) error
Get ( id int ) ( * User , error )
GetUnsafe ( id int ) ( * User , error )
CascadeGet ( id int ) ( * User , error )
2017-08-06 15:22:18 +00:00
//BulkCascadeGet(ids []int) ([]*User, error)
BulkCascadeGetMap ( ids [ ] int ) ( map [ int ] * User , error )
2017-06-13 07:12:58 +00:00
BypassGet ( id int ) ( * User , error )
Set ( item * User ) error
Add ( item * User ) error
AddUnsafe ( item * User ) error
Remove ( id int ) error
RemoveUnsafe ( id int ) error
2017-06-25 09:56:39 +00:00
CreateUser ( username string , password string , email string , group int , active int ) ( int , error )
2017-06-13 07:12:58 +00:00
GetLength ( ) int
GetCapacity ( ) int
}
2017-06-15 11:40:35 +00:00
type MemoryUserStore struct {
2017-06-13 07:12:58 +00:00
items map [ int ] * User
length int
capacity int
2017-06-13 08:56:48 +00:00
get * sql . Stmt
2017-06-25 09:56:39 +00:00
register * sql . Stmt
username_exists * sql . Stmt
2017-06-13 07:12:58 +00:00
sync . RWMutex
}
2017-06-15 11:40:35 +00:00
func NewMemoryUserStore ( capacity int ) * MemoryUserStore {
2017-06-25 09:56:39 +00:00
get_stmt , err := qgen . Builder . SimpleSelect ( "users" , "name, group, is_super_admin, session, email, avatar, message, url_prefix, url_name, level, score, last_ip" , "uid = ?" , "" , "" )
2017-06-13 08:56:48 +00:00
if err != nil {
log . Fatal ( err )
}
2017-06-25 09:56:39 +00:00
// Add an admin version of register_stmt with more flexibility?
// create_account_stmt, err = db.Prepare("INSERT INTO
register_stmt , err := qgen . Builder . SimpleInsert ( "users" , "name, email, password, salt, group, is_super_admin, session, active, message" , "?,?,?,?,?,0,'',?,''" )
if err != nil {
log . Fatal ( err )
}
username_exists_stmt , err := qgen . Builder . SimpleSelect ( "users" , "name" , "name = ?" , "" , "" )
if err != nil {
log . Fatal ( err )
}
2017-06-15 11:40:35 +00:00
return & MemoryUserStore {
2017-06-13 08:56:48 +00:00
items : make ( map [ int ] * User ) ,
capacity : capacity ,
2017-06-25 09:56:39 +00:00
get : get_stmt ,
register : register_stmt ,
username_exists : username_exists_stmt ,
2017-06-13 08:56:48 +00:00
}
2017-06-13 07:12:58 +00:00
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) Get ( id int ) ( * User , error ) {
2017-06-13 08:56:48 +00:00
sus . RLock ( )
item , ok := sus . items [ id ]
sus . RUnlock ( )
2017-06-13 07:12:58 +00:00
if ok {
return item , nil
}
2017-06-28 12:05:26 +00:00
return item , ErrNoRows
2017-06-13 07:12:58 +00:00
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) GetUnsafe ( id int ) ( * User , error ) {
2017-06-13 08:56:48 +00:00
item , ok := sus . items [ id ]
2017-06-13 07:12:58 +00:00
if ok {
return item , nil
}
2017-06-28 12:05:26 +00:00
return item , ErrNoRows
2017-06-13 07:12:58 +00:00
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) CascadeGet ( id int ) ( * User , error ) {
2017-06-13 08:56:48 +00:00
sus . RLock ( )
user , ok := sus . items [ id ]
sus . RUnlock ( )
2017-06-13 07:12:58 +00:00
if ok {
return user , nil
}
user = & User { ID : id , Loggedin : true }
2017-06-13 08:56:48 +00:00
err := sus . get . QueryRow ( id ) . Scan ( & user . Name , & user . Group , & user . Is_Super_Admin , & user . Session , & user . Email , & user . Avatar , & user . Message , & user . URLPrefix , & user . URLName , & user . Level , & user . Score , & user . Last_IP )
2017-06-13 07:12:58 +00:00
if user . Avatar != "" {
if user . Avatar [ 0 ] == '.' {
user . Avatar = "/uploads/avatar_" + strconv . Itoa ( user . ID ) + user . Avatar
}
} else {
2017-07-17 10:23:42 +00:00
user . Avatar = strings . Replace ( config . Noavatar , "{id}" , strconv . Itoa ( user . ID ) , 1 )
2017-06-13 07:12:58 +00:00
}
2017-07-29 14:04:20 +00:00
user . Link = build_profile_url ( name_to_slug ( user . Name ) , id )
2017-06-13 07:12:58 +00:00
user . Tag = groups [ user . Group ] . Tag
init_user_perms ( user )
if err == nil {
2017-06-13 08:56:48 +00:00
sus . Set ( user )
2017-06-13 07:12:58 +00:00
}
return user , err
}
2017-08-06 15:22:18 +00:00
// WARNING: We did a little hack to make this as thin and quick as possible to reduce lock contention, use the * Cascade* methods instead for normal use
func ( sus * MemoryUserStore ) bulkGet ( ids [ ] int ) ( list [ ] * User ) {
list = make ( [ ] * User , len ( ids ) )
sus . RLock ( )
for i , id := range ids {
list [ i ] = sus . items [ id ]
}
sus . RUnlock ( )
return list
}
// TO-DO: Optimise the query to avoid preparing it on the spot? Maybe, use knowledge of the most common IN() parameter counts?
// TO-DO: ID of 0 should always error?
func ( sus * MemoryUserStore ) BulkCascadeGetMap ( ids [ ] int ) ( list map [ int ] * User , err error ) {
var id_count int = len ( ids )
list = make ( map [ int ] * User )
if id_count == 0 {
return list , nil
}
var still_here [ ] int
slice_list := sus . bulkGet ( ids )
for i , slice_item := range slice_list {
if slice_item != nil {
list [ slice_item . ID ] = slice_item
} else {
still_here = append ( still_here , ids [ i ] )
}
}
ids = still_here
// If every user is in the cache, then return immediately
if len ( ids ) == 0 {
return list , nil
}
var qlist string
var uidList [ ] interface { }
for _ , id := range ids {
uidList = append ( uidList , strconv . Itoa ( id ) )
qlist += "?,"
}
qlist = qlist [ 0 : len ( qlist ) - 1 ]
stmt , err := qgen . Builder . SimpleSelect ( "users" , "uid, name, group, is_super_admin, session, email, avatar, message, url_prefix, url_name, level, score, last_ip" , "uid IN(" + qlist + ")" , "" , "" )
if err != nil {
return nil , err
}
rows , err := stmt . Query ( uidList ... )
if err != nil {
return nil , err
}
for rows . Next ( ) {
user := & User { Loggedin : true }
err := rows . Scan ( & user . ID , & user . Name , & user . Group , & user . Is_Super_Admin , & user . Session , & user . Email , & user . Avatar , & user . Message , & user . URLPrefix , & user . URLName , & user . Level , & user . Score , & user . Last_IP )
if err != nil {
return nil , err
}
// Initialise the user
if user . Avatar != "" {
if user . Avatar [ 0 ] == '.' {
user . Avatar = "/uploads/avatar_" + strconv . Itoa ( user . ID ) + user . Avatar
}
} else {
user . Avatar = strings . Replace ( config . Noavatar , "{id}" , strconv . Itoa ( user . ID ) , 1 )
}
user . Link = build_profile_url ( name_to_slug ( user . Name ) , user . ID )
user . Tag = groups [ user . Group ] . Tag
init_user_perms ( user )
// Add it to the cache...
sus . Set ( user )
// Add it to the list to be returned
list [ user . ID ] = user
}
// Did we miss any users?
if id_count > len ( list ) {
var sid_list string
for _ , id := range ids {
_ , ok := list [ id ]
if ! ok {
sid_list += strconv . Itoa ( id ) + ","
}
}
// We probably don't need this, but it might be useful in case of bugs in BulkCascadeGetMap
if sid_list == "" {
if dev . DebugMode {
2017-08-13 11:22:34 +00:00
log . Print ( "This data is sampled later in the BulkCascadeGetMap function, so it might miss the cached IDs" )
log . Print ( "id_count" , id_count )
log . Print ( "ids" , ids )
log . Print ( "list" , list )
2017-08-06 15:22:18 +00:00
}
return list , errors . New ( "We weren't able to find a user, but we don't know which one" )
}
sid_list = sid_list [ 0 : len ( sid_list ) - 1 ]
return list , errors . New ( "Unable to find the users with the following IDs: " + sid_list )
}
return list , nil
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) BypassGet ( id int ) ( * User , error ) {
2017-06-13 07:12:58 +00:00
user := & User { ID : id , Loggedin : true }
2017-06-13 08:56:48 +00:00
err := sus . get . QueryRow ( id ) . Scan ( & user . Name , & user . Group , & user . Is_Super_Admin , & user . Session , & user . Email , & user . Avatar , & user . Message , & user . URLPrefix , & user . URLName , & user . Level , & user . Score , & user . Last_IP )
2017-06-13 07:12:58 +00:00
if user . Avatar != "" {
if user . Avatar [ 0 ] == '.' {
user . Avatar = "/uploads/avatar_" + strconv . Itoa ( user . ID ) + user . Avatar
}
} else {
2017-07-17 10:23:42 +00:00
user . Avatar = strings . Replace ( config . Noavatar , "{id}" , strconv . Itoa ( user . ID ) , 1 )
2017-06-13 07:12:58 +00:00
}
2017-07-29 14:04:20 +00:00
user . Link = build_profile_url ( name_to_slug ( user . Name ) , id )
2017-06-13 07:12:58 +00:00
user . Tag = groups [ user . Group ] . Tag
init_user_perms ( user )
return user , err
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) Load ( id int ) error {
2017-06-13 07:12:58 +00:00
user := & User { ID : id , Loggedin : true }
2017-06-13 08:56:48 +00:00
err := sus . get . QueryRow ( id ) . Scan ( & user . Name , & user . Group , & user . Is_Super_Admin , & user . Session , & user . Email , & user . Avatar , & user . Message , & user . URLPrefix , & user . URLName , & user . Level , & user . Score , & user . Last_IP )
2017-06-13 07:12:58 +00:00
if err != nil {
2017-06-13 08:56:48 +00:00
sus . Remove ( id )
2017-06-13 07:12:58 +00:00
return err
}
if user . Avatar != "" {
if user . Avatar [ 0 ] == '.' {
user . Avatar = "/uploads/avatar_" + strconv . Itoa ( user . ID ) + user . Avatar
}
} else {
2017-07-17 10:23:42 +00:00
user . Avatar = strings . Replace ( config . Noavatar , "{id}" , strconv . Itoa ( user . ID ) , 1 )
2017-06-13 07:12:58 +00:00
}
2017-07-29 14:04:20 +00:00
user . Link = build_profile_url ( name_to_slug ( user . Name ) , id )
2017-06-13 07:12:58 +00:00
user . Tag = groups [ user . Group ] . Tag
init_user_perms ( user )
2017-06-13 08:56:48 +00:00
sus . Set ( user )
2017-06-13 07:12:58 +00:00
return nil
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) Set ( item * User ) error {
2017-06-13 08:56:48 +00:00
sus . Lock ( )
user , ok := sus . items [ item . ID ]
2017-06-13 07:12:58 +00:00
if ok {
2017-06-13 08:56:48 +00:00
sus . Unlock ( )
2017-06-13 07:12:58 +00:00
* user = * item
2017-06-13 08:56:48 +00:00
} else if sus . length >= sus . capacity {
sus . Unlock ( )
2017-06-13 07:12:58 +00:00
return ErrStoreCapacityOverflow
} else {
2017-06-13 08:56:48 +00:00
sus . items [ item . ID ] = item
sus . Unlock ( )
sus . length ++
2017-06-13 07:12:58 +00:00
}
return nil
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) Add ( item * User ) error {
2017-06-13 08:56:48 +00:00
if sus . length >= sus . capacity {
2017-06-13 07:12:58 +00:00
return ErrStoreCapacityOverflow
}
2017-06-13 08:56:48 +00:00
sus . Lock ( )
sus . items [ item . ID ] = item
sus . Unlock ( )
sus . length ++
2017-06-13 07:12:58 +00:00
return nil
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) AddUnsafe ( item * User ) error {
2017-06-13 08:56:48 +00:00
if sus . length >= sus . capacity {
2017-06-13 07:12:58 +00:00
return ErrStoreCapacityOverflow
}
2017-06-13 08:56:48 +00:00
sus . items [ item . ID ] = item
sus . length ++
2017-06-13 07:12:58 +00:00
return nil
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) Remove ( id int ) error {
2017-06-13 08:56:48 +00:00
sus . Lock ( )
delete ( sus . items , id )
sus . Unlock ( )
sus . length --
2017-06-13 07:12:58 +00:00
return nil
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) RemoveUnsafe ( id int ) error {
2017-06-13 08:56:48 +00:00
delete ( sus . items , id )
sus . length --
2017-06-13 07:12:58 +00:00
return nil
}
2017-06-25 09:56:39 +00:00
func ( sus * MemoryUserStore ) CreateUser ( username string , password string , email string , group int , active int ) ( int , error ) {
// Is this username already taken..?
err := sus . username_exists . QueryRow ( username ) . Scan ( & username )
2017-06-28 12:05:26 +00:00
if err != ErrNoRows {
2017-06-25 09:56:39 +00:00
return 0 , err_account_exists
}
salt , err := GenerateSafeString ( saltLength )
if err != nil {
return 0 , err
}
hashed_password , err := bcrypt . GenerateFromPassword ( [ ] byte ( password + salt ) , bcrypt . DefaultCost )
if err != nil {
return 0 , err
}
res , err := sus . register . Exec ( username , email , string ( hashed_password ) , salt , group , active )
if err != nil {
return 0 , err
}
lastId , err := res . LastInsertId ( )
return int ( lastId ) , err
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) GetLength ( ) int {
2017-06-13 08:56:48 +00:00
return sus . length
2017-06-13 07:12:58 +00:00
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) SetCapacity ( capacity int ) {
2017-06-13 08:56:48 +00:00
sus . capacity = capacity
2017-06-13 07:12:58 +00:00
}
2017-06-15 11:40:35 +00:00
func ( sus * MemoryUserStore ) GetCapacity ( ) int {
2017-06-13 08:56:48 +00:00
return sus . capacity
2017-06-13 07:12:58 +00:00
}
type SqlUserStore struct {
2017-06-14 07:09:44 +00:00
get * sql . Stmt
2017-06-25 09:56:39 +00:00
register * sql . Stmt
username_exists * sql . Stmt
2017-06-13 07:12:58 +00:00
}
func NewSqlUserStore ( ) * SqlUserStore {
2017-06-25 09:56:39 +00:00
get_stmt , err := qgen . Builder . SimpleSelect ( "users" , "name, group, is_super_admin, session, email, avatar, message, url_prefix, url_name, level, score, last_ip" , "uid = ?" , "" , "" )
2017-06-13 08:56:48 +00:00
if err != nil {
log . Fatal ( err )
}
2017-06-25 09:56:39 +00:00
// Add an admin version of register_stmt with more flexibility?
// create_account_stmt, err = db.Prepare("INSERT INTO
register_stmt , err := qgen . Builder . SimpleInsert ( "users" , "name, email, password, salt, group, is_super_admin, session, active, message" , "?,?,?,?,?,0,'',?,''" )
if err != nil {
log . Fatal ( err )
}
username_exists_stmt , err := qgen . Builder . SimpleSelect ( "users" , "name" , "name = ?" , "" , "" )
if err != nil {
log . Fatal ( err )
}
return & SqlUserStore {
get : get_stmt ,
register : register_stmt ,
username_exists : username_exists_stmt ,
}
2017-06-13 07:12:58 +00:00
}
func ( sus * SqlUserStore ) Get ( id int ) ( * User , error ) {
user := User { ID : id , Loggedin : true }
2017-06-13 08:56:48 +00:00
err := sus . get . QueryRow ( id ) . Scan ( & user . Name , & user . Group , & user . Is_Super_Admin , & user . Session , & user . Email , & user . Avatar , & user . Message , & user . URLPrefix , & user . URLName , & user . Level , & user . Score , & user . Last_IP )
2017-06-13 07:12:58 +00:00
if user . Avatar != "" {
if user . Avatar [ 0 ] == '.' {
user . Avatar = "/uploads/avatar_" + strconv . Itoa ( user . ID ) + user . Avatar
}
} else {
2017-07-17 10:23:42 +00:00
user . Avatar = strings . Replace ( config . Noavatar , "{id}" , strconv . Itoa ( user . ID ) , 1 )
2017-06-13 07:12:58 +00:00
}
2017-07-29 14:04:20 +00:00
user . Link = build_profile_url ( name_to_slug ( user . Name ) , id )
2017-06-13 07:12:58 +00:00
user . Tag = groups [ user . Group ] . Tag
init_user_perms ( & user )
return & user , err
}
func ( sus * SqlUserStore ) GetUnsafe ( id int ) ( * User , error ) {
user := User { ID : id , Loggedin : true }
2017-06-13 08:56:48 +00:00
err := sus . get . QueryRow ( id ) . Scan ( & user . Name , & user . Group , & user . Is_Super_Admin , & user . Session , & user . Email , & user . Avatar , & user . Message , & user . URLPrefix , & user . URLName , & user . Level , & user . Score , & user . Last_IP )
2017-06-13 07:12:58 +00:00
if user . Avatar != "" {
if user . Avatar [ 0 ] == '.' {
user . Avatar = "/uploads/avatar_" + strconv . Itoa ( user . ID ) + user . Avatar
}
} else {
2017-07-17 10:23:42 +00:00
user . Avatar = strings . Replace ( config . Noavatar , "{id}" , strconv . Itoa ( user . ID ) , 1 )
2017-06-13 07:12:58 +00:00
}
2017-07-29 14:04:20 +00:00
user . Link = build_profile_url ( name_to_slug ( user . Name ) , id )
2017-06-13 07:12:58 +00:00
user . Tag = groups [ user . Group ] . Tag
init_user_perms ( & user )
return & user , err
}
func ( sus * SqlUserStore ) CascadeGet ( id int ) ( * User , error ) {
user := User { ID : id , Loggedin : true }
2017-06-13 08:56:48 +00:00
err := sus . get . QueryRow ( id ) . Scan ( & user . Name , & user . Group , & user . Is_Super_Admin , & user . Session , & user . Email , & user . Avatar , & user . Message , & user . URLPrefix , & user . URLName , & user . Level , & user . Score , & user . Last_IP )
2017-06-13 07:12:58 +00:00
if user . Avatar != "" {
if user . Avatar [ 0 ] == '.' {
user . Avatar = "/uploads/avatar_" + strconv . Itoa ( user . ID ) + user . Avatar
}
} else {
2017-07-17 10:23:42 +00:00
user . Avatar = strings . Replace ( config . Noavatar , "{id}" , strconv . Itoa ( user . ID ) , 1 )
2017-06-13 07:12:58 +00:00
}
2017-07-29 14:04:20 +00:00
user . Link = build_profile_url ( name_to_slug ( user . Name ) , id )
2017-06-13 07:12:58 +00:00
user . Tag = groups [ user . Group ] . Tag
init_user_perms ( & user )
return & user , err
}
2017-08-06 15:22:18 +00:00
// TO-DO: Optimise the query to avoid preparing it on the spot? Maybe, use knowledge of the most common IN() parameter counts?
func ( sus * SqlUserStore ) BulkCascadeGetMap ( ids [ ] int ) ( list map [ int ] * User , err error ) {
var qlist string
var uidList [ ] interface { }
for _ , id := range ids {
uidList = append ( uidList , strconv . Itoa ( id ) )
qlist += "?,"
}
qlist = qlist [ 0 : len ( qlist ) - 1 ]
stmt , err := qgen . Builder . SimpleSelect ( "users" , "uid, name, group, is_super_admin, session, email, avatar, message, url_prefix, url_name, level, score, last_ip" , "uid IN(" + qlist + ")" , "" , "" )
if err != nil {
return nil , err
}
rows , err := stmt . Query ( uidList ... )
if err != nil {
return nil , err
}
list = make ( map [ int ] * User )
for rows . Next ( ) {
user := & User { Loggedin : true }
err := rows . Scan ( & user . ID , & user . Name , & user . Group , & user . Is_Super_Admin , & user . Session , & user . Email , & user . Avatar , & user . Message , & user . URLPrefix , & user . URLName , & user . Level , & user . Score , & user . Last_IP )
if err != nil {
return nil , err
}
// Initialise the user
if user . Avatar != "" {
if user . Avatar [ 0 ] == '.' {
user . Avatar = "/uploads/avatar_" + strconv . Itoa ( user . ID ) + user . Avatar
}
} else {
user . Avatar = strings . Replace ( config . Noavatar , "{id}" , strconv . Itoa ( user . ID ) , 1 )
}
user . Link = build_profile_url ( name_to_slug ( user . Name ) , user . ID )
user . Tag = groups [ user . Group ] . Tag
init_user_perms ( user )
// Add it to the list to be returned
list [ user . ID ] = user
}
return list , nil
}
2017-06-13 07:12:58 +00:00
func ( sus * SqlUserStore ) BypassGet ( id int ) ( * User , error ) {
user := User { ID : id , Loggedin : true }
2017-06-13 08:56:48 +00:00
err := sus . get . QueryRow ( id ) . Scan ( & user . Name , & user . Group , & user . Is_Super_Admin , & user . Session , & user . Email , & user . Avatar , & user . Message , & user . URLPrefix , & user . URLName , & user . Level , & user . Score , & user . Last_IP )
2017-06-13 07:12:58 +00:00
if user . Avatar != "" {
if user . Avatar [ 0 ] == '.' {
user . Avatar = "/uploads/avatar_" + strconv . Itoa ( user . ID ) + user . Avatar
}
} else {
2017-07-17 10:23:42 +00:00
user . Avatar = strings . Replace ( config . Noavatar , "{id}" , strconv . Itoa ( user . ID ) , 1 )
2017-06-13 07:12:58 +00:00
}
2017-07-29 14:04:20 +00:00
user . Link = build_profile_url ( name_to_slug ( user . Name ) , id )
2017-06-13 07:12:58 +00:00
user . Tag = groups [ user . Group ] . Tag
init_user_perms ( & user )
return & user , err
}
func ( sus * SqlUserStore ) Load ( id int ) error {
user := & User { ID : id }
// Simplify this into a quick check whether the user exists
2017-06-13 08:56:48 +00:00
err := sus . get . QueryRow ( id ) . Scan ( & user . Name , & user . Group , & user . Is_Super_Admin , & user . Session , & user . Email , & user . Avatar , & user . Message , & user . URLPrefix , & user . URLName , & user . Level , & user . Score , & user . Last_IP )
2017-06-13 07:12:58 +00:00
return err
}
2017-06-25 09:56:39 +00:00
func ( sus * SqlUserStore ) CreateUser ( username string , password string , email string , group int , active int ) ( int , error ) {
// Is this username already taken..?
err := sus . username_exists . QueryRow ( username ) . Scan ( & username )
2017-06-28 12:05:26 +00:00
if err != ErrNoRows {
2017-06-25 09:56:39 +00:00
return 0 , err_account_exists
}
salt , err := GenerateSafeString ( saltLength )
if err != nil {
return 0 , err
}
hashed_password , err := bcrypt . GenerateFromPassword ( [ ] byte ( password + salt ) , bcrypt . DefaultCost )
if err != nil {
return 0 , err
}
res , err := sus . register . Exec ( username , email , string ( hashed_password ) , salt , group , active )
if err != nil {
return 0 , err
}
lastId , err := res . LastInsertId ( )
return int ( lastId ) , err
}
2017-06-28 12:05:26 +00:00
// Placeholder methods, as we're not don't need to do any cache management with this implementation ofr the UserStore
2017-06-13 07:12:58 +00:00
func ( sus * SqlUserStore ) Set ( item * User ) error {
return nil
}
func ( sus * SqlUserStore ) Add ( item * User ) error {
return nil
}
func ( sus * SqlUserStore ) AddUnsafe ( item * User ) error {
return nil
}
func ( sus * SqlUserStore ) Remove ( id int ) error {
return nil
}
func ( sus * SqlUserStore ) RemoveUnsafe ( id int ) error {
return nil
}
func ( sus * SqlUserStore ) GetCapacity ( ) int {
return 0
}
func ( sus * SqlUserStore ) GetLength ( ) int {
return 0 // Return the total number of users registered on the forums?
}