131 lines
2.7 KiB
Go
131 lines
2.7 KiB
Go
package driver
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/vmware/govmomi/object"
|
|
"github.com/vmware/govmomi/vim25/mo"
|
|
"github.com/vmware/govmomi/vim25/soap"
|
|
"github.com/vmware/govmomi/vim25/types"
|
|
)
|
|
|
|
type Datastore struct {
|
|
ds *object.Datastore
|
|
driver *Driver
|
|
}
|
|
|
|
func (d *Driver) NewDatastore(ref *types.ManagedObjectReference) *Datastore {
|
|
return &Datastore{
|
|
ds: object.NewDatastore(d.client.Client, *ref),
|
|
driver: d,
|
|
}
|
|
}
|
|
|
|
// If name is an empty string, then resolve host's one
|
|
func (d *Driver) FindDatastore(name string, host string) (*Datastore, error) {
|
|
if name == "" {
|
|
h, err := d.FindHost(host)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
i, err := h.Info("datastore")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(i.Datastore) > 1 {
|
|
return nil, fmt.Errorf("Host has multiple datastores. Specify it explicitly")
|
|
}
|
|
|
|
ds := d.NewDatastore(&i.Datastore[0])
|
|
inf, err := ds.Info("name")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
name = inf.Name
|
|
}
|
|
|
|
ds, err := d.finder.Datastore(d.ctx, name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Datastore{
|
|
ds: ds,
|
|
driver: d,
|
|
}, nil
|
|
}
|
|
|
|
func (ds *Datastore) Info(params ...string) (*mo.Datastore, error) {
|
|
var p []string
|
|
if len(params) == 0 {
|
|
p = []string{"*"}
|
|
} else {
|
|
p = params
|
|
}
|
|
var info mo.Datastore
|
|
err := ds.ds.Properties(ds.driver.ctx, ds.ds.Reference(), p, &info)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &info, nil
|
|
}
|
|
|
|
func (ds *Datastore) FileExists(path string) bool {
|
|
_, err := ds.ds.Stat(ds.driver.ctx, path)
|
|
return err == nil
|
|
}
|
|
|
|
func (ds *Datastore) Name() string {
|
|
return ds.ds.Name()
|
|
}
|
|
|
|
func (ds *Datastore) ResolvePath(path string) string {
|
|
return ds.ds.Path(path)
|
|
}
|
|
|
|
func (ds *Datastore) UploadFile(src, dst string, host string) error {
|
|
p := soap.DefaultUpload
|
|
ctx := ds.driver.ctx
|
|
|
|
if host != "" {
|
|
h, err := ds.driver.FindHost(host)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ctx = ds.ds.HostContext(ctx, h.host)
|
|
}
|
|
|
|
return ds.ds.UploadFile(ctx, src, dst, &p)
|
|
}
|
|
|
|
func (ds *Datastore) Delete(path string) error {
|
|
dc, err := ds.driver.finder.Datacenter(ds.driver.ctx, ds.ds.DatacenterPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fm := ds.ds.NewFileManager(dc, false)
|
|
return fm.Delete(ds.driver.ctx, path)
|
|
}
|
|
|
|
func (ds *Datastore) MakeDirectory(path string) error {
|
|
dc, err := ds.driver.finder.Datacenter(ds.driver.ctx, ds.ds.DatacenterPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fm := ds.ds.NewFileManager(dc, false)
|
|
return fm.FileManager.MakeDirectory(ds.driver.ctx, path, dc, true)
|
|
}
|
|
|
|
// Cuts out the datastore prefix
|
|
// Example: "[datastore1] file.ext" --> "file.ext"
|
|
func RemoveDatastorePrefix(path string) string {
|
|
res := object.DatastorePath{}
|
|
if hadPrefix := res.FromString(path); hadPrefix {
|
|
return res.Path
|
|
} else {
|
|
return path
|
|
}
|
|
}
|