// NOTE: lot's of code compied from Gitea with slight modification to use go-git objects

package gitea

import (
	"bufio"
	"bytes"
	"context"
	"fmt"
	"io"
	"math"
	"os/exec"
	"strconv"
	"strings"

	"github.com/djherbis/buffer"
	"github.com/djherbis/nio/v3"
	"github.com/go-git/go-git/v5/plumbing"
	"github.com/go-git/go-git/v5/plumbing/filemode"
	"github.com/go-git/go-git/v5/plumbing/hash"
	"github.com/go-git/go-git/v5/plumbing/object"
)

func GetCommit(ctx context.Context, repoPath, rev string) (*object.Commit, error) {
	wr, rd, cancel := CatFileBatch(ctx, repoPath)
	defer cancel()

	if _, err := wr.Write([]byte(rev + "\n")); err != nil {
		return nil, fmt.Errorf("write rev: %w", err)
	}
	sha, typ, size, err := ReadBatchLine(rd)
	if err != nil {
		return nil, err
	}
	if typ != "commit" {
		if err := DiscardFull(rd, size+1); err != nil {
			return nil, err
		}
		return nil, fmt.Errorf("unexpected type: %s for commit: %s", typ, rev)
	}
	commit, err := ReadCommit(plumbing.NewHash(string(sha)), io.LimitReader(rd, size))
	if err != nil {
		return nil, fmt.Errorf("read commit %s: %w", rev, err)
	}
	if _, err := rd.Discard(1); err != nil {
		return nil, err
	}
	return commit, nil
}

func GetTree(ctx context.Context, repoPath, rev string) (*object.Tree, error) {
	bw, br, cancel := CatFileBatch(ctx, repoPath)
	defer cancel()

	return BatchGetTree(bw, br, rev)
}

func BatchGetTree(bw io.WriteCloser, br *bufio.Reader, rev string) (*object.Tree, error) {
	if _, err := bw.Write([]byte(rev + "\n")); err != nil {
		return nil, fmt.Errorf("write rev: %w", err)
	}
	sha, typ, size, err := ReadBatchLine(br)
	if err != nil {
		return nil, fmt.Errorf("resolve %s: %w", rev, err)
	}
	if typ != "tree" {
		if err := DiscardFull(br, size+1); err != nil {
			return nil, err
		}
		return nil, fmt.Errorf("unexpected type: %s for tree: %s", typ, rev)
	}

	entries, err := catBatchParseTreeEntries(br, size)
	if err != nil {
		return nil, fmt.Errorf("read tree %s: %w", rev, err)
	}
	return &object.Tree{
		Hash:    plumbing.NewHash(string(sha)),
		Entries: entries,
	}, nil
}

func catBatchParseTreeEntries(rd *bufio.Reader, sz int64) ([]object.TreeEntry, error) {
	entries := make([]object.TreeEntry, 0, 10)
loop:
	for sz > 0 {
		mode, fname, sha, count, err := ParseCatFileTreeLine(rd)
		if err != nil {
			if err == io.EOF {
				break loop
			}
			return nil, err
		}
		modeNum, err := strconv.ParseUint(string(mode), 8, 32)
		if err != nil {
			return nil, err
		}
		sz -= int64(count)
		entry := object.TreeEntry{
			Name: string(fname),
			Mode: filemode.FileMode(modeNum),
			Hash: plumbing.Hash(sha),
		}
		entries = append(entries, entry)
	}
	if _, err := rd.Discard(1); err != nil {
		return entries, err
	}
	return entries, nil
}

func CatFileBatchCheck(ctx context.Context, repoPath string) (io.WriteCloser, *bufio.Reader, func()) {
	batchStdinReader, batchStdinWriter := io.Pipe()
	batchStdoutReader, batchStdoutWriter := nio.Pipe(buffer.New(32 * 1024))
	ctx, ctxCancel := context.WithCancel(ctx)
	closed := make(chan struct{})
	cancel := func() {
		ctxCancel()
		_ = batchStdinWriter.Close()
		_ = batchStdoutReader.Close()
		<-closed
	}

	// Ensure cancel is called as soon as the provided context is cancelled
	go func() {
		<-ctx.Done()
		cancel()
	}()

	go func() {
		stderr := &strings.Builder{}
		cmd := exec.CommandContext(ctx, "git", "-C", repoPath, "cat-file", "--batch-check")
		cmd.Stdin = batchStdinReader
		cmd.Stdout = batchStdoutWriter
		cmd.Stderr = stderr
		if err := cmd.Run(); err != nil {
			_ = batchStdinReader.CloseWithError(fmt.Errorf("%w\n%s", err, stderr.String()))
			_ = batchStdoutWriter.CloseWithError(fmt.Errorf("%w\n%s", err, stderr.String()))
		} else {
			_ = batchStdoutWriter.Close()
			_ = batchStdinReader.Close()
		}
		close(closed)
	}()

	batchReader := bufio.NewReaderSize(batchStdoutReader, 32*1024)
	return batchStdinWriter, batchReader, cancel
}

