storage

Author: sp-yduck

api/storage_type.go

@@ -1,5 +1,9 @@
 package api
 
+import (
+	"encoding/json"
+)
+
 type Storage struct {
 	Active       int
 	Avail        int
@@ -26,3 +30,20 @@ type StorageCreateOptions struct {
 	Mkdir       bool   `json:"mkdir,omitempty"`
 	Path        string `json:"path,omitempty"`
 }
+
+type StorageContent struct {
+	Storage string `json:",omitempty"`
+	Content string `json:",omitempty"`
+	// to do : use custom type instead of json.Number
+	CTime     json.Number `json:",omitempty"`
+	Encrypted string
+	Format    string
+	Notes     string
+	Parent    string
+	Protected bool
+	Size      int
+	Used      int
+	// to do : Verificateion
+	VMID  int
+	VolID string `josn:"volid,omitempty"`
+}

proxmox/storage.go

@@ -2,6 +2,8 @@ package proxmox
 
 import (
 	"context"
+	"errors"
+	"fmt"
 
 	"github.com/sp-yduck/proxmox-go/api"
 	"github.com/sp-yduck/proxmox-go/rest"
@@ -10,6 +12,7 @@ import (
 type Storage struct {
 	restclient *rest.RESTClient
 	Storage    *api.Storage
+	Node       string
 }
 
 func (s *Service) Storage(ctx context.Context, name string) (*Storage, error) {
@@ -29,3 +32,32 @@ func (s *Service) CreateStorage(ctx context.Context, name, storageType string, o
 	}
 	return &Storage{restclient: s.restclient, Storage: storage}, nil
 }
+
+func (s *Storage) Delete(ctx context.Context) error {
+	return s.restclient.DeleteStorage(ctx, s.Storage.Storage)
+}
+
+func (s *Storage) GetContents(ctx context.Context) ([]*api.StorageContent, error) {
+	var contents []*api.StorageContent
+	if s.Node == "" {
+		return nil, errors.New("Node must not be empty")
+	}
+	path := fmt.Sprintf("/nodes/%s/storage/%s/content", s.Node, s.Storage.Storage)
+	if err := s.restclient.Get(ctx, path, &contents); err != nil {
+		return nil, err
+	}
+	return contents, nil
+}
+
+func (s *Storage) GetContent(ctx context.Context, volumeID string) (*api.StorageContent, error) {
+	contents, err := s.GetContents(ctx)
+	if err != nil {
+		return nil, err
+	}
+	for _, content := range contents {
+		if content.VolID == volumeID {
+			return content, nil
+		}
+	}
+	return nil, rest.NotFoundErr
+}