// Copyright 2020 The Gogs Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package repo

import (
	"fmt"

	"github.com/gogs/git-module"

	"gogs.io/gogs/internal/context"
	"gogs.io/gogs/internal/gitutil"
)

func GetRepoGitTree(c *context.APIContext) {
	gitRepo, err := git.Open(c.Repo.Repository.RepoPath())
	if err != nil {
		c.ServerError("open repository", err)
		return
	}

	sha := c.Params(":sha")
	tree, err := gitRepo.LsTree(sha)
	if err != nil {
		c.NotFoundOrServerError("get tree", gitutil.IsErrRevisionNotExist, err)
		return
	}

	entries, err := tree.Entries()
	if err != nil {
		c.ServerError("list entries", err)
		return
	}

	type repoGitTreeEntry struct {
		Path string `json:"path"`
		Mode string `json:"mode"`
		Type string `json:"type"`
		Size int64  `json:"size"`
		Sha  string `json:"sha"`
		URL  string `json:"url"`
	}
	type repoGitTree struct {
		Sha  string              `json:"sha"`
		URL  string              `json:"url"`
		Tree []*repoGitTreeEntry `json:"tree"`
	}

	treesURL := fmt.Sprintf("%s/repos/%s/%s/git/trees", c.BaseURL, c.Params(":username"), c.Params(":reponame"))

	if len(entries) == 0 {
		c.JSONSuccess(&repoGitTree{
			Sha: sha,
			URL: fmt.Sprintf(treesURL+"/%s", sha),
		})
		return
	}

	children := make([]*repoGitTreeEntry, 0, len(entries))
	for _, entry := range entries {
		var mode string
		switch entry.Type() {
		case git.ObjectCommit:
			mode = "160000"
		case git.ObjectTree:
			mode = "040000"
		case git.ObjectBlob:
			mode = "120000"
		case git.ObjectTag:
			mode = "100644"
		default:
			panic("unreachable")
		}
		children = append(children, &repoGitTreeEntry{
			Path: entry.Name(),
			Mode: mode,
			Type: string(entry.Type()),
			Size: entry.Size(),
			Sha:  entry.ID().String(),
			URL:  fmt.Sprintf(treesURL+"/%s", entry.ID().String()),
		})
	}
	c.JSONSuccess(&repoGitTree{
		Sha:  c.Params(":sha"),
		URL:  fmt.Sprintf(treesURL+"/%s", sha),
		Tree: children,
	})
}