134 lines
3.0 KiB
Go
134 lines
3.0 KiB
Go
/*
|
|
Copyright (c) 2015 VMware, Inc. All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package logs
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"path"
|
|
|
|
"golang.org/x/net/context"
|
|
|
|
"github.com/vmware/govmomi/govc/cli"
|
|
"github.com/vmware/govmomi/govc/flags"
|
|
"github.com/vmware/govmomi/object"
|
|
"github.com/vmware/govmomi/vim25"
|
|
"github.com/vmware/govmomi/vim25/soap"
|
|
"github.com/vmware/govmomi/vim25/types"
|
|
)
|
|
|
|
type download struct {
|
|
*flags.DatacenterFlag
|
|
|
|
IncludeDefault bool
|
|
}
|
|
|
|
func init() {
|
|
cli.Register("logs.download", &download{})
|
|
}
|
|
|
|
func (cmd *download) Register(ctx context.Context, f *flag.FlagSet) {
|
|
cmd.DatacenterFlag, ctx = flags.NewDatacenterFlag(ctx)
|
|
cmd.DatacenterFlag.Register(ctx, f)
|
|
|
|
f.BoolVar(&cmd.IncludeDefault, "default", true, "Specifies if the bundle should include the default server")
|
|
}
|
|
|
|
func (cmd *download) Process(ctx context.Context) error {
|
|
if err := cmd.DatacenterFlag.Process(ctx); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (cmd *download) Usage() string {
|
|
return "[PATH]..."
|
|
}
|
|
|
|
func (cmd *download) DownloadFile(c *vim25.Client, b string) error {
|
|
u, err := c.Client.ParseURL(b)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
dst := path.Base(u.Path)
|
|
p := soap.DefaultDownload
|
|
if cmd.OutputFlag.TTY {
|
|
logger := cmd.ProgressLogger(fmt.Sprintf("Downloading %s... ", dst))
|
|
defer logger.Wait()
|
|
p.Progress = logger
|
|
}
|
|
|
|
return c.Client.DownloadFile(dst, u, &p)
|
|
}
|
|
|
|
func (cmd *download) GenerateLogBundles(m *object.DiagnosticManager, host []*object.HostSystem) ([]types.DiagnosticManagerBundleInfo, error) {
|
|
logger := cmd.ProgressLogger("Generating log bundles... ")
|
|
defer logger.Wait()
|
|
|
|
task, err := m.GenerateLogBundles(context.TODO(), cmd.IncludeDefault, host)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
r, err := task.WaitForResult(context.TODO(), logger)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return r.Result.(types.ArrayOfDiagnosticManagerBundleInfo).DiagnosticManagerBundleInfo, nil
|
|
}
|
|
|
|
func (cmd *download) Run(ctx context.Context, f *flag.FlagSet) error {
|
|
finder, err := cmd.Finder()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var host []*object.HostSystem
|
|
|
|
for _, arg := range f.Args() {
|
|
hs, err := finder.HostSystemList(context.TODO(), arg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
host = append(host, hs...)
|
|
}
|
|
|
|
c, err := cmd.Client()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
m := object.NewDiagnosticManager(c)
|
|
|
|
bundles, err := cmd.GenerateLogBundles(m, host)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, bundle := range bundles {
|
|
err := cmd.DownloadFile(c, bundle.Url)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|