95 lines
1.8 KiB
Go
95 lines
1.8 KiB
Go
|
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
|
||
|
|
}
|