stuff
This commit is contained in:
BIN
RBU_account
BIN
RBU_account
Binary file not shown.
@ -1,3 +1,4 @@
|
|||||||
{
|
{
|
||||||
"createGiteaAccount": false
|
"createGiteaAccount": false,
|
||||||
|
"port": 8080
|
||||||
}
|
}
|
||||||
|
|||||||
88
src/main.go
88
src/main.go
@ -1,7 +1,6 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"crypto/rand"
|
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"github.com/bwmarrin/discordgo"
|
"github.com/bwmarrin/discordgo"
|
||||||
@ -13,11 +12,10 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"regexp"
|
"regexp"
|
||||||
"golang.org/x/crypto/argon2"
|
|
||||||
"context"
|
|
||||||
"time"
|
|
||||||
"github.com/cornelk/hashmap"
|
"github.com/cornelk/hashmap"
|
||||||
"code.gitea.io/sdk/gitea"
|
"code.gitea.io/sdk/gitea"
|
||||||
|
"fmt"
|
||||||
|
"golang.org/x/crypto/argon2"
|
||||||
)
|
)
|
||||||
var discord *discordgo.Session
|
var discord *discordgo.Session
|
||||||
var secret secrets_json
|
var secret secrets_json
|
||||||
@ -26,6 +24,7 @@ var cacheAccounts hashmap.HashMap
|
|||||||
var db *sql.DB
|
var db *sql.DB
|
||||||
var giteaClient *gitea.Client
|
var giteaClient *gitea.Client
|
||||||
var registerTmpl *template.Template
|
var registerTmpl *template.Template
|
||||||
|
var submitTmpl *template.Template
|
||||||
type account struct {
|
type account struct {
|
||||||
email string
|
email string
|
||||||
username string
|
username string
|
||||||
@ -60,6 +59,7 @@ type secrets_json struct {
|
|||||||
}
|
}
|
||||||
type config_json struct {
|
type config_json struct {
|
||||||
CreateGiteaAccount bool `json:"createGiteaAccount"`
|
CreateGiteaAccount bool `json:"createGiteaAccount"`
|
||||||
|
Port uint16 `json:"port"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
@ -99,65 +99,13 @@ func main() {
|
|||||||
moodle := moodle.NewMoodleApi("https://exam.redstoneunion.de/", secret.MoodleToken)
|
moodle := moodle.NewMoodleApi("https://exam.redstoneunion.de/", secret.MoodleToken)
|
||||||
_ = moodle
|
_ = moodle
|
||||||
registerTmpl = template.Must(template.ParseFiles("tmpl/register.html"))
|
registerTmpl = template.Must(template.ParseFiles("tmpl/register.html"))
|
||||||
submitTmpl := template.Must(template.ParseFiles("tmpl/submit.html"))
|
submitTmpl = template.Must(template.ParseFiles("tmpl/submit.html"))
|
||||||
remail = regexp2.MustCompile("^(?=.{0,255}$)(?:[a-z0-9!#$%&'*+/=?^_`{|}~-]+(?:\\.[a-z0-9!#$%&'*+/=?^_`{|}~-]+)*|\"(?:[\\x01-\\x08\\x0b\\x0c\\x0e-\\x1f\\x21\\x23-\\x5b\\x5d-\\x7f]|\\\\[\\x01-\\x09\\x0b\\x0c\\x0e-\\x7f])*\")@(?:(?:[a-z0-9](?:[a-z0-9-]*[a-z0-9])?\\.)+[a-z0-9](?:[a-z0-9-]*[a-z0-9])?|\\[(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?|[a-z0-9-]*[a-z0-9]:(?:[\\x01-\\x08\\x0b\\x0c\\x0e-\\x1f\\x21-\\x5a\\x53-\\x7f]|\\\\[\\x01-\\x09\\x0b\\x0c\\x0e-\\x7f])+)\\])$", 0)
|
remail = regexp2.MustCompile("^(?=.{0,255}$)(?:[a-z0-9!#$%&'*+/=?^_`{|}~-]+(?:\\.[a-z0-9!#$%&'*+/=?^_`{|}~-]+)*|\"(?:[\\x01-\\x08\\x0b\\x0c\\x0e-\\x1f\\x21\\x23-\\x5b\\x5d-\\x7f]|\\\\[\\x01-\\x09\\x0b\\x0c\\x0e-\\x7f])*\")@(?:(?:[a-z0-9](?:[a-z0-9-]*[a-z0-9])?\\.)+[a-z0-9](?:[a-z0-9-]*[a-z0-9])?|\\[(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?|[a-z0-9-]*[a-z0-9]:(?:[\\x01-\\x08\\x0b\\x0c\\x0e-\\x1f\\x21-\\x5a\\x53-\\x7f]|\\\\[\\x01-\\x09\\x0b\\x0c\\x0e-\\x7f])+)\\])$", 0)
|
||||||
rusername = regexp.MustCompile("^([[:lower:]]|\\d|_|-|\\.){1,40}$")
|
rusername = regexp.MustCompile("^([[:lower:]]|\\d|_|-|\\.){1,40}$")
|
||||||
rpassword = regexp2.MustCompile("^(?=.{8,255}$)(?=.*[a-z])(?=.*[A-Z])(?=.*[0-9])(?=.*\\W).*$", 0)
|
rpassword = regexp2.MustCompile("^(?=.{8,255}$)(?=.*[a-z])(?=.*[A-Z])(?=.*[0-9])(?=.*\\W).*$", 0)
|
||||||
http.HandleFunc("/register", register)
|
http.HandleFunc("/register", register)
|
||||||
http.HandleFunc("/submit", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
var submitStruct SubmitStruct
|
|
||||||
token := r.FormValue("token")
|
|
||||||
var accInter interface{}
|
|
||||||
accInter, submitStruct.Success = cacheAccounts.GetStringKey(token)
|
|
||||||
if !submitStruct.Success {
|
|
||||||
goto submitReturn
|
|
||||||
}
|
|
||||||
{
|
|
||||||
var account account = accInter.(account)
|
|
||||||
cacheAccounts.Del(token)
|
|
||||||
salt := make([]byte, 32)
|
|
||||||
_, err := rand.Read(salt)
|
|
||||||
log(err)
|
|
||||||
hash := argon2.IDKey([]byte(account.password), salt, 1, 64*1024, 4, 32)
|
|
||||||
// add user to the database
|
|
||||||
query := "INSERT INTO account(username, email, hash, salt, discordUserId) VALUES (?, ?, ?, ?, ?)"
|
|
||||||
ctx, cancelfunc := context.WithTimeout(context.Background(), 5*time.Second)
|
|
||||||
defer cancelfunc()
|
|
||||||
stmt, err := db.PrepareContext(ctx, query)
|
|
||||||
log(err)
|
|
||||||
defer stmt.Close()
|
|
||||||
_, err = stmt.ExecContext(ctx, account.username, account.email, hash, salt, account.discordId)
|
|
||||||
log(err)
|
|
||||||
//_, err = moodle.AddUser(account.username + "wg", account.username, account.email, account.username, account.password)
|
|
||||||
log(err)
|
|
||||||
if config.CreateGiteaAccount {
|
|
||||||
opt := gitea.CreateUserOption{
|
|
||||||
Email: account.email,
|
|
||||||
Username: account.username,
|
|
||||||
SourceID: 0,
|
|
||||||
Password: account.password,
|
|
||||||
SendNotify: false,
|
|
||||||
}
|
|
||||||
_, _, err = giteaClient.AdminCreateUser(opt)
|
|
||||||
log(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
submitReturn: err = submitTmpl.Execute(w, submitStruct)
|
http.ListenAndServe(":" + fmt.Sprint(config.Port), nil)
|
||||||
log(err)
|
|
||||||
})
|
|
||||||
|
|
||||||
http.ListenAndServe(":8080", nil)
|
|
||||||
}
|
|
||||||
func getRbuMember(user string, tag string) (*discordgo.Member, bool) {
|
|
||||||
allUsers, err := discord.GuildMembers(secret.DiscordServerID, "0", 1000)
|
|
||||||
log(err)
|
|
||||||
for _, element := range allUsers {
|
|
||||||
if element.User.Username == user && element.User.Discriminator == tag{
|
|
||||||
return element, true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil, false
|
|
||||||
}
|
}
|
||||||
func log(err error) {
|
func log(err error) {
|
||||||
if err!=nil {
|
if err!=nil {
|
||||||
@ -165,26 +113,6 @@ func log(err error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func UsernameExistsInMem(username string) bool {
|
func hash(password []byte, salt []byte) []byte {
|
||||||
for key := range cacheAccounts.Iter() {
|
return argon2.IDKey(password, salt, 1, 64*1024, 4, 32)
|
||||||
var accInter interface{}
|
|
||||||
accInter, _ = cacheAccounts.Get(key)
|
|
||||||
var account account = accInter.(account)
|
|
||||||
if account.username == username {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func discordUsernameExistsInMem(id string) bool {
|
|
||||||
for key := range cacheAccounts.Iter() {
|
|
||||||
var accInter interface{}
|
|
||||||
accInter, _ = cacheAccounts.Get(key)
|
|
||||||
var account account = accInter.(account)
|
|
||||||
if account.discordId == id {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
}
|
||||||
|
|||||||
@ -6,6 +6,10 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
"regexp"
|
"regexp"
|
||||||
"github.com/dlclark/regexp2"
|
"github.com/dlclark/regexp2"
|
||||||
|
"context"
|
||||||
|
"time"
|
||||||
|
"code.gitea.io/sdk/gitea"
|
||||||
|
"crypto/rand"
|
||||||
)
|
)
|
||||||
var rusername *regexp.Regexp
|
var rusername *regexp.Regexp
|
||||||
var remail *regexp2.Regexp
|
var remail *regexp2.Regexp
|
||||||
@ -56,4 +60,80 @@ func register(w http.ResponseWriter, r *http.Request) {
|
|||||||
}
|
}
|
||||||
registerReturn: err = registerTmpl.Execute(w, registerstruct)
|
registerReturn: err = registerTmpl.Execute(w, registerstruct)
|
||||||
log(err)
|
log(err)
|
||||||
|
}
|
||||||
|
func submit(w http.ResponseWriter, r *http.Request) {
|
||||||
|
var err error
|
||||||
|
var submitStruct SubmitStruct
|
||||||
|
token := r.FormValue("token")
|
||||||
|
var accInter interface{}
|
||||||
|
accInter, submitStruct.Success = cacheAccounts.GetStringKey(token)
|
||||||
|
if !submitStruct.Success {
|
||||||
|
goto submitReturn
|
||||||
|
}
|
||||||
|
{
|
||||||
|
var account account = accInter.(account)
|
||||||
|
cacheAccounts.Del(token)
|
||||||
|
salt := make([]byte, 32)
|
||||||
|
_, err := rand.Read(salt)
|
||||||
|
log(err)
|
||||||
|
hash := hash([]byte(account.password), salt)
|
||||||
|
// add user to the database
|
||||||
|
query := "INSERT INTO account(username, email, hash, salt, discordUserId) VALUES (?, ?, ?, ?, ?)"
|
||||||
|
ctx, cancelfunc := context.WithTimeout(context.Background(), 5*time.Second)
|
||||||
|
defer cancelfunc()
|
||||||
|
stmt, err := db.PrepareContext(ctx, query)
|
||||||
|
log(err)
|
||||||
|
defer stmt.Close()
|
||||||
|
_, err = stmt.ExecContext(ctx, account.username, account.email, hash, salt, account.discordId)
|
||||||
|
log(err)
|
||||||
|
//_, err = moodle.AddUser(account.username + "wg", account.username, account.email, account.username, account.password)
|
||||||
|
log(err)
|
||||||
|
if config.CreateGiteaAccount {
|
||||||
|
opt := gitea.CreateUserOption{
|
||||||
|
Email: account.email,
|
||||||
|
Username: account.username,
|
||||||
|
SourceID: 0,
|
||||||
|
Password: account.password,
|
||||||
|
SendNotify: false,
|
||||||
|
}
|
||||||
|
_, _, err = giteaClient.AdminCreateUser(opt)
|
||||||
|
log(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
submitReturn: err = submitTmpl.Execute(w, submitStruct)
|
||||||
|
log(err)
|
||||||
|
}
|
||||||
|
func getRbuMember(user string, tag string) (*discordgo.Member, bool) {
|
||||||
|
allUsers, err := discord.GuildMembers(secret.DiscordServerID, "0", 1000)
|
||||||
|
log(err)
|
||||||
|
for _, element := range allUsers {
|
||||||
|
if element.User.Username == user && element.User.Discriminator == tag{
|
||||||
|
return element, true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
func UsernameExistsInMem(username string) bool {
|
||||||
|
for key := range cacheAccounts.Iter() {
|
||||||
|
var accInter interface{}
|
||||||
|
accInter, _ = cacheAccounts.Get(key)
|
||||||
|
var account account = accInter.(account)
|
||||||
|
if account.username == username {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func discordUsernameExistsInMem(id string) bool {
|
||||||
|
for key := range cacheAccounts.Iter() {
|
||||||
|
var accInter interface{}
|
||||||
|
accInter, _ = cacheAccounts.Get(key)
|
||||||
|
var account account = accInter.(account)
|
||||||
|
if account.discordId == id {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user