feat(size): add size check for environment or globally

Author: jacopobonomi

README.md

@@ -14,6 +14,7 @@ A powerful CLI tool for managing Python virtual environments with ease.
 - Clean cache and temporary files
 - Smart environment activation
 - Shell completion for bash, zsh, fish, and powershell
+- Size check for environments
 
 | Feature | venv-manager | virtualenv | pyenv-virtualenv | Poetry | Pipenv |
 |---------|-------------|------------|-----------------|--------|--------|
@@ -34,8 +35,8 @@ curl -sSL https://raw.githubusercontent.com/jacopobonomi/venv_manager/main/insta
 
 ```bash
 # Clone repository
-git clone https://github.com/jacopobonomi/venv-manager
-cd venv-manager
+git clone https://github.com/jacopobonomi/venv_manager
+cd venv_manager
 
 # Install 
 make install
@@ -59,6 +60,12 @@ venv-manager packages myenv
 # Upgrade all packages
 venv-manager upgrade myenv
 
+# Check environment size
+venv-manager size myenv
+
+# Check all environments size
+venv-manager size --global
+
 # Global operations
 venv-manager --global clean
 
@@ -80,11 +87,13 @@ source <(venv-manager completion bash)
 | `install <n> <reqs>` | Install requirements |
 | `upgrade <n>` | Upgrade packages |
 | `clean <n>` | Clean cache files |
+| `size <n>` | Check environment size |
 | `completion [bash|zsh|fish|powershell]` | Generate shell completion scripts |
 
 ## Development 🛠️
 
 Requirements:
+
 - Go 1.21+
 - Python 3.x
 

cmd/venv-manager/main.go

@@ -6,6 +6,7 @@ import (
 	"path/filepath"
 
 	"github.com/jacopobonomi/venv-manager/internal/manager"
+	"github.com/jacopobonomi/venv-manager/internal/utils"
 	"github.com/spf13/cobra"
 )
 
@@ -42,6 +43,7 @@ func init() {
 	rootCmd.AddCommand(cleanCmd())
 	rootCmd.AddCommand(activateCmd())
 	rootCmd.AddCommand(deactivateCmd())
+	rootCmd.AddCommand(sizeCmd())
 	rootCmd.AddCommand(completionCmd())
 }
 
@@ -266,6 +268,41 @@ func deactivateCmd() *cobra.Command {
 	return cmd
 }
 
+func sizeCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "size [name]",
+		Short: "Check the size of a virtual environment",
+		Run: func(cmd *cobra.Command, args []string) {
+			mgr.SetGlobal(globalFlag)
+			name := ""
+			if len(args) > 0 && !globalFlag {
+				name = args[0]
+			}
+			
+			// Display appropriate header
+			if name != "" {
+				fmt.Printf("%s📊 Size of virtual environment '%s':%s\n", colorYellow, name, colorReset)
+			} else if globalFlag {
+				fmt.Printf("%s📊 Sizes of all virtual environments:%s\n", colorYellow, colorReset)
+			}
+			
+			// Get sizes from manager
+			sizes, err := mgr.GetSize(name)
+			if err != nil {
+				fmt.Fprintf(os.Stderr, "%s%v%s\n", colorRed, err, colorReset)
+				os.Exit(1)
+			}
+			
+			// Display results
+			for venvName, size := range sizes {
+				sizeStr := utils.FormatSize(size)
+				fmt.Printf("- %s: %s\n", venvName, sizeStr)
+			}
+		},
+	}
+	return cmd
+}
+
 func main() {
 	if err := rootCmd.Execute(); err != nil {
 		fmt.Fprintf(os.Stderr, "%s%v%s\n", colorRed, err, colorReset)

internal/manager/manager.go

@@ -301,3 +301,38 @@ func (m *Manager) ListPackages(name string) ([]string, error) {
 	}
 	return result, nil
 }
+
+func (m *Manager) GetSize(name string) (map[string]int64, error) {
+	sizes := make(map[string]int64)
+	
+	if name != "" {
+		venvPath := filepath.Join(m.baseDir, name)
+		if !m.fs.Exists(venvPath) {
+			return nil, fmt.Errorf("venv '%s' does not exist", name)
+		}
+		
+		size, err := m.fs.GetDirSize(venvPath)
+		if err != nil {
+			return nil, err
+		}
+		sizes[name] = size
+	} else if m.global {
+		venvs, err := m.List()
+		if err != nil {
+			return nil, err
+		}
+		
+		for _, venv := range venvs {
+			venvPath := filepath.Join(m.baseDir, venv)
+			size, err := m.fs.GetDirSize(venvPath)
+			if err != nil {
+				return nil, err
+			}
+			sizes[venv] = size
+		}
+	} else {
+		return nil, fmt.Errorf("please specify a venv name or use --global flag")
+	}
+	
+	return sizes, nil
+}

internal/utils/filesystem.go

@@ -1,7 +1,9 @@
 package utils
 
 import (
+	"fmt"
 	"os"
+	"path/filepath"
 )
 
 type FileSystem interface {
@@ -10,6 +12,7 @@ type FileSystem interface {
 	Exists(path string) bool
 	IsDir(path string) bool
 	ReadDir(path string) ([]os.DirEntry, error)
+	GetDirSize(path string) (int64, error)
 }
 
 type RealFileSystem struct{}
@@ -39,3 +42,34 @@ func (fs *RealFileSystem) IsDir(path string) bool {
 func (fs *RealFileSystem) ReadDir(path string) ([]os.DirEntry, error) {
 	return os.ReadDir(path)
 }
+
+func (fs *RealFileSystem) GetDirSize(path string) (int64, error) {
+	var size int64
+	err := filepath.Walk(path, func(_ string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if !info.IsDir() {
+			size += info.Size()
+		}
+		return nil
+	})
+	return size, err
+}
+
+func FormatSize(bytes int64) string {
+	const (
+		KB = 1024
+		MB = 1024 * KB
+		GB = 1024 * MB
+	)
+
+	if bytes < KB {
+		return fmt.Sprintf("%d B", bytes)
+	} else if bytes < MB {
+		return fmt.Sprintf("%.2f KB", float64(bytes)/KB)
+	} else if bytes < GB {
+		return fmt.Sprintf("%.2f MB", float64(bytes)/MB)
+	}
+	return fmt.Sprintf("%.2f GB", float64(bytes)/GB)
+}