package main import ( "net/http" "encoding/json" "time" ) func domainExportBeginError(email string, toName string, domain string, err error) { // we're not using err at the moment because it's all errorInternal if err2 := smtpDomainExportError(email, toName, domain); err2 != nil { logger.Errorf("cannot send domain export error email for %s: %v", domain, err2) return } } func domainExportBegin(email string, toName string, domain string) { type dataExport struct { Version int `json:"version"` Comments []comment `json:"comments"` Commenters []commenter `json:"commenters"` } e := dataExport{Version: 1, Comments: []comment{}, Commenters: []commenter{}} statement := ` SELECT commentHex, domain, path, commenterHex, markdown, parentHex, score, state, creationDate FROM comments WHERE domain = $1; ` rows1, err := db.Query(statement, domain) if err != nil { logger.Errorf("cannot select comments while exporting %s: %v", domain, err) domainExportBeginError(email, toName, domain, errorInternal) return } defer rows1.Close() for rows1.Next() { c := comment{} if err = rows1.Scan(&c.CommentHex, &c.Domain, &c.Path, &c.CommenterHex, &c.Markdown, &c.ParentHex, &c.Score, &c.State, &c.CreationDate); err != nil { logger.Errorf("cannot scan comment while exporting %s: %v", domain, err) domainExportBeginError(email, toName, domain, errorInternal) return } e.Comments = append(e.Comments, c) } statement = ` SELECT commenters.commenterHex, commenters.email, commenters.name, commenters.link, commenters.photo, commenters.provider, commenters.joinDate FROM commenters, comments WHERE comments.domain = $1 AND commenters.commenterHex = comments.commenterHex; ` rows2, err := db.Query(statement, domain) if err != nil { logger.Errorf("cannot select commenters while exporting %s: %v", domain, err) domainExportBeginError(email, toName, domain, errorInternal) return } defer rows2.Close() for rows2.Next() { c := commenter{} if err := rows2.Scan(&c.CommenterHex, &c.Email, &c.Name, &c.Link, &c.Photo, &c.Provider, &c.JoinDate); err != nil { logger.Errorf("cannot scan commenter while exporting %s: %v", domain, err) domainExportBeginError(email, toName, domain, errorInternal) return } e.Commenters = append(e.Commenters, c) } je, err := json.Marshal(e) if err != nil { logger.Errorf("cannot marshall JSON while exporting %s: %v", domain, err) domainExportBeginError(email, toName, domain, errorInternal) return } gje, err := gzipStatic(je) if err != nil { logger.Errorf("cannot gzip JSON while exporting %s: %v", domain, err) domainExportBeginError(email, toName, domain, errorInternal) return } exportHex, err := randomHex(32) if err != nil { logger.Errorf("cannot generate exportHex while exporting %s: %v", domain, err) domainExportBeginError(email, toName, domain, errorInternal) return } statement = ` INSERT INTO exports (exportHex, binData, domain, creationDate) VALUES ($1, $2, $3 , $4 ); ` _, err = db.Exec(statement, exportHex, gje, domain, time.Now().UTC()) if err != nil { logger.Errorf("error inserting expiry binary data while exporting %s: %v", domain, err) domainExportBeginError(email, toName, domain, errorInternal) return } err = smtpDomainExport(email, toName, domain, exportHex) if err != nil { logger.Errorf("error sending data export email for %s: %v", domain, err) return } } func domainExportBeginHandler(w http.ResponseWriter, r *http.Request) { type request struct { OwnerToken *string `json:"ownerToken"` Domain *string `json:"domain"` } var x request if err := bodyUnmarshal(r, &x); err != nil { bodyMarshal(w, response{"success": false, "message": err.Error()}) return } if !smtpConfigured { bodyMarshal(w, response{"success": false, "message": errorSmtpNotConfigured.Error()}) return } o, err := ownerGetByOwnerToken(*x.OwnerToken) if err != nil { bodyMarshal(w, response{"success": false, "message": err.Error()}) return } isOwner, err := domainOwnershipVerify(o.OwnerHex, *x.Domain) if err != nil { bodyMarshal(w, response{"success": false, "message": err.Error()}) return } if !isOwner { bodyMarshal(w, response{"success": false, "message": errorNotAuthorised.Error()}) return } go domainExportBegin(o.Email, o.Name, *x.Domain); bodyMarshal(w, response{"success": true}) }