package main 

import (

	"fmt"

	"github.com/noahdesu/go-ceph/rados"

	"github.com/noahdesu/go-ceph/rbd"

)

const (

	DefaultRadosConfigFile = "/etc/ceph/ceph.conf"

	DefaultBaseImageSize = 10 * 1024 * 1024 * 1024

	DefaultPoolName = "rbd"

)

func main() {

	// connect to the cluster

	conn,_ := rados.NewConn()

	if err := conn.ReadConfigFile(DefaultRadosConfigFile); err != nil {

		fmt.Printf("Rbd read config failed: %v",err)

		return

	}

	if err := conn.Connect(); err != nil {

		fmt.Printf("Rbd connect failed: %v",err)

		return

	}

	// connect to the pool

	ioctx,err := conn.OpenIOContext(DefaultPoolName)

	if err != nil {

		fmt.Printf("Rbd open pool failed: %v",err)

		return

	}

	// create base image

	baseImageName := "test"

	_,err = rbd.Create(ioctx,baseImageName,DefaultBaseImageSize,rbd.RbdFeatureLayering)

	if err != nil {

		fmt.Printf("Rbd create image failed: %v",err)

		return

	}

	img := rbd.GetImage(ioctx,baseImageName)

	// we should open base image first

	if err := img.Open(); err != nil {

		fmt.Printf("Rbd open image  failed: %v",err)

		return

	}

	defer img.Close()

	// create snapshot

	snapName := "test-snap"

	snapshot,err := img.CreateSnapshot(snapName)

	if err != nil {

		fmt.Printf("Rbd create snapshot failed: %v",err)

		return

	}

	// protect snapshot

	if err := snapshot.Protect(); err != nil {

		fmt.Printf("Rbd create snapshot failed: %v",err)

		return

	}

	// make a clone image based on the snap shot

	cloneImageName := "clone-test"

	_,err = img.Clone(snapName,ioctx,cloneImageName,rbd.RbdFeatureLayering)

	if err != nil {

		fmt.Printf("Rbd clone snapshot failed: %v",err)

		return

	}

	return

}