bf851bd9fc
Added support for dyntmpl to the template system. The Account Dashboard now sort of uses dyntmpl, more work needed here. Renamed the pre_render_view_topic hook to pre_render_topic. Added the GetCurrentLangPack() function. Added the alerts_no_new_alerts phrase. Added the account_level_list phrase. Refactored the route rename logic in the patcher to cut down on the amount of boilerplate. Added more route renames to the patcher. You will need to run the patcher / updater in this commit.
81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
package common
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"github.com/Azareal/Gosora/query_gen"
|
|
)
|
|
|
|
var IPSearch IPSearcher
|
|
|
|
type IPSearcher interface {
|
|
Lookup(ip string) (uids []int, err error)
|
|
}
|
|
|
|
type DefaultIPSearcher struct {
|
|
searchUsers *sql.Stmt
|
|
searchTopics *sql.Stmt
|
|
searchReplies *sql.Stmt
|
|
searchUsersReplies *sql.Stmt
|
|
}
|
|
|
|
// NewDefaultIPSearcher gives you a new instance of DefaultIPSearcher
|
|
func NewDefaultIPSearcher() (*DefaultIPSearcher, error) {
|
|
acc := qgen.NewAcc()
|
|
return &DefaultIPSearcher{
|
|
searchUsers: acc.Select("users").Columns("uid").Where("last_ip = ?").Prepare(),
|
|
searchTopics: acc.Select("users").Columns("uid").InQ("uid", acc.Select("topics").Columns("createdBy").Where("ipaddress = ?")).Prepare(),
|
|
searchReplies: acc.Select("users").Columns("uid").InQ("uid", acc.Select("replies").Columns("createdBy").Where("ipaddress = ?")).Prepare(),
|
|
searchUsersReplies: acc.Select("users").Columns("uid").InQ("uid", acc.Select("users_replies").Columns("createdBy").Where("ipaddress = ?")).Prepare(),
|
|
}, acc.FirstError()
|
|
}
|
|
|
|
func (searcher *DefaultIPSearcher) Lookup(ip string) (uids []int, err error) {
|
|
var uid int
|
|
var reqUserList = make(map[int]bool)
|
|
|
|
var runQuery = func(stmt *sql.Stmt) error {
|
|
rows, err := stmt.Query(ip)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer rows.Close()
|
|
|
|
for rows.Next() {
|
|
err := rows.Scan(&uid)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
reqUserList[uid] = true
|
|
}
|
|
return rows.Err()
|
|
}
|
|
|
|
err = runQuery(searcher.searchUsers)
|
|
if err != nil {
|
|
return uids, err
|
|
}
|
|
err = runQuery(searcher.searchTopics)
|
|
if err != nil {
|
|
return uids, err
|
|
}
|
|
err = runQuery(searcher.searchReplies)
|
|
if err != nil {
|
|
return uids, err
|
|
}
|
|
err = runQuery(searcher.searchUsersReplies)
|
|
if err != nil {
|
|
return uids, err
|
|
}
|
|
|
|
// Convert the user ID map to a slice, then bulk load the users
|
|
uids = make([]int, len(reqUserList))
|
|
var i int
|
|
for userID := range reqUserList {
|
|
uids[i] = userID
|
|
i++
|
|
}
|
|
|
|
return uids, nil
|
|
}
|