subscribe-bot/web/repo.go

package web

import (
	"archive/zip"
	"errors"
	"fmt"
	"io"
	"net/http"
	"path"
	"strconv"
	"time"

	"github.com/dustin/go-humanize"
	"github.com/gin-gonic/gin"
	"github.com/go-git/go-git/v5"
	"github.com/go-git/go-git/v5/plumbing"
	"github.com/go-git/go-git/v5/plumbing/object"
)

func (web *Web) mapVersions(c *gin.Context) {
	userId := c.Param("userId")
	mapId := c.Param("mapId")

	id, _ := strconv.Atoi(mapId)
	bs, _ := web.api.GetBeatmapSet(id)

	repoDir := path.Join(web.config.Repos, userId, mapId)
	repo, _ := git.PlainOpen(repoDir)

	type Revision struct {
		Date      time.Time
		HumanDate string
		Summary   string
		Hash      string
		HasParent bool
	}

	versions := make([]Revision, 0)
	logIter, _ := repo.Log(&git.LogOptions{})
	for i := 0; i < 20; i++ {
		commit, err := logIter.Next()
		if err == io.EOF {
			break
		}

		stats, _ := commit.Stats()
		_, err = commit.Parent(0)
		hasParent := !errors.Is(err, object.ErrParentNotFound)

		versions = append(versions, Revision{
			Date:      commit.Author.When,
			HumanDate: humanize.Time(commit.Author.When),
			Summary:   stats.String(),
			Hash:      commit.Hash.String(),
			HasParent: hasParent,
		})
	}

	c.HTML(http.StatusOK, "map-version.html", gin.H{
		"Beatmapset": bs,
		"LoggedIn":   isLoggedIn(c),
		"Versions":   versions,
	})
}

func (web *Web) mapPatch(c *gin.Context) {
	userId := c.Param("userId")
	mapId := c.Param("mapId")
	hash := c.Param("hash")

	repoDir := path.Join(web.config.Repos, userId, mapId)
	repo, _ := git.PlainOpen(repoDir)

	hashObj := plumbing.NewHash(hash)
	commit, _ := repo.CommitObject(hashObj)
	parent, _ := commit.Parent(0)
	patch, _ := parent.Patch(commit)

	c.String(http.StatusOK, "text/plain", patch.String())
}

func (web *Web) mapZip(c *gin.Context) {
	userId := c.Param("userId")
	mapId := c.Param("mapId")
	hash := c.Param("hash")

	repoDir := path.Join(web.config.Repos, userId, mapId)
	repo, _ := git.PlainOpen(repoDir)

	hashObj := plumbing.NewHash(hash)
	commit, _ := repo.CommitObject(hashObj)
	tree, _ := commit.Tree()

	files := tree.Files()

	c.Writer.Header().Set("Content-type", "application/octet-stream")
	c.Writer.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s-%s.zip", mapId, hash))
	c.Stream(func(w io.Writer) bool {
		ar := zip.NewWriter(w)
		for {
			file, err := files.Next()
			if err == io.EOF {
				break
			}

			reader, _ := file.Reader()
			fdest, _ := ar.Create(file.Name)
			io.Copy(fdest, reader)
		}
		ar.Close()

		return false
	})
}