Cloudpods/backend/pkg/mcclient/options/ansibleplaybooks.go

// Copyright 2019 Yunion
//
// 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 options

import (
	"fmt"
	"io/ioutil"
	"strings"

	"yunion.io/x/jsonutils"

	apis "yunion.io/x/onecloud/pkg/apis/ansible"
	"yunion.io/x/onecloud/pkg/util/ansible"
)

type AnsiblePlaybookIdOptions struct {
	ID string `help:"name/id of the playbook"`
}

type AnsiblePlaybookListOptions struct {
	BaseListOptions
}

type AnsiblePlaybookCommonOptions struct {
	Host []string `help:"name or id of server or host in format '<[server:]id|host:id>|ipaddr var=val'"`
	Mod  []string `help:"ansible modules and their arguments in format 'name k1=v1 k2=v2'"`
	File []string `help:"files for use by modules, e.g. name=content, name=@file"`
}

func (opts *AnsiblePlaybookCommonOptions) ToPlaybook() (*ansible.Playbook, error) {
	if len(opts.Mod) == 0 {
		return nil, fmt.Errorf("Requires at least one --mod argument")
	}
	if len(opts.Host) == 0 {
		return nil, fmt.Errorf("Requires at least one server/host to operate on")
	}
	pb := ansible.NewPlaybook()
	hosts := []ansible.Host{}
	for _, s := range opts.Host {
		host, err := ansible.ParseHostLine(s)
		if err != nil {
			return nil, err
		}
		hosts = append(hosts, host)
	}
	pb.Inventory = ansible.Inventory{Hosts: hosts}
	for _, s := range opts.Mod {
		module, err := ansible.ParseModuleLine(s)
		if err != nil {
			return nil, err
		}
		pb.Modules = append(pb.Modules, module)
	}
	files := map[string][]byte{}
	for _, s := range opts.File {
		i := strings.IndexByte(s, '=')
		if i < 0 {
			return nil, fmt.Errorf("missing '=' in argument for --file.  Read command help")
		}
		name := strings.TrimSpace(s[:i])
		if name == "" {
			return nil, fmt.Errorf("empty file name: %s", s)
		}
		v := s[i+1:]
		if len(v) > 0 && v[0] == '@' {
			path := v[1:]
			d, err := ioutil.ReadFile(path)
			if err != nil {
				return nil, fmt.Errorf("read file %s: %v", path, err)
			}
			files[name] = d
		} else {
			files[name] = []byte(v)
		}
	}
	pb.Files = files
	return pb, nil
}

type AnsiblePlaybookCreateOptions struct {
	NAME string `help:"name of the playbook"`
	AnsiblePlaybookCommonOptions
}

func (opts *AnsiblePlaybookCreateOptions) Params() (*jsonutils.JSONDict, error) {
	pb, err := opts.AnsiblePlaybookCommonOptions.ToPlaybook()
	if err != nil {
		return nil, err
	}
	input := &apis.AnsiblePlaybookCreateInput{
		Name:     opts.NAME,
		Playbook: *pb,
	}
	params := input.JSON(input)
	return params, nil
}

type AnsiblePlaybookUpdateOptions struct {
	ID   string `json:"-" help:"name/id of the playbook"`
	Name string
	AnsiblePlaybookCommonOptions
}

func (opts *AnsiblePlaybookUpdateOptions) Params() (*jsonutils.JSONDict, error) {
	pb, err := opts.AnsiblePlaybookCommonOptions.ToPlaybook()
	if err != nil {
		return nil, err
	}
	input := &apis.AnsiblePlaybookUpdateInput{
		Name:     opts.Name,
		Playbook: *pb,
	}
	params := input.JSON(input)
	return params, nil
}