forked from mirror/gitea
4eb2a29910
The 4 functions are duplicated, especially as interface methods. I think we just need to keep `MustID` the only one and remove other 3. ``` MustID(b []byte) ObjectID MustIDFromString(s string) ObjectID NewID(b []byte) (ObjectID, error) NewIDFromString(s string) (ObjectID, error) ``` Introduced the new interfrace method `ComputeHash` which will replace the interface `HasherInterface`. Now we don't need to keep two interfaces. Reintroduced `git.NewIDFromString` and `git.MustIDFromString`. The new function will detect the hash length to decide which objectformat of it. If it's 40, then it's SHA1. If it's 64, then it's SHA256. This will be right if the commitID is a full one. So the parameter should be always a full commit id. @AdamMajer Please review.
102 lines
2.3 KiB
Go
102 lines
2.3 KiB
Go
// Copyright 2014 The Gogs Authors. All rights reserved.
|
|
// Copyright 2019 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package git
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"strings"
|
|
)
|
|
|
|
// ObjectType git object type
|
|
type ObjectType string
|
|
|
|
const (
|
|
// ObjectCommit commit object type
|
|
ObjectCommit ObjectType = "commit"
|
|
// ObjectTree tree object type
|
|
ObjectTree ObjectType = "tree"
|
|
// ObjectBlob blob object type
|
|
ObjectBlob ObjectType = "blob"
|
|
// ObjectTag tag object type
|
|
ObjectTag ObjectType = "tag"
|
|
// ObjectBranch branch object type
|
|
ObjectBranch ObjectType = "branch"
|
|
)
|
|
|
|
// Bytes returns the byte array for the Object Type
|
|
func (o ObjectType) Bytes() []byte {
|
|
return []byte(o)
|
|
}
|
|
|
|
type EmptyReader struct{}
|
|
|
|
func (EmptyReader) Read(p []byte) (int, error) {
|
|
return 0, io.EOF
|
|
}
|
|
|
|
func (repo *Repository) GetObjectFormat() (ObjectFormat, error) {
|
|
if repo != nil && repo.objectFormat != nil {
|
|
return repo.objectFormat, nil
|
|
}
|
|
|
|
str, err := repo.hashObject(EmptyReader{}, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
hash, err := NewIDFromString(str)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
repo.objectFormat = hash.Type()
|
|
|
|
return repo.objectFormat, nil
|
|
}
|
|
|
|
// HashObject takes a reader and returns hash for that reader
|
|
func (repo *Repository) HashObject(reader io.Reader) (ObjectID, error) {
|
|
idStr, err := repo.hashObject(reader, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewIDFromString(idStr)
|
|
}
|
|
|
|
func (repo *Repository) hashObject(reader io.Reader, save bool) (string, error) {
|
|
var cmd *Command
|
|
if save {
|
|
cmd = NewCommand(repo.Ctx, "hash-object", "-w", "--stdin")
|
|
} else {
|
|
cmd = NewCommand(repo.Ctx, "hash-object", "--stdin")
|
|
}
|
|
stdout := new(bytes.Buffer)
|
|
stderr := new(bytes.Buffer)
|
|
err := cmd.Run(&RunOpts{
|
|
Dir: repo.Path,
|
|
Stdin: reader,
|
|
Stdout: stdout,
|
|
Stderr: stderr,
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return strings.TrimSpace(stdout.String()), nil
|
|
}
|
|
|
|
// GetRefType gets the type of the ref based on the string
|
|
func (repo *Repository) GetRefType(ref string) ObjectType {
|
|
if repo.IsTagExist(ref) {
|
|
return ObjectTag
|
|
} else if repo.IsBranchExist(ref) {
|
|
return ObjectBranch
|
|
} else if repo.IsCommitExist(ref) {
|
|
return ObjectCommit
|
|
} else if _, err := repo.GetBlob(ref); err == nil {
|
|
return ObjectBlob
|
|
}
|
|
return ObjectType("invalid")
|
|
}
|