/*
 * Copyright 1999-2020 Alibaba Group Holding Ltd.
 *
 * 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 cmd

import (
	"context"
	"fmt"
	"strings"

	"github.com/spf13/cobra"

	"github.com/chaosblade-io/chaosblade-spec-go/channel"
	"github.com/chaosblade-io/chaosblade-spec-go/spec"
)

type QueryDiskCommand struct {
	baseCommand
}

const MountPointArg = "mount-point"

func (qdc *QueryDiskCommand) Init() {
	qdc.command = &cobra.Command{
		Use:   "disk device",
		Short: "Query disk information",
		Long:  "Query disk information for chaos experiments of disk",
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			return qdc.queryDiskInfo(cmd, args[0])
		},
		Example: qdc.queryDiskExample(),
	}
}

func (qdc *QueryDiskCommand) queryDiskExample() string {
	return `blade query disk mount-point`
}

func (qdc *QueryDiskCommand) queryDiskInfo(command *cobra.Command, arg string) error {
	switch arg {
	case MountPointArg:
		response := channel.NewLocalChannel().Run(context.TODO(), "df",
			fmt.Sprintf(`-h | awk 'NR!=1 {print $1","$NF}' | tr '\n' ' '`))
		if !response.Success {
			return response
		}
		disks := response.Result.(string)
		fields := strings.Fields(disks)
		result := make([]string, 0)
		for _, disk := range fields {
			// TODO Check the file system prefix, but should check the file system type
			if strings.HasPrefix(disk, "/") {
				arr := strings.Split(disk, ",")
				if len(arr) < 2 {
					continue
				}
				result = append(result, arr[1])
			}
		}
		command.Println(spec.ReturnSuccess(result))
	default:
		return fmt.Errorf("the %s argument not found", arg)
	}
	return nil
}