func CatFileBatch(ctx context.Context, repoPath string) (io.WriteCloser, *bufio.Reader, func()) {
	batchStdinReader, batchStdinWriter := io.Pipe()
	batchStdoutReader, batchStdoutWriter := nio.Pipe(buffer.New(32 * 1024))
	ctx, ctxCancel := context.WithCancel(ctx)
	closed := make(chan struct{})
	cancel := func() {
		ctxCancel()
		_ = batchStdinWriter.Close()
		_ = batchStdoutReader.Close()
		<-closed
	}

	// Ensure cancel is called as soon as the provided context is cancelled
	go func() {
		<-ctx.Done()
		cancel()
	}()

	go func() {
		stderr := &strings.Builder{}
		cmd := exec.CommandContext(ctx, "git", "-C", repoPath, "cat-file", "--batch")
		cmd.Stdin = batchStdinReader
		cmd.Stdout = batchStdoutWriter
		cmd.Stderr = stderr
		if err := cmd.Run(); err != nil {
			_ = batchStdinReader.CloseWithError(fmt.Errorf("%w\n%s", err, stderr.String()))
			_ = batchStdoutWriter.CloseWithError(fmt.Errorf("%w\n%s", err, stderr.String()))
		} else {
			_ = batchStdoutWriter.Close()
			_ = batchStdinReader.Close()
		}
		close(closed)
	}()

	batchReader := bufio.NewReaderSize(batchStdoutReader, 32*1024)
	return batchStdinWriter, batchReader, cancel
}

func ReadBatchLine(reader io.Reader) (sha []byte, typ string, size int64, err error) {
	rd, ok := reader.(*bufio.Reader)
	if !ok {
		rd = bufio.NewReader(reader)
	}
	typ, err = rd.ReadString('\n')
	if err != nil {
		return sha, typ, size, err
	}
	if len(typ) == 1 {
		typ, err = rd.ReadString('\n')
		if err != nil {
			return sha, typ, size, err
		}
	}
	idx := strings.IndexByte(typ, ' ')
	if idx < 0 {
		return sha, typ, size, fmt.Errorf("missing sha: %s", sha)
	}
	sha = []byte(typ[:idx])
	typ = typ[idx+1:]

	idx = strings.IndexByte(typ, ' ')
	if idx < 0 {
		return sha, typ, size, fmt.Errorf("missing size: %s", sha)
	}

	sizeStr := typ[idx+1 : len(typ)-1]
	typ = typ[:idx]

	size, err = strconv.ParseInt(sizeStr, 10, 64)
	return sha, typ, size, err
}

// NOTE: readCommit doesn't return complete go-git [object.Commit] object!
// The embedded object store is missing, so calling method from returned commit
// can lead to panic.
func ReadCommit(oid plumbing.Hash, reader io.Reader) (*object.Commit, error) {
	commit := &object.Commit{
		Hash:         oid,
		ExtraHeaders: make(map[string][]byte),
	}

	bufReader := bufio.NewReader(reader)
	inHeader := true
	var payloadSB, messageSB bytes.Buffer
	var headerKey string
	var headerValue []byte
	for {
		line, err := bufReader.ReadBytes('\n')
		if err != nil && err != io.EOF {
			return nil, err
		}
		if len(line) == 0 {
			break
		}

		if inHeader {
			inHeader = !(len(line) == 1 && line[0] == '\n') // a bare newline ends the header block
			k, v, _ := bytes.Cut(line, []byte{' '})
			if len(k) != 0 || !inHeader {
				if headerKey != "" {
					assignCommitHeader(commit, headerKey, headerValue)
				}
				headerKey = string(k) // also resets headerValue via the assignment below
				headerValue = v
			} else {
				headerValue = append(headerValue, v...)
			}
			if headerKey != "gpgsig" && headerKey != "gpgsig-sha256" {
				_, _ = payloadSB.Write(line)
			}
		} else {
			_, _ = messageSB.Write(line)
			_, _ = payloadSB.Write(line)
		}

		if err == io.EOF {
			break
		}
	}

	commit.Message = messageSB.String()
	// TODO: pass raw payload so we can verify it without reconstructing the payload
	// if commit.Signature != nil {
	// 	commit.Signature.Payload = payloadSB.String()
	// }
	return commit, nil
}

