mirror of https://github.com/octoleo/restic.git
120 lines
1.7 KiB
Go
120 lines
1.7 KiB
Go
package khepri
|
|
|
|
import "os"
|
|
|
|
type Object struct {
|
|
repo *Repository
|
|
|
|
id ID
|
|
tpe Type
|
|
|
|
hw HashingWriter
|
|
file *os.File
|
|
}
|
|
|
|
func (repo *Repository) NewObject(t Type) (*Object, error) {
|
|
obj := &Object{
|
|
repo: repo,
|
|
tpe: t,
|
|
}
|
|
|
|
return obj, obj.open()
|
|
}
|
|
|
|
func (obj *Object) open() error {
|
|
if obj.isFinal() {
|
|
panic("object is finalized")
|
|
}
|
|
|
|
if obj.isOpen() {
|
|
panic("object already open")
|
|
}
|
|
|
|
// create tempfile in repository
|
|
if obj.hw == nil {
|
|
// save contents to tempfile, hash while writing
|
|
var err error
|
|
obj.file, err = obj.repo.tempFile()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// create hashing writer
|
|
obj.hw = NewHashingWriter(obj.file, obj.repo.hash)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (obj *Object) isOpen() bool {
|
|
return obj.file != nil && obj.hw != nil
|
|
}
|
|
|
|
func (obj *Object) isFinal() bool {
|
|
return obj.id != nil
|
|
}
|
|
|
|
func (obj *Object) Write(data []byte) (int, error) {
|
|
if !obj.isOpen() {
|
|
panic("object not open")
|
|
}
|
|
|
|
return obj.hw.Write(data)
|
|
}
|
|
|
|
func (obj *Object) Close() error {
|
|
if obj.file == nil || obj.hw == nil {
|
|
panic("object is not open")
|
|
}
|
|
|
|
obj.file.Close()
|
|
|
|
hash := obj.hw.Hash()
|
|
|
|
// move file to final name using hash of contents
|
|
id := ID(hash)
|
|
err := obj.repo.renameFile(obj.file, obj.tpe, id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
obj.hw = nil
|
|
obj.file = nil
|
|
|
|
obj.id = id
|
|
return nil
|
|
}
|
|
|
|
func (obj *Object) ID() ID {
|
|
if !obj.isFinal() {
|
|
panic("object not finalized")
|
|
}
|
|
|
|
return obj.id
|
|
}
|
|
|
|
func (obj *Object) Type() Type {
|
|
return obj.tpe
|
|
}
|
|
|
|
func (obj *Object) Remove() error {
|
|
if obj.id != nil {
|
|
return obj.repo.Remove(obj.tpe, obj.id)
|
|
}
|
|
|
|
if obj.file != nil {
|
|
file := obj.file
|
|
obj.hw = nil
|
|
obj.file = nil
|
|
|
|
err := file.Close()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return os.Remove(file.Name())
|
|
}
|
|
|
|
return nil
|
|
}
|