add storage.GetStatus()

Author: sp-yduck

proxmox/storage.go

@@ -38,6 +38,18 @@ func (s *Storage) Delete(ctx context.Context) error {
 	return s.restclient.DeleteStorage(ctx, s.Storage.Storage)
 }
 
+func (s *Storage) GetStatus(ctx context.Context) ([]*api.Storage, error) {
+	var status []*api.Storage
+	if s.Node == "" {
+		return nil, errors.New("Node must not be empty")
+	}
+	path := fmt.Sprintf("/nodes/%s/storage/%s/status", s.Node, s.Storage.Storage)
+	if err := s.restclient.Get(ctx, path, &status); err != nil {
+		return nil, err
+	}
+	return status, nil
+}
+
 func (s *Storage) GetContents(ctx context.Context) ([]*api.StorageContent, error) {
 	var contents []*api.StorageContent
 	if s.Node == "" {

rest/node.go

@@ -55,3 +55,25 @@ func (c *RESTClient) GetNodeVNCWebSocket(ctx context.Context, nodeName, port, vn
 	}
 	return websocket, nil
 }
+
+func (c *RESTClient) GetNodeStorages(ctx context.Context, nodeName string) ([]*api.Storage, error) {
+	path := fmt.Sprintf("/nodes/%s/storage", nodeName)
+	var storages []*api.Storage
+	if err := c.Get(ctx, path, &storages); err != nil {
+		return nil, err
+	}
+	return storages, nil
+}
+
+func (c *RESTClient) GetNodeStorage(ctx context.Context, nodeName, storageName string) (*api.Storage, error) {
+	storages, err := c.GetNodeStorages(ctx, nodeName)
+	if err != nil {
+		return nil, err
+	}
+	for _, s := range storages {
+		if s.Storage == storageName {
+			return s, nil
+		}
+	}
+	return nil, NotFoundErr
+}