66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
var accountsSchema = `create table accounts
|
|
(
|
|
id serial
|
|
constraint accounts_pkey
|
|
primary key,
|
|
username varchar(13) not null,
|
|
password varchar not null,
|
|
creation date not null,
|
|
last_login date not null,
|
|
last_ip varchar not null,
|
|
ban integer not null,
|
|
admin integer not null,
|
|
gender smallint default 0 not null
|
|
);`
|
|
|
|
var authlog *SecurityLog
|
|
|
|
func install(db *sqlx.DB) {
|
|
if _, err := db.Exec(accountsSchema); err != nil {
|
|
panic(err)
|
|
}
|
|
println("accounts table installed!")
|
|
}
|
|
|
|
func argParse() {
|
|
var kayos *User
|
|
var err error
|
|
for _, arg := range os.Args {
|
|
if arg == "--install" {
|
|
println("installing...")
|
|
install(db)
|
|
return
|
|
}
|
|
if arg == "--regkayos" {
|
|
println("registering kayos...")
|
|
if err := RegisterNewUser("kayos", "yeet", "127.0.0.1", 0, true); err != nil {
|
|
println("REG_TEST_FAIL: " + err.Error())
|
|
return
|
|
}
|
|
println("kayos registered!")
|
|
return
|
|
}
|
|
if arg == "--loginkayos" {
|
|
println("logging in as kayos...")
|
|
if kayos, err = AttemptWebLogin(
|
|
"kayos",
|
|
"yeet",
|
|
"192.168.69.5",
|
|
"yeetBrowser 420"); err != nil {
|
|
println("FAIL: " + err.Error())
|
|
return
|
|
}
|
|
println("logged in as kayos!")
|
|
println(kayos.Password)
|
|
}
|
|
}
|
|
}
|