mirror of https://github.com/writeas/writefreely
Matt Baer
6 years ago
parent
af872127c6
commit
ebeacff43c
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,114 @@ |
|||||||
|
package writefreely |
||||||
|
|
||||||
|
import ( |
||||||
|
"archive/zip" |
||||||
|
"bytes" |
||||||
|
"encoding/csv" |
||||||
|
"github.com/writeas/web-core/log" |
||||||
|
"strings" |
||||||
|
"time" |
||||||
|
) |
||||||
|
|
||||||
|
func exportPostsCSV(u *User, posts *[]PublicPost) []byte { |
||||||
|
var b bytes.Buffer |
||||||
|
|
||||||
|
r := [][]string{ |
||||||
|
{"id", "slug", "blog", "url", "created", "title", "body"}, |
||||||
|
} |
||||||
|
for _, p := range *posts { |
||||||
|
var blog string |
||||||
|
if p.Collection != nil { |
||||||
|
blog = p.Collection.Alias |
||||||
|
} |
||||||
|
f := []string{p.ID, p.Slug.String, blog, p.CanonicalURL(), p.Created8601(), p.Title.String, strings.Replace(p.Content, "\n", "\\n", -1)} |
||||||
|
r = append(r, f) |
||||||
|
} |
||||||
|
|
||||||
|
w := csv.NewWriter(&b) |
||||||
|
w.WriteAll(r) // calls Flush internally
|
||||||
|
if err := w.Error(); err != nil { |
||||||
|
log.Info("error writing csv:", err) |
||||||
|
} |
||||||
|
|
||||||
|
return b.Bytes() |
||||||
|
} |
||||||
|
|
||||||
|
type exportedTxt struct { |
||||||
|
Name, Body string |
||||||
|
Mod time.Time |
||||||
|
} |
||||||
|
|
||||||
|
func exportPostsZip(u *User, posts *[]PublicPost) []byte { |
||||||
|
// Create a buffer to write our archive to.
|
||||||
|
b := new(bytes.Buffer) |
||||||
|
|
||||||
|
// Create a new zip archive.
|
||||||
|
w := zip.NewWriter(b) |
||||||
|
|
||||||
|
// Add some files to the archive.
|
||||||
|
var filename string |
||||||
|
files := []exportedTxt{} |
||||||
|
for _, p := range *posts { |
||||||
|
filename = "" |
||||||
|
if p.Collection != nil { |
||||||
|
filename += p.Collection.Alias + "/" |
||||||
|
} |
||||||
|
if p.Slug.String != "" { |
||||||
|
filename += p.Slug.String + "_" |
||||||
|
} |
||||||
|
filename += p.ID + ".txt" |
||||||
|
files = append(files, exportedTxt{filename, p.Content, p.Created}) |
||||||
|
} |
||||||
|
|
||||||
|
for _, file := range files { |
||||||
|
head := &zip.FileHeader{Name: file.Name} |
||||||
|
head.SetModTime(file.Mod) |
||||||
|
f, err := w.CreateHeader(head) |
||||||
|
if err != nil { |
||||||
|
log.Error("export zip header: %v", err) |
||||||
|
} |
||||||
|
_, err = f.Write([]byte(file.Body)) |
||||||
|
if err != nil { |
||||||
|
log.Error("export zip write: %v", err) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Make sure to check the error on Close.
|
||||||
|
err := w.Close() |
||||||
|
if err != nil { |
||||||
|
log.Error("export zip close: %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
return b.Bytes() |
||||||
|
} |
||||||
|
|
||||||
|
func compileFullExport(app *app, u *User) *ExportUser { |
||||||
|
exportUser := &ExportUser{ |
||||||
|
User: u, |
||||||
|
} |
||||||
|
|
||||||
|
colls, err := app.db.GetCollections(u) |
||||||
|
if err != nil { |
||||||
|
log.Error("unable to fetch collections: %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
posts, err := app.db.GetAnonymousPosts(u) |
||||||
|
if err != nil { |
||||||
|
log.Error("unable to fetch anon posts: %v", err) |
||||||
|
} |
||||||
|
exportUser.AnonymousPosts = *posts |
||||||
|
|
||||||
|
var collObjs []CollectionObj |
||||||
|
for _, c := range *colls { |
||||||
|
co := &CollectionObj{Collection: c} |
||||||
|
co.Posts, err = app.db.GetPosts(&c, 0, true) |
||||||
|
if err != nil { |
||||||
|
log.Error("unable to get collection posts: %v", err) |
||||||
|
} |
||||||
|
app.db.GetPostsCount(co, true) |
||||||
|
collObjs = append(collObjs, *co) |
||||||
|
} |
||||||
|
exportUser.Collections = &collObjs |
||||||
|
|
||||||
|
return exportUser |
||||||
|
} |
||||||
@ -0,0 +1,100 @@ |
|||||||
|
package writefreely |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
. "github.com/gorilla/feeds" |
||||||
|
"github.com/gorilla/mux" |
||||||
|
stripmd "github.com/writeas/go-strip-markdown" |
||||||
|
"github.com/writeas/web-core/log" |
||||||
|
"net/http" |
||||||
|
"time" |
||||||
|
) |
||||||
|
|
||||||
|
func ViewFeed(app *app, w http.ResponseWriter, req *http.Request) error { |
||||||
|
alias := collectionAliasFromReq(req) |
||||||
|
|
||||||
|
// Display collection if this is a collection
|
||||||
|
var c *Collection |
||||||
|
var err error |
||||||
|
if app.cfg.App.SingleUser { |
||||||
|
c, err = app.db.GetCollection(alias) |
||||||
|
} else { |
||||||
|
c, err = app.db.GetCollectionByID(1) |
||||||
|
} |
||||||
|
if err != nil { |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
if c.IsPrivate() || c.IsProtected() { |
||||||
|
return ErrCollectionNotFound |
||||||
|
} |
||||||
|
|
||||||
|
// Fetch extra data about the Collection
|
||||||
|
// TODO: refactor out this logic, shared in collection.go:fetchCollection()
|
||||||
|
coll := &DisplayCollection{CollectionObj: &CollectionObj{Collection: *c}} |
||||||
|
if c.PublicOwner { |
||||||
|
u, err := app.db.GetUserByID(coll.OwnerID) |
||||||
|
if err != nil { |
||||||
|
// Log the error and just continue
|
||||||
|
log.Error("Error getting user for collection: %v", err) |
||||||
|
} else { |
||||||
|
coll.Owner = u |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
tag := mux.Vars(req)["tag"] |
||||||
|
if tag != "" { |
||||||
|
coll.Posts, _ = app.db.GetPostsTagged(c, tag, 1, false) |
||||||
|
} else { |
||||||
|
coll.Posts, _ = app.db.GetPosts(c, 1, false) |
||||||
|
} |
||||||
|
|
||||||
|
author := "" |
||||||
|
if coll.Owner != nil { |
||||||
|
author = coll.Owner.Username |
||||||
|
} |
||||||
|
|
||||||
|
collectionTitle := coll.DisplayTitle() |
||||||
|
if tag != "" { |
||||||
|
collectionTitle = tag + " — " + collectionTitle |
||||||
|
} |
||||||
|
|
||||||
|
baseUrl := coll.CanonicalURL() |
||||||
|
basePermalinkUrl := baseUrl |
||||||
|
siteURL := baseUrl |
||||||
|
if tag != "" { |
||||||
|
siteURL += "tag:" + tag |
||||||
|
} |
||||||
|
|
||||||
|
feed := &Feed{ |
||||||
|
Title: collectionTitle, |
||||||
|
Link: &Link{Href: siteURL}, |
||||||
|
Description: coll.Description, |
||||||
|
Author: &Author{author, ""}, |
||||||
|
Created: time.Now(), |
||||||
|
} |
||||||
|
|
||||||
|
var title, permalink string |
||||||
|
for _, p := range *coll.Posts { |
||||||
|
title = p.PlainDisplayTitle() |
||||||
|
permalink = fmt.Sprintf("%s%s", baseUrl, p.Slug.String) |
||||||
|
feed.Items = append(feed.Items, &Item{ |
||||||
|
Id: fmt.Sprintf("%s%s", basePermalinkUrl, p.Slug.String), |
||||||
|
Title: title, |
||||||
|
Link: &Link{Href: permalink}, |
||||||
|
Description: "<![CDATA[" + stripmd.Strip(p.Content) + "]]>", |
||||||
|
Content: applyMarkdown([]byte(p.Content)), |
||||||
|
Author: &Author{author, ""}, |
||||||
|
Created: p.Created, |
||||||
|
Updated: p.Updated, |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
rss, err := feed.ToRss() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
fmt.Fprint(w, rss) |
||||||
|
return nil |
||||||
|
} |
||||||
@ -0,0 +1,8 @@ |
|||||||
|
package writefreely |
||||||
|
|
||||||
|
import "mime" |
||||||
|
|
||||||
|
func IsJSON(h string) bool { |
||||||
|
ct, _, _ := mime.ParseMediaType(h) |
||||||
|
return ct == "application/json" |
||||||
|
} |
||||||
@ -0,0 +1,94 @@ |
|||||||
|
package writefreely |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"github.com/gorilla/mux" |
||||||
|
"github.com/ikeikeikeike/go-sitemap-generator/stm" |
||||||
|
"github.com/writeas/web-core/log" |
||||||
|
"net/http" |
||||||
|
"time" |
||||||
|
) |
||||||
|
|
||||||
|
func buildSitemap(host, alias string) *stm.Sitemap { |
||||||
|
sm := stm.NewSitemap() |
||||||
|
sm.SetDefaultHost(host) |
||||||
|
if alias != "/" { |
||||||
|
sm.SetSitemapsPath(alias) |
||||||
|
} |
||||||
|
|
||||||
|
sm.Create() |
||||||
|
|
||||||
|
// Note: Do not call `sm.Finalize()` because it flushes
|
||||||
|
// the underlying datastructure from memory to disk.
|
||||||
|
|
||||||
|
return sm |
||||||
|
} |
||||||
|
|
||||||
|
func handleViewSitemap(app *app, w http.ResponseWriter, r *http.Request) error { |
||||||
|
vars := mux.Vars(r) |
||||||
|
|
||||||
|
// Determine canonical blog URL
|
||||||
|
alias := vars["collection"] |
||||||
|
subdomain := vars["subdomain"] |
||||||
|
isSubdomain := subdomain != "" |
||||||
|
if isSubdomain { |
||||||
|
alias = subdomain |
||||||
|
} |
||||||
|
|
||||||
|
host := fmt.Sprintf("%s/%s/", app.cfg.App.Host, alias) |
||||||
|
var c *Collection |
||||||
|
var err error |
||||||
|
pre := "/" |
||||||
|
if app.cfg.App.SingleUser { |
||||||
|
c, err = app.db.GetCollectionByID(1) |
||||||
|
} else { |
||||||
|
c, err = app.db.GetCollection(alias) |
||||||
|
} |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if !isSubdomain { |
||||||
|
pre += alias + "/" |
||||||
|
} |
||||||
|
host = c.CanonicalURL() |
||||||
|
|
||||||
|
sm := buildSitemap(host, pre) |
||||||
|
posts, err := app.db.GetPosts(c, 0, false) |
||||||
|
if err != nil { |
||||||
|
log.Error("Error getting posts: %v", err) |
||||||
|
return err |
||||||
|
} |
||||||
|
lastSiteMod := time.Now() |
||||||
|
for i, p := range *posts { |
||||||
|
if i == 0 { |
||||||
|
lastSiteMod = p.Updated |
||||||
|
} |
||||||
|
u := stm.URL{ |
||||||
|
"loc": p.Slug.String, |
||||||
|
"changefreq": "weekly", |
||||||
|
"mobile": true, |
||||||
|
"lastmod": p.Updated, |
||||||
|
} |
||||||
|
if len(p.Images) > 0 { |
||||||
|
imgs := []stm.URL{} |
||||||
|
for _, i := range p.Images { |
||||||
|
imgs = append(imgs, stm.URL{"loc": i, "title": ""}) |
||||||
|
} |
||||||
|
u["image"] = imgs |
||||||
|
} |
||||||
|
sm.Add(u) |
||||||
|
} |
||||||
|
|
||||||
|
// Add top URL
|
||||||
|
sm.Add(stm.URL{ |
||||||
|
"loc": pre, |
||||||
|
"changefreq": "daily", |
||||||
|
"priority": "1.0", |
||||||
|
"lastmod": lastSiteMod, |
||||||
|
}) |
||||||
|
|
||||||
|
w.Write(sm.XMLContent()) |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
@ -0,0 +1,121 @@ |
|||||||
|
package writefreely |
||||||
|
|
||||||
|
import ( |
||||||
|
"database/sql" |
||||||
|
"encoding/json" |
||||||
|
"github.com/writeas/impart" |
||||||
|
"github.com/writeas/web-core/log" |
||||||
|
"net/http" |
||||||
|
) |
||||||
|
|
||||||
|
func handleWebSignup(app *app, w http.ResponseWriter, r *http.Request) error { |
||||||
|
reqJSON := IsJSON(r.Header.Get("Content-Type")) |
||||||
|
|
||||||
|
// Get params
|
||||||
|
var ur userRegistration |
||||||
|
if reqJSON { |
||||||
|
decoder := json.NewDecoder(r.Body) |
||||||
|
err := decoder.Decode(&ur) |
||||||
|
if err != nil { |
||||||
|
log.Error("Couldn't parse signup JSON request: %v\n", err) |
||||||
|
return ErrBadJSON |
||||||
|
} |
||||||
|
} else { |
||||||
|
err := r.ParseForm() |
||||||
|
if err != nil { |
||||||
|
log.Error("Couldn't parse signup form request: %v\n", err) |
||||||
|
return ErrBadFormData |
||||||
|
} |
||||||
|
|
||||||
|
err = app.formDecoder.Decode(&ur, r.PostForm) |
||||||
|
if err != nil { |
||||||
|
log.Error("Couldn't decode signup form request: %v\n", err) |
||||||
|
return ErrBadFormData |
||||||
|
} |
||||||
|
} |
||||||
|
ur.Web = true |
||||||
|
|
||||||
|
_, err := signupWithRegistration(app, ur, w, r) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
return impart.HTTPError{http.StatusFound, "/"} |
||||||
|
} |
||||||
|
|
||||||
|
// { "username": "asdf" }
|
||||||
|
// result: { code: 204 }
|
||||||
|
func handleUsernameCheck(app *app, w http.ResponseWriter, r *http.Request) error { |
||||||
|
reqJSON := IsJSON(r.Header.Get("Content-Type")) |
||||||
|
|
||||||
|
// Get params
|
||||||
|
var d struct { |
||||||
|
Username string `json:"username"` |
||||||
|
} |
||||||
|
if reqJSON { |
||||||
|
decoder := json.NewDecoder(r.Body) |
||||||
|
err := decoder.Decode(&d) |
||||||
|
if err != nil { |
||||||
|
log.Error("Couldn't decode username check: %v\n", err) |
||||||
|
return ErrBadFormData |
||||||
|
} |
||||||
|
} else { |
||||||
|
return impart.HTTPError{http.StatusNotAcceptable, "Must be JSON request"} |
||||||
|
} |
||||||
|
|
||||||
|
// Check if username is okay
|
||||||
|
finalUsername := getSlug(d.Username, "") |
||||||
|
if finalUsername == "" { |
||||||
|
errMsg := "Invalid username" |
||||||
|
if d.Username != "" { |
||||||
|
// Username was provided, but didn't convert into valid latin characters
|
||||||
|
errMsg += " - must have at least 2 letters or numbers" |
||||||
|
} |
||||||
|
return impart.HTTPError{http.StatusBadRequest, errMsg + "."} |
||||||
|
} |
||||||
|
if app.db.PostIDExists(finalUsername) { |
||||||
|
return impart.HTTPError{http.StatusConflict, "Username is already taken."} |
||||||
|
} |
||||||
|
var un string |
||||||
|
err := app.db.QueryRow("SELECT username FROM users WHERE username = ?", finalUsername).Scan(&un) |
||||||
|
switch { |
||||||
|
case err == sql.ErrNoRows: |
||||||
|
return impart.WriteSuccess(w, finalUsername, http.StatusOK) |
||||||
|
case err != nil: |
||||||
|
log.Error("Couldn't SELECT username: %v", err) |
||||||
|
return impart.HTTPError{http.StatusInternalServerError, "We messed up."} |
||||||
|
} |
||||||
|
|
||||||
|
// Username was found, so it's taken
|
||||||
|
return impart.HTTPError{http.StatusConflict, "Username is already taken."} |
||||||
|
} |
||||||
|
|
||||||
|
func getValidUsername(app *app, reqName, prevName string) (string, *impart.HTTPError) { |
||||||
|
// Check if username is okay
|
||||||
|
finalUsername := getSlug(reqName, "") |
||||||
|
if finalUsername == "" { |
||||||
|
errMsg := "Invalid username" |
||||||
|
if reqName != "" { |
||||||
|
// Username was provided, but didn't convert into valid latin characters
|
||||||
|
errMsg += " - must have at least 2 letters or numbers" |
||||||
|
} |
||||||
|
return "", &impart.HTTPError{http.StatusBadRequest, errMsg + "."} |
||||||
|
} |
||||||
|
if finalUsername == prevName { |
||||||
|
return "", &impart.HTTPError{http.StatusNotModified, "Username unchanged."} |
||||||
|
} |
||||||
|
if app.db.PostIDExists(finalUsername) { |
||||||
|
return "", &impart.HTTPError{http.StatusConflict, "Username is already taken."} |
||||||
|
} |
||||||
|
var un string |
||||||
|
err := app.db.QueryRow("SELECT username FROM users WHERE username = ?", finalUsername).Scan(&un) |
||||||
|
switch { |
||||||
|
case err == sql.ErrNoRows: |
||||||
|
return finalUsername, nil |
||||||
|
case err != nil: |
||||||
|
log.Error("Couldn't SELECT username: %v", err) |
||||||
|
return "", &impart.HTTPError{http.StatusInternalServerError, "We messed up."} |
||||||
|
} |
||||||
|
|
||||||
|
// Username was found, so it's taken
|
||||||
|
return "", &impart.HTTPError{http.StatusConflict, "Username is already taken."} |
||||||
|
} |
||||||
Loading…
Reference in new issue