dify-ee-helm-chart-values-generator/version_manager.py

"""Version management for Dify EE"""

import sys
from typing import Dict, Any, Optional

from utils import Colors, print_header, print_info, print_success, print_error, print_warning
from i18n import get_translator

_t = get_translator()


class VersionManager:
    """Version manager - manages Dify EE configuration modules for different versions"""

    # Version configuration: defines modules supported by each version
    # Uses major version format (3.x, 2.x) to avoid confusion with specific chart versions (e.g., 3.5.6)
    VERSION_CONFIGS = {
        "3.x": {
            "name": "Dify Enterprise Edition 3.x",
            "modules": [
                "global",
                "infrastructure",
                "networking",
                "mail",
                "plugins",  # Version 3.x supports plugin module
                "services"
            ],
            "description": "Full version with plugin support"
        },
        "2.x": {
            "name": "Dify Enterprise Edition 2.x",
            "modules": [
                "global",
                "infrastructure",
                "networking",
                "mail",
                # Version 2.x does not support plugin module
                "services"
            ],
            "description": "Version without plugin support"
        },
        # More version configurations can be added here (e.g., "4.x")
    }

    @classmethod
    def get_available_versions(cls) -> list:
        """Get list of available versions"""
        return list(cls.VERSION_CONFIGS.keys())

    @classmethod
    def get_version_info(cls, version: str) -> Optional[Dict[str, Any]]:
        """Get version information"""
        return cls.VERSION_CONFIGS.get(version)

    @classmethod
    def get_version_modules(cls, version: str) -> list:
        """Get list of modules supported by version"""
        config = cls.get_version_info(version)
        if config:
            return config.get("modules", [])
        return []

    @classmethod
    def is_module_supported(cls, version: str, module: str) -> bool:
        """Check if version supports a module"""
        modules = cls.get_version_modules(version)
        return module in modules

    @classmethod
    def prompt_version_selection(cls) -> str:
        """Interactive version selection"""
        print_header(_t('select_dify_version'))
        print_info(_t('select_version_prompt'))
        print()

        versions = cls.get_available_versions()
        version_options = []

        for i, version in enumerate(versions, 1):
            config = cls.get_version_info(version)
            name = config.get("name", f"Version {version}")
            desc = config.get("description", "")
            modules = config.get("modules", [])

            print(f"  {i}. {name}")
            print(f"     {_t('version')}: {version}")
            if desc:
                print(f"     {_t('description')}: {desc}")
            print(f"     {_t('supported_modules')}: {', '.join(modules)}")
            print()

            version_options.append(version)

        while True:
            try:
                default_text = _t('default')
                choice = input(f"{Colors.BOLD}{_t('select_version_range')} [1-{len(versions)}] ({default_text}: 1): {Colors.ENDC}").strip()

                if not choice:
                    choice = "1"

                idx = int(choice) - 1
                if 0 <= idx < len(versions):
                    selected_version = version_options[idx]
                    config = cls.get_version_info(selected_version)
                    print_success(f"{_t('selected')}: {config.get('name', selected_version)}")
                    return selected_version
                else:
                    range_text = _t('enter_number_range')
                    print_error(f"{_t('invalid_selection')} {range_text} 1-{len(versions)}")
            except ValueError:
                print_error(_t('enter_valid_number'))
            except KeyboardInterrupt:
                print("\n")
                print_warning(_t('user_interrupted'))
                sys.exit(0)

    @classmethod
    def detect_version_from_values(cls, values: Dict[str, Any]) -> Optional[str]:
        """Detect version from values.yaml if possible"""
        # Try to detect version from Chart.yaml or other sources
        # This can be implemented based on actual requirements
        # For now, return None to let user manually select
        return None

    @classmethod
    def map_chart_version_to_ee_version(cls, chart_version: Optional[str]) -> Optional[str]:
        """
        Map Helm Chart version to Dify EE version

        Mapping rules:
        - Chart version 3.x.x -> Dify EE 3.x
        - Chart version 2.x.x -> Dify EE 2.x
        - Chart version 1.x.x -> Dify EE 2.x (legacy)
        - Chart version 4.x.x -> Dify EE 4.x (future)
        - Unknown or None -> None (requires manual selection)

        Args:
            chart_version: Helm Chart version string (e.g., "3.5.6", "3.6.0-beta.1")

        Returns:
            Dify EE version string (e.g., "3.x", "2.x") or None if cannot be determined
        """
        if not chart_version:
            return None

        # Extract major version number
        try:
            # Handle versions like "3.5.6", "3.6.0-beta.1", etc.
            major_version = int(chart_version.split('.')[0])

            # Map major version to EE version (using x format for extensibility)
            # Future versions (4.x, 5.x, etc.) will automatically map correctly
            if major_version >= 4:
                return f"{major_version}.x"
            elif major_version >= 3:
                return "3.x"
            elif major_version >= 2:
                return "2.x"
            elif major_version >= 1:
                # Chart version 1.x maps to EE 2.x (legacy support)
                return "2.x"
            else:
                return None
        except (ValueError, IndexError):
            # If version format is unexpected, return None
            return None