| import os
|
| import subprocess
|
| import time
|
| import uuid
|
| from dataclasses import dataclass
|
|
|
| from openhands.core.logger import openhands_logger as logger
|
| from openhands.runtime.plugins.requirement import Plugin, PluginRequirement
|
| from openhands.runtime.utils.system import check_port_available
|
| from openhands.utils.shutdown_listener import should_continue
|
|
|
|
|
| @dataclass
|
| class VSCodeRequirement(PluginRequirement):
|
| name: str = 'vscode'
|
|
|
|
|
| class VSCodePlugin(Plugin):
|
| name: str = 'vscode'
|
|
|
| async def initialize(self, username: str):
|
| self.vscode_port = int(os.environ['VSCODE_PORT'])
|
| self.vscode_connection_token = str(uuid.uuid4())
|
| assert check_port_available(self.vscode_port)
|
| cmd = (
|
| f"su - {username} -s /bin/bash << 'EOF'\n"
|
| f'sudo chown -R {username}:{username} /openhands/.openvscode-server\n'
|
| 'cd /workspace\n'
|
| f'exec /openhands/.openvscode-server/bin/openvscode-server --host 0.0.0.0 --connection-token {self.vscode_connection_token} --port {self.vscode_port}\n'
|
| 'EOF'
|
| )
|
| print(cmd)
|
| self.gateway_process = subprocess.Popen(
|
| cmd,
|
| stderr=subprocess.STDOUT,
|
| shell=True,
|
| )
|
|
|
| output = ''
|
| while should_continue() and self.gateway_process.stdout is not None:
|
| line = self.gateway_process.stdout.readline().decode('utf-8')
|
| print(line)
|
| output += line
|
| if 'at' in line:
|
| break
|
| time.sleep(1)
|
| logger.debug('Waiting for VSCode server to start...')
|
|
|
| logger.debug(
|
| f'VSCode server started at port {self.vscode_port}. Output: {output}'
|
| )
|
|
|
