from dataclasses import dataclass, field
from typing import Type
import tyro
from loguru import logger

from deepagents.backends import LocalShellBackend

from lang_agent.fs_bkends.base import BaseFilesystemBackend, FilesystemBackendConfig


@tyro.conf.configure(tyro.conf.SuppressFixed)
@dataclass
class LocalShellConfig(FilesystemBackendConfig):
    _target:Type = field(default_factory=lambda:LocalShell)

    workspace_dir:str = "./workspace"
    """path to workspace directory"""

    skills_dir:str = "./workspace/skills"
    """path to directory containing skill files"""

    rt_skills_dir:str = "/skills"
    """path to directory with skills in runtime directory"""


class LocalShell(BaseFilesystemBackend):
    def __init__(self, config:LocalShellConfig):
        logger.warning("Caution: The LocalShell backend grants direct access to the local system shell. Improper use can pose significant security and safety risks, including unintended code execution and file access. Use this backend with extreme care.")
        self.config = config
        self._build_backend()
    
    def _build_backend(self):
        self.backend = LocalShellBackend(root_dir=self.config.workspace_dir,
                                         virtual_mode=True,
                                        #  env={"PATH": "/usr/bin:/bin"}
                                         inherit_env=True)



if __name__ == "__main__":
    import sys

    # Instantiate a LocalShell instance with the default config
    config = LocalShellConfig()
    shell = LocalShell(config)

    # Try checking access to 'npx'
    try:
        result = shell.backend.execute("npx --version")
        if result.exit_code == 0:
            print("npx is available, version:", result.output.strip())
        else:
            print("npx returned non-zero exit code:", result.exit_code, file=sys.stderr)
            print("output:", result.output, file=sys.stderr)
    except Exception as e:
        print("Could not access 'npx':", str(e), file=sys.stderr)