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(repo, topicsTree, commit.Hash)) {
481 categories = append(categories, atom.Category{Term: topic})
483 htmlized := make([]string, 0, len(lines))
484 htmlized = append(htmlized, "<pre>")
485 for _, l := range lines[2:] {
486 htmlized = append(htmlized, lineURLize(cfg.AtomBaseURL+cfg.URLPrefix, l))
488 htmlized = append(htmlized, "</pre>")
489 feed.Entry = append(feed.Entry, &atom.Entry{
491 ID: "urn:uuid:" + bytes2uuid(commit.Hash[:]),
494 Href: cfg.AtomBaseURL + cfg.URLPrefix + "/" + commit.Hash.String(),
496 Published: atom.Time(commit.Author.When),
497 Updated: atom.Time(commit.Author.When),
498 Summary: &atom.Text{Type: "text", Body: lines[0]},
501 Body: strings.Join(htmlized, "\n"),
503 Category: categories,
506 data, err := xml.MarshalIndent(&feed, "", " ")
508 makeErr(err, http.StatusInternalServerError)
512 } else if pathInfo == "/"+AtomCommentsFeed {
513 commit, err := repo.CommitObject(commentsRef.Hash())
515 makeErr(err, http.StatusInternalServerError)
517 etagHash.Write([]byte("ATOM COMMENTS"))
518 etagHash.Write(commit.Hash[:])
520 idHasher, err := blake2b.New256(nil)
524 idHasher.Write([]byte("ATOM COMMENTS"))
525 idHasher.Write([]byte(cfg.AtomId))
527 Title: cfg.Title + " comments",
528 ID: "urn:uuid:" + bytes2uuid(idHasher.Sum(nil)),
529 Updated: atom.Time(commit.Author.When),
532 Href: cfg.AtomBaseURL + cfg.URLPrefix + "/" + AtomCommentsFeed,
534 Author: &atom.Person{Name: cfg.AtomAuthor},
536 repoLog, err := repo.Log(&git.LogOptions{From: commentsRef.Hash()})
538 makeErr(err, http.StatusInternalServerError)
540 for i := 0; i < PageEntries; i++ {
541 commit, err = repoLog.Next()
545 fileStats, err := commit.Stats()
547 makeErr(err, http.StatusInternalServerError)
549 t, err := commit.Tree()
551 makeErr(err, http.StatusInternalServerError)
553 commentedHash := plumbing.NewHash(strings.ReplaceAll(
554 fileStats[0].Name, "/", "",
556 commit, err = repo.CommitObject(commentedHash)
560 comments := sgblog.ParseComments(sgblog.GetNote(repo, t, commentedHash))
561 if len(comments) == 0 {
564 commentN := strconv.Itoa(len(comments) - 1)
565 lines := strings.Split(comments[len(comments)-1], "\n")
566 from := strings.TrimPrefix(lines[0], "From: ")
567 date := strings.TrimPrefix(lines[1], "Date: ")
568 htmlized := make([]string, 0, len(lines))
569 htmlized = append(htmlized, "<pre>")
570 for _, l := range lines[2:] {
571 htmlized = append(htmlized, lineURLize(cfg.AtomBaseURL+cfg.URLPrefix, l))
573 htmlized = append(htmlized, "</pre>")
575 idHasher.Write([]byte("COMMENT"))
576 idHasher.Write(commit.Hash[:])
577 idHasher.Write([]byte(commentN))
578 feed.Entry = append(feed.Entry, &atom.Entry{
580 "Comment %s for \"%s\" by %s",
581 commentN, msgSplit(commit.Message)[0], from,
583 Author: &atom.Person{Name: from},
584 ID: "urn:uuid:" + bytes2uuid(idHasher.Sum(nil)),
587 Href: strings.Join([]string{
588 cfg.AtomBaseURL, cfg.URLPrefix, "/",
589 commit.Hash.String(), "#comment", commentN,
592 Published: atom.TimeStr(strings.Replace(date, " ", "T", -1)),
593 Updated: atom.TimeStr(strings.Replace(date, " ", "T", -1)),
596 Body: strings.Join(htmlized, "\n"),
600 data, err := xml.MarshalIndent(&feed, "", " ")
602 makeErr(err, http.StatusInternalServerError)
606 } else if sha1DigestRe.MatchString(pathInfo[1:]) {
607 commit, err := repo.CommitObject(plumbing.NewHash(pathInfo[1 : 1+sha1.Size*2]))
609 makeErr(err, http.StatusBadRequest)
611 for _, data := range etagHashForWeb {
612 etagHash.Write([]byte(data))
614 etagHash.Write([]byte("ENTRY"))
615 etagHash.Write(commit.Hash[:])
616 atomCommentsURL := strings.Join([]string{
617 cfg.AtomBaseURL, cfg.URLPrefix, "/",
618 commit.Hash.String(), "/", AtomCommentsFeed,
620 commentsRaw := sgblog.GetNote(repo, commentsTree, commit.Hash)
621 etagHash.Write(commentsRaw)
622 topicsRaw := sgblog.GetNote(repo, topicsTree, commit.Hash)
623 etagHash.Write(topicsRaw)
624 if strings.HasSuffix(pathInfo, AtomCommentsFeed) {
625 etagHash.Write([]byte("ATOM COMMENTS"))
627 type Comment struct {
633 commentsRaw := sgblog.ParseComments(commentsRaw)
635 if len(commentsRaw) > PageEntries {
636 toSkip = len(commentsRaw) - PageEntries
638 comments := make([]Comment, 0, len(commentsRaw)-toSkip)
639 for i := len(commentsRaw) - 1; i >= toSkip; i-- {
640 lines := strings.Split(commentsRaw[i], "\n")
641 from := strings.TrimPrefix(lines[0], "From: ")
642 date := strings.TrimPrefix(lines[1], "Date: ")
643 comments = append(comments, Comment{
646 date: strings.Replace(date, " ", "T", 1),
650 idHasher, err := blake2b.New256(nil)
654 idHasher.Write([]byte("ATOM COMMENTS"))
655 idHasher.Write(commit.Hash[:])
657 Title: fmt.Sprintf("\"%s\" comments", msgSplit(commit.Message)[0]),
658 ID: "urn:uuid:" + bytes2uuid(idHasher.Sum(nil)),
659 Link: []atom.Link{{Rel: "self", Href: atomCommentsURL}},
660 Author: &atom.Person{Name: cfg.AtomAuthor},
662 if len(comments) > 0 {
663 feed.Updated = atom.TimeStr(comments[0].date)
665 feed.Updated = atom.Time(commit.Author.When)
667 for _, comment := range comments {
669 idHasher.Write([]byte("COMMENT"))
670 idHasher.Write(commit.Hash[:])
671 idHasher.Write([]byte(comment.n))
672 htmlized := make([]string, 0, len(comment.body))
673 htmlized = append(htmlized, "<pre>")
674 for _, l := range comment.body {
677 lineURLize(cfg.AtomBaseURL+cfg.URLPrefix, l),
680 htmlized = append(htmlized, "</pre>")
681 feed.Entry = append(feed.Entry, &atom.Entry{
682 Title: fmt.Sprintf("Comment %s by %s", comment.n, comment.from),
683 Author: &atom.Person{Name: comment.from},
684 ID: "urn:uuid:" + bytes2uuid(idHasher.Sum(nil)),
687 Href: strings.Join([]string{
690 commit.Hash.String(),
691 "#comment", comment.n,
694 Published: atom.TimeStr(
695 strings.Replace(comment.date, " ", "T", -1),
697 Updated: atom.TimeStr(
698 strings.Replace(comment.date, " ", "T", -1),
702 Body: strings.Join(htmlized, "\n"),
706 data, err := xml.MarshalIndent(&feed, "", " ")
708 makeErr(err, http.StatusInternalServerError)
713 notesRaw := sgblog.GetNote(repo, notesTree, commit.Hash)
714 etagHash.Write(notesRaw)
717 lines := msgSplit(commit.Message)
719 when := commit.Author.When.Format(sgblog.WhenFmt)
721 if len(commit.ParentHashes) > 0 {
722 parent = commit.ParentHashes[0].String()
724 commentsParsed := sgblog.ParseComments(commentsRaw)
725 comments := make([]CommentEntry, 0, len(commentsParsed))
726 for _, comment := range commentsParsed {
727 lines := strings.Split(comment, "\n")
728 comments = append(comments, CommentEntry{lines[:3], lines[3:]})
730 var notesLines []string
731 if len(notesRaw) > 0 {
732 notesLines = strings.Split(string(notesRaw), "\n")
735 os.Stdout.Write([]byte(startHeader(etagHash, gzipWriter != nil)))
736 err = TmplHTMLEntry.Execute(out, struct {
742 AtomCommentsURL string
744 Commit *object.Commit
747 Comments []CommentEntry
750 Version: sgblog.Version,
753 TitleEscaped: url.PathEscape(fmt.Sprintf("Re: %s (%s)", title, commit.Hash)),
755 AtomCommentsURL: atomCommentsURL,
759 NoteLines: notesLines,
761 Topics: sgblog.ParseTopics(topicsRaw),
764 makeErr(err, http.StatusInternalServerError)
767 makeErr(errors.New("unknown URL action"), http.StatusNotFound)
769 out.Write([]byte("</body></html>\n"))
770 if gzipWriter != nil {
773 os.Stdout.Write(outBuf.Bytes())
777 os.Stdout.WriteString("Content-Type: application/atom+xml; charset=UTF-8\n")
778 os.Stdout.WriteString("ETag: " + etagString(etagHash) + "\n")
779 if gzipWriter != nil {
780 os.Stdout.WriteString("Content-Encoding: gzip\n")
783 os.Stdout.WriteString("\n")
784 os.Stdout.Write(outBuf.Bytes())