feat: restructure Vimana to use ~/.vimana directory layout

- Move framework to ~/.vimana/repo (VIMANA_PATH)
- Migrate plugin cache from ~/vimana to ~/.vimana/cache
- Update install scripts for new directory structure
- Add backward compatibility for existing installations
- Enable lab discovery at ~/.vimana/repo/siddhis/*/lab/^X

Author: s4dhulabs

core/labs/vmnf_labs.py

@@ -40,7 +40,7 @@
 import io
 import os
 
-vimana_path = os.getenv("vimana_path")
+vimana_path = os.getenv("VIMANA_PATH") or os.getenv("vimana_path")
 
 class xc_handler:
     def __init__(self, vmnf_handler:dict) -> None:

core/navi/collections/exceptions.py

@@ -178,6 +178,10 @@ def parse_args(self):
         run_cmd.add_argument('--workflow', action='store', nargs='?', default=False, dest='workflow')
         run_cmd.add_argument('--channel', action='store', nargs='?', default=False, dest='cmd_channel')
         run_cmd.add_argument('--ci-mode', action='store_true', dest='ci_mode')
+        run_cmd.add_argument('--lab', action='store', dest='run_lab', help='Run a lab for the specified plugin')
+        run_cmd.add_argument('--lab-operation', action='store', dest='lab_operation', 
+                           choices=['start', 'stop', 'status', 'list', 'cleanup'], 
+                           default='start', help='Lab operation to perform (default: start)')
         #run_cmd.add_argument('--cmd', action='store', nargs='?', default=False, dest='cmd')
         #run_cmd.add_argument('--pycode', action='store', nargs='?', default=False, dest='pycode')
 
@@ -333,6 +337,8 @@ def start_handler(self):
             module_run      = False,
             siddhi_run      = False,
             plugin_run      = False,
+            run_lab         = False,
+            lab_operation   = False,
             external_disabled = False,
             module_list     = False,
             list_payloads   = False,

core/vmnf_arg_parser.py

@@ -187,6 +187,11 @@ def abduct():
 
                 handler_ns.start_resource.append(arg)
 
+    if handler_ns.run_lab:
+        from core.labs.vmnf_labs import handle_lab_command
+        handle_lab_command(handler_ns)
+        return True
+
     # ~ new in vimana 1.0 - run plugin by name directly
     if hasattr(handler_ns, 'plugin_name') and handler_ns.plugin_name:
 
@@ -301,7 +306,12 @@ def abduct():
 
     #~ list all channels
     elif handler_ns.list_channels:
-        from core.vmnf_channels import list_channels, get_channels_by_type, get_channels_by_plugin, get_channels_by_target
+        from core.vmnf_channels import (
+            list_channels, 
+            get_channels_by_type, 
+            get_channels_by_plugin, 
+            get_channels_by_target
+        )
         from neotermcolor import colored
 
         # Check if summary mode is requested

core/vmnf_engine.py

@@ -32,7 +32,7 @@
 from urllib.parse import urlparse, urljoin
 from datetime import datetime
 
-vimana_path = os.getenv("vimana_path")
+vimana_path = os.getenv("VIMANA_PATH") or os.getenv("vimana_path")
 
 
 class naviSessions:

core/vmnf_navi_sessions.py

@@ -40,7 +40,7 @@
 import io
 import os
 
-vimana_path = os.getenv("vimana_path")
+vimana_path = os.getenv("VIMANA_PATH") or os.getenv("vimana_path")
 
 class naviScan:
     def __init__(self, vmnf_handler:dict) -> None:

core/vmnf_scan_tools.py

@@ -6,37 +6,47 @@
 # One-liner installer for Vimana Framework
 # Usage: curl -s https://raw.githubusercontent.com/s4dhulabs/vimana-framework/develop/scripts/install | bash
 
-echo "🚀 Installing Vimana Framework to ~/vimana-framework..."
+echo "🚀 Installing Vimana Framework to ~/.vimana/repo..."
 
-# Change to home directory
-cd ~
+# Create .vimana directory structure
+VIMANA_HOME="$HOME/.vimana"
+VIMANA_REPO="$VIMANA_HOME/repo"
+
+echo "📁 Creating Vimana directory structure..."
+mkdir -p "$VIMANA_HOME"/{cache,logs,outputs}
 
 # Remove existing installation if it exists
-if [ -d "vimana-framework" ]; then
+if [ -d "$VIMANA_REPO" ]; then
     echo "🗑️  Removing existing installation..."
-    rm -rf vimana-framework
+    rm -rf "$VIMANA_REPO"
 fi
 
-# Clone the repository to home directory
-if ! git clone -b develop https://github.com/s4dhulabs/vimana-framework.git; then
+# Clone the repository directly to ~/.vimana/repo
+if ! git clone -b develop https://github.com/s4dhulabs/vimana-framework.git "$VIMANA_REPO"; then
     echo "❌ Failed to clone repository"
     exit 1
 fi
 
 # Change to directory and run setup
-cd vimana-framework
+cd "$VIMANA_REPO"
 
 # Make the script executable and source it
 chmod +x scripts/setup-uv
 source scripts/setup-uv
 
 # Create symlink for vfe (Vimana Framework Environment)
 echo "🔗 Creating vfe symlink for easy access..."
-sudo ln -sf "$PWD/scripts/vfe" /usr/bin/vfe
+sudo ln -sf "$VIMANA_REPO/scripts/vfe" /usr/bin/vfe
+
+# Set environment variables
+echo "🔧 Setting environment variables..."
+export VIMANA_HOME="$HOME/.vimana"
+export VIMANA_PATH="$VIMANA_HOME/repo"
 
 echo ""
 echo "🎉 Vimana Framework installation complete!"
-echo "📁 Installed to: ~/vimana-framework"
+echo "📁 Installed to: ~/.vimana/repo"
+echo "📁 Data directory: ~/.vimana"
 echo ""
 echo "🚀 Quick start:"
 echo "   source vfe        # Activate environment from anywhere"

scripts/install

@@ -24,6 +24,14 @@ fi
 echo "🔧 Creating virtual environment..."
 uv sync
 
+# Set environment variables
+echo "🔧 Setting environment variables..."
+export VIMANA_HOME="${VIMANA_HOME:-$HOME/.vimana}"
+export VIMANA_PATH="${VIMANA_PATH:-$PWD}"
+
+# Create .vimana directory structure if it doesn't exist
+mkdir -p "$VIMANA_HOME"/{cache,logs,outputs}
+
 # Create symlinks
 echo "🔗 Creating symlinks..."
 sudo ln -sf $PWD/vimana.py /usr/bin/vimana
@@ -54,6 +62,8 @@ echo ""
 echo "🎊 Vimana Framework Setup completed successfully!"
 echo "💡 Run 'vimana' or 'vf' to start the framework"
 echo "🌐 Virtual environment: $PWD/.venv (in project directory)"
+echo "🏠 VIMANA_HOME: $VIMANA_HOME"
+echo "📁 VIMANA_PATH: $VIMANA_PATH"
 echo "💡 Use 'uv run vf' to run with proper environment"
 echo ""
 

scripts/setup-uv

@@ -103,7 +103,7 @@ def __init__(self, *args,**handler):
 
         issue_type = 'dast/output'
         plugin_scope = f'django/{issue_type}'
-        self.cache_dir = f"vimana/__cache__/{plugin_scope}"
+        self.cache_dir = f".vimana/cache/{plugin_scope}"
         self.abs_cache_path = os.path.join(os.path.expanduser("~"), self.cache_dir)
 
         self.agressive_mode = self.handler.get('agressive_mode')

siddhis/d4m8/engines/_d4m8_engine.py

@@ -49,7 +49,7 @@
 import io
 import os
 
-vimana_path = os.getenv("vimana_path")
+vimana_path = os.getenv("VIMANA_PATH") or os.getenv("vimana_path")
 
 class navi_handler:
     def __init__(self, vmnf_handler:dict) -> None: