cli/main.go

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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 main

// import "github.com/apache/ozone-go"
import (
	"encoding/json"
	"fmt"
	"github.com/apache/ozone-go/api"
	"github.com/urfave/cli"
	"os"
	"strings"
)

var version string
var commit string
var date string

func main() {

	app := cli.NewApp()
	app.Name = "ozone"
	app.Usage = "Ozone command line client"
	app.Description = "Native Ozone command line client"
	app.Version = fmt.Sprintf("%s (%s, %s)", version, commit, date)
	app.Flags = []cli.Flag{
		cli.StringFlag{
			Name:     "om",
			Required: true,
			Value:    "localhost",
			Usage:    "Host (or host:port) address of the OzoneManager",
		},
	}
	app.Commands = []cli.Command{
		{
			Name:    "volume",
			Aliases: []string{"v", "vol"},
			Usage:   "Ozone volume related operations",
			Flags:   []cli.Flag{},
			Subcommands: []cli.Command{
				{
					Name:    "list",
					Aliases: []string{"ls"},
					Usage:   "List volumes.",
					Action: func(c *cli.Context) error {
						ozoneClient := api.CreateOzoneClient(c.GlobalString("om"))
						volumes, err := ozoneClient.ListVolumes()
						if err != nil {
							return err
						}
						for _, volume := range volumes {
							println(volume.Name)
						}
						return nil
					},
				},
				{
					Name:    "create",
					Aliases: []string{"mk"},
					Usage:   "Create volume.",
					Action: func(c *cli.Context) error {
						ozoneClient := api.CreateOzoneClient(c.GlobalString("om"))
						address := OzoneObjectAddressFromString(c.Args().Get(0))
						err := ozoneClient.CreateVolume(*address.Volume)
						if err != nil {
							return err
						}
						return nil
					},
				},
			},
		},
		{
			Name:    "bucket",
			Aliases: []string{"b"},
			Usage:   "Ozone bucket related operations",
			Flags:   []cli.Flag{},
			Subcommands: []cli.Command{
				{
					Name:    "create",
					Aliases: []string{"mk"},
					Usage:   "Create bucket.",
					Action: func(c *cli.Context) error {
						ozoneClient := api.CreateOzoneClient(c.GlobalString("om"))
						address := OzoneObjectAddressFromString(c.Args().Get(0))
						err := ozoneClient.CreateBucket(*address.Volume, *address.Bucket)
						if err != nil {
							return err
						}
						return nil
					},
				},
				{
					Name:    "list",
					Aliases: []string{"ls"},
					Usage:   "List buckets",
					Action: func(c *cli.Context) error {
						ozoneClient := api.CreateOzoneClient(c.GlobalString("om"))
						address := OzoneObjectAddressFromString(c.Args().Get(0))
						buckets, err := ozoneClient.ListBucket(*address.Volume)
						if err != nil {
							return err
						}
						for _, bucket := range buckets {
							println(bucket.Name)
						}
						return nil
					},
				},
			},
		},
		{
			Name:    "key",
			Aliases: []string{"k"},
			Usage:   "Ozone key related operations",
			Flags:   []cli.Flag{},
			Subcommands: []cli.Command{
				{
					Name:    "list",
					Aliases: []string{"ls"},
					Usage:   "List keys.",
					Action: func(c *cli.Context) error {
						ozoneClient := api.CreateOzoneClient(c.GlobalString("om"))
						address := OzoneObjectAddressFromString(c.Args().Get(0))
						keys, err := ozoneClient.ListKeys(*address.Volume, *address.Bucket)
						if err != nil {
							return err
						}
						out, err := json.MarshalIndent(keys, "", "   ")
						if err != nil {
							return err
						}

						println(string(out))
						return nil
					},
				},
				{
					Name:    "info",
					Aliases: []string{"show"},
					Usage:   "Show information about one key",
					Action: func(c *cli.Context) error {
						ozoneClient := api.CreateOzoneClient(c.GlobalString("om"))
						address := OzoneObjectAddressFromString(c.Args().Get(0))
						key, err := ozoneClient.InfoKey(*address.Volume, *address.Bucket, *address.Key)
						if err != nil {
							return err
						}
						out, err := json.MarshalIndent(key, "", "   ")
						if err != nil {
							return err
						}

						println(string(out))
						return nil
					},
				},
				{
					Name:    "cat",
					Aliases: []string{"c"},
					Usage:   "Show content of a file",
					Action: func(c *cli.Context) error {
						ozoneClient := api.CreateOzoneClient(c.GlobalString("om"))
						address := OzoneObjectAddressFromString(c.Args().Get(0))
						_, err := ozoneClient.GetKey(*address.Volume, *address.Bucket, *address.Key, os.Stdout)
						if err != nil {
							return err
						}

						return nil
					},
				},
				{
					Name:    "put",
					Aliases: []string{"p"},
					Usage:   "Put file to Ozone",
					Action: func(c *cli.Context) error {
						ozoneClient := api.CreateOzoneClient(c.GlobalString("om"))
						address := OzoneObjectAddressFromString(c.Args().Get(0))
						f, err := os.Open(c.Args().Get(1))
						if err != nil {
							return err
						}
						_, err = ozoneClient.PutKey(*address.Volume, *address.Bucket, *address.Key, f)
						if err != nil {
							return err
						}

						return nil
					},
				},
			},
		},
	}
	err := app.Run(os.Args)
	if err != nil {
		panic(err)
	}
}

type OzoneObjectAddress struct {
	Volume *string
	Bucket *string
	Key    *string
}

func OzoneObjectAddressFromString(get string) OzoneObjectAddress {
	volumeBucketKey := strings.SplitN(get, "/", 3)
	o := OzoneObjectAddress{Volume: &volumeBucketKey[0]}
	if len(volumeBucketKey) > 1 {
		o.Bucket = &volumeBucketKey[1]
	}
	if len(volumeBucketKey) > 2 {
		o.Key = &volumeBucketKey[2]
	}
	return o

}