forked from tildetown/town
dump script for user db
parent
f624483614
commit
c5590cac95
|
@ -9,11 +9,14 @@ import (
|
|||
|
||||
"git.tilde.town/tildetown/town/models"
|
||||
"git.tilde.town/tildetown/town/signup"
|
||||
"git.tilde.town/tildetown/town/stats"
|
||||
"git.tilde.town/tildetown/town/towndb"
|
||||
"github.com/AlecAivazis/survey/v2"
|
||||
)
|
||||
|
||||
// TODO ok so right now this just does one hyper specific thing, but I want to
|
||||
// add more similar scripts here. this might warrant the introduction of cobra.
|
||||
// this is basically a pile of scripts. no hope is to be found here. this is
|
||||
// dirty, one off code stored in case any useful patterns are worth extracting
|
||||
// or for copypasta fodder.
|
||||
|
||||
func confirmContinue(msg string) {
|
||||
var serr error
|
||||
|
@ -41,6 +44,108 @@ type jsonSignup struct {
|
|||
}
|
||||
|
||||
func main() {
|
||||
db, err := towndb.ConnectDB()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
lol, err := os.ReadFile("/town/var/users.json")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
td, err := stats.Stats()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
lines := strings.Split(string(lol), "\n")
|
||||
errs := []error{}
|
||||
signups := make([]jsonSignup, len(lines))
|
||||
for i, l := range lines {
|
||||
l = strings.TrimSpace(l)
|
||||
if l == "" {
|
||||
continue
|
||||
}
|
||||
s := jsonSignup{}
|
||||
err := json.Unmarshal([]byte(l), &s)
|
||||
if err != nil {
|
||||
fmt.Printf("%s %s", l, err.Error())
|
||||
errs = append(errs, err)
|
||||
} else {
|
||||
signups[i] = s
|
||||
}
|
||||
}
|
||||
if len(errs) > 0 {
|
||||
confirmContinue(fmt.Sprintf("%d errors found deserializing; continue?", len(errs)))
|
||||
}
|
||||
|
||||
ttbirth, err := time.Parse("2006-01-02 3:04pm", "2014-10-11 11:49pm")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
me := towndb.TownUser{
|
||||
Created: ttbirth,
|
||||
Username: "vilmibm",
|
||||
Emails: []string{"vilmibm@protonmail.com"},
|
||||
State: towndb.StateActive,
|
||||
IsAdmin: true,
|
||||
}
|
||||
err = me.Insert(db)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
notFound := []jsonSignup{}
|
||||
found := []jsonSignup{}
|
||||
for _, su := range signups {
|
||||
fl := len(found)
|
||||
for _, u := range td.Users {
|
||||
if su.Username == u.Username {
|
||||
found = append(found, su)
|
||||
break
|
||||
}
|
||||
}
|
||||
if len(found) == fl {
|
||||
notFound = append(notFound, su)
|
||||
}
|
||||
}
|
||||
if len(notFound) > 0 {
|
||||
confirmContinue(fmt.Sprintf("%d of those were not found. continue?", len(notFound)))
|
||||
}
|
||||
|
||||
for _, su := range found {
|
||||
var emails []string
|
||||
if su.Email != "" {
|
||||
emails = []string{su.Email}
|
||||
}
|
||||
|
||||
u := towndb.TownUser{
|
||||
Created: time.Unix(int64(su.Created), 0),
|
||||
Emails: emails,
|
||||
Username: su.Username,
|
||||
State: towndb.StateActive,
|
||||
}
|
||||
if err = u.Insert(db); err != nil {
|
||||
confirmContinue(fmt.Sprintf("%#v led to error %s; continue?", u, err.Error()))
|
||||
}
|
||||
|
||||
if su.Notes != "" {
|
||||
note := towndb.AdminNote{
|
||||
Created: time.Time{},
|
||||
AuthorID: me.ID,
|
||||
Content: su.Notes,
|
||||
UserID: u.ID,
|
||||
}
|
||||
if err = note.Insert(db); err != nil {
|
||||
confirmContinue(fmt.Sprintf("%#v led to error %s; continue?", note, err.Error()))
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func importSignups() {
|
||||
lol, err := os.ReadFile("/town/var/signups.json")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
|
|
|
@ -27,34 +27,57 @@ type AdminNote struct {
|
|||
}
|
||||
|
||||
func (n *AdminNote) Insert(db *sql.DB) error {
|
||||
n.Created = time.Now()
|
||||
stmt, err := db.Prepare(`
|
||||
INSERT INTO notes (created, authorid, content, userid)
|
||||
VALUES (
|
||||
?, ?, ?, ?
|
||||
)`)
|
||||
var (
|
||||
err error
|
||||
stmt *sql.Stmt
|
||||
result sql.Result
|
||||
liid int64
|
||||
)
|
||||
|
||||
tx, err := db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
result, err := stmt.Exec(
|
||||
defer func() {
|
||||
if err != nil {
|
||||
tx.Rollback()
|
||||
}
|
||||
}()
|
||||
stmt, err = tx.Prepare(`
|
||||
INSERT INTO notes (created, author, content)
|
||||
VALUES (?, ?, ?)`)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
result, err = stmt.Exec(
|
||||
n.Created.Unix(),
|
||||
n.AuthorID,
|
||||
n.Content,
|
||||
n.UserID)
|
||||
n.Content)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer stmt.Close()
|
||||
|
||||
liid, err := result.LastInsertId()
|
||||
liid, err = result.LastInsertId()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
n.ID = liid
|
||||
|
||||
return nil
|
||||
stmt, err = tx.Prepare(`
|
||||
INSERT INTO user_notes (noteid, userid) VALUES (?, ?)`)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = stmt.Exec(n.ID, n.UserID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return tx.Commit()
|
||||
}
|
||||
|
||||
type TownUser struct {
|
||||
|
|
Loading…
Reference in New Issue