85 lines
3.2 KiB
Python
85 lines
3.2 KiB
Python
import os
|
||
import shutil
|
||
import subprocess
|
||
from functools import cached_property
|
||
from typing import Union
|
||
|
||
|
||
class VpnCommand:
|
||
@cached_property
|
||
def install_dir(self) -> str:
|
||
return os.path.dirname(__file__).rstrip("/vpn")
|
||
|
||
@cached_property
|
||
def install_file(self) -> str:
|
||
return self.install_dir + "/bin/openvpn-install.sh"
|
||
|
||
def get_vpn_status(self, file_path="/etc/openvpn/server/server.conf") -> str:
|
||
if os.path.exists(file_path):
|
||
return "OpenVPN has been installed"
|
||
return f"Please install OpenVPN: {self.install_file}"
|
||
|
||
def install_vpn(self) -> str:
|
||
return self.create_client("init")
|
||
|
||
def create_client(self, name) -> str:
|
||
result = self.install_cmd(name, "1")
|
||
if isinstance(result, str):
|
||
return result
|
||
if result:
|
||
if not name.startswith('/'):
|
||
new_client = f"/root/{name.strip('.ovpn')}.ovpn"
|
||
else:
|
||
new_client = name
|
||
# move vpn file
|
||
if os.path.exists(new_client):
|
||
shutil.copy(new_client, self.install_dir)
|
||
return name + " has been created"
|
||
return name + " failed to create"
|
||
|
||
def revoke_client(self, name) -> str:
|
||
result = self.install_cmd(name, "2")
|
||
if isinstance(result, str):
|
||
return result
|
||
if result:
|
||
new_client = f"{name.strip('.ovpn')}.ovpn"
|
||
# move vpn file
|
||
if os.path.exists(new_client):
|
||
shutil.rmtree(self.install_dir + new_client)
|
||
return name + " has been revoked"
|
||
return name + " failed to revoke"
|
||
|
||
def install_cmd(self, name: str, option: str,
|
||
file_path="/etc/openvpn/server/server.conf") -> Union[bool, str]:
|
||
if (self.get_vpn_status(file_path) != "OpenVPN has been installed"
|
||
and name != "init"):
|
||
return "Please install OpenVPN"
|
||
|
||
name = name.strip('.ovpn')
|
||
new_client = f"~/{name}.ovpn"
|
||
# 创建一个Popen对象，指定shell脚本路径，并设置stdin参数为subprocess.PIPE
|
||
# 这样我们就可以通过stdin管道向脚本发送输入
|
||
process = subprocess.Popen(
|
||
['bash', self.install_file], stdin=subprocess.PIPE, stdout=subprocess.PIPE,
|
||
stderr=subprocess.PIPE, text=True)
|
||
# 示例交互式输入列表
|
||
input_line = f"{option}\n{name}\n"
|
||
process.stdin.write(input_line)
|
||
process.stdin.flush() # 刷新缓冲区，确保输入立即发送到子进程
|
||
print(input_line, self.install_file, new_client)
|
||
# 等待脚本执行完成并获取stdout/stderr输出
|
||
try:
|
||
stdout_data, stderr_data = process.communicate(timeout=5)
|
||
print(stdout_data, stderr_data)
|
||
except Exception as e:
|
||
with open('a.txt', "w") as f:
|
||
f.write(str(vars(process)))
|
||
f.write(str(vars(e)))
|
||
return False
|
||
# 输出脚本执行结果和状态
|
||
print("shell output: ", new_client, stdout_data, stderr_data,
|
||
process.returncode, process.stderr)
|
||
if self.get_vpn_status(file_path) != "OpenVPN has been installed":
|
||
return stdout_data
|
||
return True
|