func assignCommitHeader(commit *object.Commit, headerKey string, headerValue []byte) {
	value := bytes.TrimSuffix(headerValue, []byte{'\n'})
	switch headerKey {
	case "tree":
		commit.TreeHash = plumbing.NewHash(string(value))
	case "parent":
		commit.ParentHashes = append(commit.ParentHashes, plumbing.NewHash(string(value)))
	case "author":
		commit.Author.Decode(value)
	case "committer":
		commit.Committer.Decode(value)
	case "gpgsig", "gpgsig-sha256":
		// if there are duplicate "gpgsig" and "gpgsig-sha256" headers, then the signature must have already been invalid
		// so we don't need to handle duplicate headers here
		commit.PGPSignature = string(value)
	case "mergetag":
		commit.MergeTag = string(value)
	default:
		commit.ExtraHeaders[headerKey] = value
	}
}

// ParseCatFileTreeLine reads an entry from a tree in a cat-file --batch stream
// This carefully avoids allocations - except where fnameBuf is too small.
// It is recommended therefore to pass in an fnameBuf large enough to avoid almost all allocations
//
// Each line is composed of:
// <mode-in-ascii-dropping-initial-zeros> SP <fname> NUL <binary HASH>
//
// We don't attempt to convert the raw HASH to save a lot of time
func ParseCatFileTreeLine(rd *bufio.Reader) (mode, fname, sha []byte, n int, err error) {
	modeBuf := make([]byte, 40)
	fnameBuf := make([]byte, 4096)
	shaBuf := make([]byte, hash.HexSize)

	var readBytes []byte

	// Read the Mode & fname
	readBytes, err = rd.ReadSlice('\x00')
	if err != nil {
		return mode, fname, sha, n, err
	}
	idx := bytes.IndexByte(readBytes, ' ')
	if idx < 0 {
		return mode, fname, sha, n, fmt.Errorf("missing")
	}

	n += idx + 1
	copy(modeBuf, readBytes[:idx])
	if len(modeBuf) >= idx {
		modeBuf = modeBuf[:idx]
	} else {
		modeBuf = append(modeBuf, readBytes[len(modeBuf):idx]...)
	}
	mode = modeBuf

	readBytes = readBytes[idx+1:]

	// Deal with the fname
	copy(fnameBuf, readBytes)
	if len(fnameBuf) > len(readBytes) {
		fnameBuf = fnameBuf[:len(readBytes)]
	} else {
		fnameBuf = append(fnameBuf, readBytes[len(fnameBuf):]...)
	}
	for err == bufio.ErrBufferFull {
		readBytes, err = rd.ReadSlice('\x00')
		fnameBuf = append(fnameBuf, readBytes...)
	}
	n += len(fnameBuf)
	if err != nil {
		return mode, fname, sha, n, err
	}
	fnameBuf = fnameBuf[:len(fnameBuf)-1]
	fname = fnameBuf

	// Deal with the binary hash
	idx = 0
	length := hash.HexSize / 2
	for idx < length {
		var read int
		read, err = rd.Read(shaBuf[idx:length])
		n += read
		if err != nil {
			return mode, fname, sha, n, err
		}
		idx += read
	}
	sha = shaBuf
	return mode, fname, sha, n, err
}

func DiscardFull(rd *bufio.Reader, discard int64) error {
	if discard > math.MaxInt32 {
		n, err := rd.Discard(math.MaxInt32)
		discard -= int64(n)
		if err != nil {
			return err
		}
	}
	for discard > 0 {
		n, err := rd.Discard(int(discard))
		discard -= int64(n)
		if err != nil {
			return err
		}
	}
	return nil
}