2 SGBlog -- Git-backed CGI/UCSPI blogging/phlogging/gemlogging engine
3 Copyright (C) 2020-2021 Sergey Matveev <stargrave@stargrave.org>
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU Affero General Public License as
7 published by the Free Software Foundation, version 3 of the License.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU Affero General Public License
15 along with this program. If not, see <http://www.gnu.org/licenses/>.
41 "github.com/go-git/go-git/v5"
42 "github.com/go-git/go-git/v5/plumbing"
43 "github.com/go-git/go-git/v5/plumbing/object"
44 "go.stargrave.org/sgblog"
45 "go.stargrave.org/sgblog/cmd/sgblog/atom"
46 "golang.org/x/crypto/blake2b"
50 AtomPostsFeed = "feed.atom"
51 AtomCommentsFeed = "comments.atom"
55 renderableSchemes = map[string]struct{}{
68 //go:embed http-index.tmpl
69 TmplHTMLIndexRaw string
70 TmplHTMLIndex = template.Must(template.New("http-index").Parse(TmplHTMLIndexRaw))
72 //go:embed http-entry.tmpl
73 TmplHTMLEntryRaw string
74 TmplHTMLEntry = template.Must(template.New("http-entry").Funcs(
75 template.FuncMap{"lineURLize": lineURLizeInTemplate},
76 ).Parse(TmplHTMLEntryRaw))
79 type TableEntry struct {
91 type CommentEntry struct {
96 func makeA(href, text string) string {
97 return `<a href="` + href + `">` + text + `</a>`
100 func etagString(etag hash.Hash) string {
101 return `"` + hex.EncodeToString(etag.Sum(nil)) + `"`
104 func urlParse(what string) *url.URL {
105 if u, err := url.ParseRequestURI(what); err == nil {
106 if _, exists := renderableSchemes[u.Scheme]; exists {
113 func lineURLize(urlPrefix, line string) string {
114 cols := strings.Split(html.EscapeString(line), " ")
115 for i, col := range cols {
116 if u := urlParse(col); u != nil {
117 cols[i] = makeA(col, col)
120 cols[i] = sha1DigestRe.ReplaceAllString(col, makeA(urlPrefix+"/$1", "$1"))
122 return strings.Join(cols, " ")
125 func lineURLizeInTemplate(urlPrefix, line interface{}) string {
126 return lineURLize(urlPrefix.(string), line.(string))
129 func startHeader(etag hash.Hash, gziped bool) string {
131 "Content-Type: text/html; charset=utf-8",
132 "ETag: " + etagString(etag),
135 lines = append(lines, "Content-Encoding: gzip")
137 lines = append(lines, "")
138 lines = append(lines, "")
139 return strings.Join(lines, "\n")
142 func makeErr(err error, status int) {
143 fmt.Println("Status:", status)
144 fmt.Print("Content-Type: text/plain; charset=utf-8\n\n")
149 func checkETag(etag hash.Hash) {
150 ifNoneMatch := os.Getenv("HTTP_IF_NONE_MATCH")
151 if ifNoneMatch != "" && ifNoneMatch == etagString(etag) {
152 fmt.Printf("Status: 304\nETag: %s\n\n", ifNoneMatch)
157 func bytes2uuid(b []byte) string {
160 raw[6] = (raw[6] & 0x0F) | uint8(4<<4) // version 4
161 return fmt.Sprintf("%x-%x-%x-%x-%x", raw[0:4], raw[4:6], raw[6:8], raw[8:10], raw[10:])
164 type CommitIterNext interface {
165 Next() (*object.Commit, error)
169 cfgPath := os.Getenv("SGBLOG_CFG")
171 log.Fatalln("SGBLOG_CFG is not set")
173 cfg, err := readCfg(cfgPath)
178 pathInfo := os.Getenv("PATH_INFO")
179 if len(pathInfo) == 0 {
182 queryValues, err := url.ParseQuery(os.Getenv("QUERY_STRING"))
184 makeErr(err, http.StatusBadRequest)
187 etagHash, err := blake2b.New256(nil)
191 for _, s := range []string{
202 if _, err = etagHash.Write([]byte(s)); err != nil {
206 etagHashForWeb := []string{
210 cfg.CommentsNotesRef,
213 for _, gitURL := range cfg.GitURLs {
214 etagHashForWeb = append(etagHashForWeb, gitURL)
217 headHash, err := initRepo(cfg)
219 makeErr(err, http.StatusInternalServerError)
222 if notes, err := repo.Notes(); err == nil {
223 var notesRef *plumbing.Reference
224 var commentsRef *plumbing.Reference
225 notes.ForEach(func(ref *plumbing.Reference) error {
226 switch string(ref.Name()) {
227 case "refs/notes/commits":
229 case cfg.CommentsNotesRef:
235 if commentsCommit, err := repo.CommitObject(notesRef.Hash()); err == nil {
236 notesTree, _ = commentsCommit.Tree()
239 if commentsRef != nil {
240 if commentsCommit, err := repo.CommitObject(commentsRef.Hash()); err == nil {
241 commentsTree, _ = commentsCommit.Tree()
246 var outBuf bytes.Buffer
249 var gzipWriter *gzip.Writer
250 acceptEncoding := os.Getenv("HTTP_ACCEPT_ENCODING")
251 for _, encoding := range strings.Split(acceptEncoding, ", ") {
252 if encoding == "gzip" {
253 gzipWriter = gzip.NewWriter(&outBuf)
260 if offsetRaw, exists := queryValues["offset"]; exists {
261 offset, err = strconv.Atoi(offsetRaw[0])
263 makeErr(err, http.StatusBadRequest)
266 repoLog, err := repo.Log(&git.LogOptions{From: *headHash})
268 makeErr(err, http.StatusInternalServerError)
270 topicsCache, err := getTopicsCache(cfg, repoLog)
272 makeErr(err, http.StatusInternalServerError)
274 repoLog, err = repo.Log(&git.LogOptions{From: *headHash})
276 makeErr(err, http.StatusInternalServerError)
280 var commits CommitIterNext
282 if t, exists := queryValues["topic"]; exists {
284 hashes := topicsCache[topic]
286 makeErr(errors.New("no posts with that topic"), http.StatusBadRequest)
288 if len(hashes) > offset {
289 hashes = hashes[offset:]
292 commits = &HashesIter{hashes}
294 for i := 0; i < offset; i++ {
295 if _, err = repoLog.Next(); err != nil {
303 entries := make([]TableEntry, 0, PageEntries)
305 for _, data := range etagHashForWeb {
306 etagHash.Write([]byte(data))
308 etagHash.Write([]byte("INDEX"))
309 etagHash.Write([]byte(topic))
310 for i := 0; i < PageEntries; i++ {
311 commit, err := commits.Next()
316 etagHash.Write(commit.Hash[:])
317 commentsRaw := sgblog.GetNote(repo, commentsTree, commit.Hash)
318 etagHash.Write(commentsRaw)
319 topicsRaw := sgblog.GetNote(repo, topicsTree, commit.Hash)
320 etagHash.Write(topicsRaw)
321 entries = append(entries, TableEntry{
323 CommentsRaw: commentsRaw,
324 TopicsRaw: topicsRaw,
329 for i, entry := range entries {
332 lines := msgSplit(entry.Commit.Message)
333 entry.Title = lines[0]
334 entry.LinesNum = len(lines) - 2
335 for _, line := range lines[2:] {
340 entry.DomainURLs = append(entry.DomainURLs, makeA(line, u.Host))
342 entry.CommentsNum = len(sgblog.ParseComments(entry.CommentsRaw))
343 entry.Topics = sgblog.ParseTopics(entry.TopicsRaw)
346 offsetPrev := offset - PageEntries
350 os.Stdout.Write([]byte(startHeader(etagHash, gzipWriter != nil)))
351 err = TmplHTMLIndex.Execute(out, struct {
359 AtomCommentsFeed string
366 Version: sgblog.Version,
369 TopicsEnabled: topicsTree != nil,
370 Topics: topicsCache.Topics(),
371 CommentsEnabled: commentsTree != nil,
372 AtomPostsFeed: AtomPostsFeed,
373 AtomCommentsFeed: AtomCommentsFeed,
375 OffsetPrev: offsetPrev,
376 OffsetNext: offset + PageEntries,
381 makeErr(err, http.StatusInternalServerError)
383 } else if pathInfo == "/twtxt.txt" {
384 commit, err := repo.CommitObject(*headHash)
386 makeErr(err, http.StatusInternalServerError)
388 etagHash.Write([]byte("TWTXT POSTS"))
389 etagHash.Write(commit.Hash[:])
391 repoLog, err := repo.Log(&git.LogOptions{From: *headHash})
393 makeErr(err, http.StatusInternalServerError)
395 for i := 0; i < PageEntries; i++ {
396 commit, err = repoLog.Next()
402 commit.Author.When.Format(time.RFC3339),
403 msgSplit(commit.Message)[0],
406 os.Stdout.WriteString("Content-Type: text/plain; charset=utf-8\n")
407 os.Stdout.WriteString("ETag: " + etagString(etagHash) + "\n")
408 if gzipWriter != nil {
409 os.Stdout.WriteString("Content-Encoding: gzip\n")
412 os.Stdout.WriteString("\n")
413 os.Stdout.Write(outBuf.Bytes())
415 } else if pathInfo == "/"+AtomPostsFeed {
416 commit, err := repo.CommitObject(*headHash)
418 makeErr(err, http.StatusInternalServerError)
422 if t, exists := queryValues["topic"]; exists {
426 etagHash.Write([]byte("ATOM POSTS"))
427 etagHash.Write([]byte(topic))
428 etagHash.Write(commit.Hash[:])
434 title = fmt.Sprintf("%s (topic: %s)", cfg.Title, topic)
436 idHasher, err := blake2b.New256(nil)
440 idHasher.Write([]byte("ATOM POSTS"))
441 idHasher.Write([]byte(cfg.AtomId))
442 idHasher.Write([]byte(topic))
445 ID: "urn:uuid:" + bytes2uuid(idHasher.Sum(nil)),
446 Updated: atom.Time(commit.Author.When),
449 Href: cfg.AtomBaseURL + cfg.URLPrefix + "/" + AtomPostsFeed,
451 Author: &atom.Person{Name: cfg.AtomAuthor},
454 repoLog, err := repo.Log(&git.LogOptions{From: *headHash})
456 makeErr(err, http.StatusInternalServerError)
458 var commits CommitIterNext
462 topicsCache, err := getTopicsCache(cfg, repoLog)
464 makeErr(err, http.StatusInternalServerError)
466 hashes := topicsCache[topic]
468 makeErr(errors.New("no posts with that topic"), http.StatusBadRequest)
470 commits = &HashesIter{hashes}
473 for i := 0; i < PageEntries; i++ {
474 commit, err = commits.Next()
478 lines := msgSplit(commit.Message)
479 var categories []atom.Category
480 for _, topic := range sgblog.ParseTopics(sgblog.GetNote(
481 repo, topicsTree, commit.Hash,
483 categories = append(categories, atom.Category{Term: topic})
485 htmlized := make([]string, 0, len(lines))
486 htmlized = append(htmlized, "<pre>")
487 for _, l := range lines[2:] {
488 htmlized = append(htmlized, lineURLize(cfg.AtomBaseURL+cfg.URLPrefix, l))
490 htmlized = append(htmlized, "</pre>")
491 feed.Entry = append(feed.Entry, &atom.Entry{
493 ID: "urn:uuid:" + bytes2uuid(commit.Hash[:]),
496 Href: cfg.AtomBaseURL + cfg.URLPrefix + "/" + commit.Hash.String(),
498 Published: atom.Time(commit.Author.When),
499 Updated: atom.Time(commit.Author.When),
500 Summary: &atom.Text{Type: "text", Body: lines[0]},
503 Body: strings.Join(htmlized, "\n"),
505 Category: categories,
508 data, err := xml.MarshalIndent(&feed, "", " ")
510 makeErr(err, http.StatusInternalServerError)
514 } else if pathInfo == "/"+AtomCommentsFeed {
515 commit, err := repo.CommitObject(commentsRef.Hash())
517 makeErr(err, http.StatusInternalServerError)
519 etagHash.Write([]byte("ATOM COMMENTS"))
520 etagHash.Write(commit.Hash[:])
522 idHasher, err := blake2b.New256(nil)
526 idHasher.Write([]byte("ATOM COMMENTS"))
527 idHasher.Write([]byte(cfg.AtomId))
529 Title: cfg.Title + " comments",
530 ID: "urn:uuid:" + bytes2uuid(idHasher.Sum(nil)),
531 Updated: atom.Time(commit.Author.When),
534 Href: cfg.AtomBaseURL + cfg.URLPrefix + "/" + AtomCommentsFeed,
536 Author: &atom.Person{Name: cfg.AtomAuthor},
538 repoLog, err := repo.Log(&git.LogOptions{From: commentsRef.Hash()})
540 makeErr(err, http.StatusInternalServerError)
542 for i := 0; i < PageEntries; i++ {
543 commit, err = repoLog.Next()
547 fileStats, err := commit.Stats()
549 makeErr(err, http.StatusInternalServerError)
551 t, err := commit.Tree()
553 makeErr(err, http.StatusInternalServerError)
555 commentedHash := plumbing.NewHash(strings.ReplaceAll(
556 fileStats[0].Name, "/", "",
558 commit, err = repo.CommitObject(commentedHash)
562 comments := sgblog.ParseComments(sgblog.GetNote(repo, t, commentedHash))
563 if len(comments) == 0 {
566 commentN := strconv.Itoa(len(comments) - 1)
567 lines := strings.Split(comments[len(comments)-1], "\n")
568 from := strings.TrimPrefix(lines[0], "From: ")
569 date := strings.TrimPrefix(lines[1], "Date: ")
570 htmlized := make([]string, 0, len(lines))
571 htmlized = append(htmlized, "<pre>")
572 for _, l := range lines[2:] {
573 htmlized = append(htmlized, lineURLize(cfg.AtomBaseURL+cfg.URLPrefix, l))
575 htmlized = append(htmlized, "</pre>")
577 idHasher.Write([]byte("COMMENT"))
578 idHasher.Write(commit.Hash[:])
579 idHasher.Write([]byte(commentN))
580 feed.Entry = append(feed.Entry, &atom.Entry{
582 "Comment %s for \"%s\" by %s",
583 commentN, msgSplit(commit.Message)[0], from,
585 Author: &atom.Person{Name: from},
586 ID: "urn:uuid:" + bytes2uuid(idHasher.Sum(nil)),
589 Href: strings.Join([]string{
590 cfg.AtomBaseURL, cfg.URLPrefix, "/",
591 commit.Hash.String(), "#comment", commentN,
594 Published: atom.TimeStr(strings.Replace(date, " ", "T", -1)),
595 Updated: atom.TimeStr(strings.Replace(date, " ", "T", -1)),
598 Body: strings.Join(htmlized, "\n"),
602 data, err := xml.MarshalIndent(&feed, "", " ")
604 makeErr(err, http.StatusInternalServerError)
608 } else if sha1DigestRe.MatchString(pathInfo[1:]) {
609 commit, err := repo.CommitObject(plumbing.NewHash(pathInfo[1 : 1+sha1.Size*2]))
611 makeErr(err, http.StatusBadRequest)
613 for _, data := range etagHashForWeb {
614 etagHash.Write([]byte(data))
616 etagHash.Write([]byte("ENTRY"))
617 etagHash.Write(commit.Hash[:])
618 atomCommentsURL := strings.Join([]string{
619 cfg.AtomBaseURL, cfg.URLPrefix, "/",
620 commit.Hash.String(), "/", AtomCommentsFeed,
622 commentsRaw := sgblog.GetNote(repo, commentsTree, commit.Hash)
623 etagHash.Write(commentsRaw)
624 topicsRaw := sgblog.GetNote(repo, topicsTree, commit.Hash)
625 etagHash.Write(topicsRaw)
626 if strings.HasSuffix(pathInfo, AtomCommentsFeed) {
627 etagHash.Write([]byte("ATOM COMMENTS"))
629 type Comment struct {
635 commentsRaw := sgblog.ParseComments(commentsRaw)
637 if len(commentsRaw) > PageEntries {
638 toSkip = len(commentsRaw) - PageEntries
640 comments := make([]Comment, 0, len(commentsRaw)-toSkip)
641 for i := len(commentsRaw) - 1; i >= toSkip; i-- {
642 lines := strings.Split(commentsRaw[i], "\n")
643 from := strings.TrimPrefix(lines[0], "From: ")
644 date := strings.TrimPrefix(lines[1], "Date: ")
645 comments = append(comments, Comment{
648 date: strings.Replace(date, " ", "T", 1),
652 idHasher, err := blake2b.New256(nil)
656 idHasher.Write([]byte("ATOM COMMENTS"))
657 idHasher.Write(commit.Hash[:])
659 Title: fmt.Sprintf("\"%s\" comments", msgSplit(commit.Message)[0]),
660 ID: "urn:uuid:" + bytes2uuid(idHasher.Sum(nil)),
661 Link: []atom.Link{{Rel: "self", Href: atomCommentsURL}},
662 Author: &atom.Person{Name: cfg.AtomAuthor},
664 if len(comments) > 0 {
665 feed.Updated = atom.TimeStr(comments[0].date)
667 feed.Updated = atom.Time(commit.Author.When)
669 for _, comment := range comments {
671 idHasher.Write([]byte("COMMENT"))
672 idHasher.Write(commit.Hash[:])
673 idHasher.Write([]byte(comment.n))
674 htmlized := make([]string, 0, len(comment.body))
675 htmlized = append(htmlized, "<pre>")
676 for _, l := range comment.body {
679 lineURLize(cfg.AtomBaseURL+cfg.URLPrefix, l),
682 htmlized = append(htmlized, "</pre>")
683 feed.Entry = append(feed.Entry, &atom.Entry{
684 Title: fmt.Sprintf("Comment %s by %s", comment.n, comment.from),
685 Author: &atom.Person{Name: comment.from},
686 ID: "urn:uuid:" + bytes2uuid(idHasher.Sum(nil)),
689 Href: strings.Join([]string{
692 commit.Hash.String(),
693 "#comment", comment.n,
696 Published: atom.TimeStr(
697 strings.Replace(comment.date, " ", "T", -1),
699 Updated: atom.TimeStr(
700 strings.Replace(comment.date, " ", "T", -1),
704 Body: strings.Join(htmlized, "\n"),
708 data, err := xml.MarshalIndent(&feed, "", " ")
710 makeErr(err, http.StatusInternalServerError)
715 notesRaw := sgblog.GetNote(repo, notesTree, commit.Hash)
716 etagHash.Write(notesRaw)
719 lines := msgSplit(commit.Message)
721 when := commit.Author.When.Format(sgblog.WhenFmt)
723 if len(commit.ParentHashes) > 0 {
724 parent = commit.ParentHashes[0].String()
726 commentsParsed := sgblog.ParseComments(commentsRaw)
727 comments := make([]CommentEntry, 0, len(commentsParsed))
728 for _, comment := range commentsParsed {
729 lines := strings.Split(comment, "\n")
730 comments = append(comments, CommentEntry{lines[:3], lines[3:]})
732 var notesLines []string
733 if len(notesRaw) > 0 {
734 notesLines = strings.Split(string(notesRaw), "\n")
737 os.Stdout.Write([]byte(startHeader(etagHash, gzipWriter != nil)))
738 err = TmplHTMLEntry.Execute(out, struct {
744 AtomCommentsURL string
746 Commit *object.Commit
749 Comments []CommentEntry
752 Version: sgblog.Version,
755 TitleEscaped: url.PathEscape(fmt.Sprintf("Re: %s (%s)", title, commit.Hash)),
757 AtomCommentsURL: atomCommentsURL,
761 NoteLines: notesLines,
763 Topics: sgblog.ParseTopics(topicsRaw),
766 makeErr(err, http.StatusInternalServerError)
769 makeErr(errors.New("unknown URL action"), http.StatusNotFound)
771 out.Write([]byte("</body></html>\n"))
772 if gzipWriter != nil {
775 os.Stdout.Write(outBuf.Bytes())
779 os.Stdout.WriteString("Content-Type: application/atom+xml; charset=utf-8\n")
780 os.Stdout.WriteString("ETag: " + etagString(etagHash) + "\n")
781 if gzipWriter != nil {
782 os.Stdout.WriteString("Content-Encoding: gzip\n")
785 os.Stdout.WriteString("\n")
786 os.Stdout.Write(outBuf.Bytes())