mirror of
https://github.com/dolthub/dolt.git
synced 2026-02-05 02:59:44 -06:00
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package ref
|
|
|
|
import (
|
|
"crypto/sha1"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"regexp"
|
|
|
|
. "github.com/attic-labs/noms/dbg"
|
|
)
|
|
|
|
var (
|
|
// In the future we will allow different digest types, so this will get more complicated. For now sha1 is fine.
|
|
pattern = regexp.MustCompile("^sha1-([0-9a-f]{40})$")
|
|
)
|
|
|
|
type Sha1Digest [sha1.Size]byte
|
|
|
|
type Ref struct {
|
|
// In the future, we will also store the algorithm, and digest will thus probably have to be a slice (because it can vary in size)
|
|
digest Sha1Digest
|
|
}
|
|
|
|
func (r Ref) String() string {
|
|
return fmt.Sprintf("sha1-%s", hex.EncodeToString(r.digest[:]))
|
|
}
|
|
|
|
func New(digest Sha1Digest) Ref {
|
|
return Ref{digest}
|
|
}
|
|
|
|
func Parse(s string) (r Ref, err error) {
|
|
match := pattern.FindStringSubmatch(s)
|
|
if match == nil {
|
|
return r, fmt.Errorf("Could not parse ref: %s", s)
|
|
}
|
|
|
|
// TODO: The new temp byte array is kinda bummer. Would be better to walk the string and decode each byte into result.digest. But can't find stdlib functions to do that.
|
|
n, err := hex.Decode(r.digest[:], []byte(match[1]))
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// If there was no error, we should have decoded exactly one digest worth of bytes.
|
|
Chk.Equal(sha1.Size, n)
|
|
return
|
|
}
|
|
|
|
func MustParse(s string) Ref {
|
|
r, err := Parse(s)
|
|
Chk.NoError(err)
|
|
return r
|
|
}
|