InPanel 1.2.1__py3-none-any.whl

inpanel/mod/ssh.py

@@ -0,0 +1,280 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2017-2026 Jackson Dou
+# All rights reserved.
+#
+# InPanel is distributed under the terms of The New BSD License.
+# The full license can be found in 'LICENSE'.
+
+'''Module for ssh Management'''
+
+import os
+import re
+import shlex
+from pathlib import Path
+
+import pexpect
+
+SSHCFG = '/etc/ssh/sshd_config'
+
+
+def loadconfig(cfgfile=None, detail=False):
+    """Read config file and parse config item to dict.
+    """
+    if not cfgfile:
+        cfgfile = SSHCFG
+
+    settings = {}
+    with open(cfgfile, encoding='utf-8') as f:
+        for line_i, line in enumerate(f):
+            line = line.strip()
+            if not line or line.startswith('# '):
+                continue
+
+            # detect if it's commented
+            if line.startswith('#'):
+                line = line.strip('#')
+                commented = True
+                if not detail:
+                    continue
+            else:
+                commented = False
+
+            fs = re.split(r'\s+', line, 1)
+            if len(fs) != 2:
+                continue
+
+            item = fs[0].strip()
+            value = fs[1].strip()
+
+            if item in settings:
+                if detail:
+                    count = settings[item]['count']+1
+                if not commented:
+                    settings[item] = detail and {
+                        'file': cfgfile,
+                        'line': line_i,
+                        'value': value,
+                        'commented': commented,
+                    } or value
+            else:
+                count = 1
+                settings[item] = detail and {
+                    'file': cfgfile,
+                    'line': line_i,
+                    'value': fs[1].strip(),
+                    'commented': commented,
+                } or value
+            if detail:
+                settings[item]['count'] = count
+
+    return settings
+
+
+def cfg_get(item, detail=False, config=None):
+    """Get value of a config item.
+    """
+    if not config:
+        config = loadconfig(detail=detail)
+    if item in config:
+        return config[item]
+    else:
+        return None
+
+
+def cfg_set(item, value, commented=False, config=None):
+    """Set value of a config item.
+    """
+    cfgfile = SSHCFG
+    v = cfg_get(item, detail=True, config=config)
+
+    if v:
+        # detect if value change
+        if v['commented'] == commented and v['value'] == value:
+            return True
+
+        # empty value should be commented
+        if value == '':
+            commented = True
+
+        # replace item in line
+        lines = []
+        with open(v['file'], encoding='utf-8') as f:
+            for line_i, line in enumerate(f):
+                if line_i == v['line']:
+                    if not v['commented']:
+                        if commented:
+                            if v['count'] > 1:
+                                # delete this line, just ignore it
+                                pass
+                            else:
+                                # comment this line
+                                lines.append('#%s %s\n' % (item, value))
+                        else:
+                            lines.append('%s %s\n' % (item, value))
+                    else:
+                        if commented:
+                            # do not allow change comment value
+                            lines.append(line)
+                            pass
+                        else:
+                            # append a new line after comment line
+                            lines.append(line)
+                            lines.append('%s %s\n' % (item, value))
+                else:
+                    lines.append(line)
+        with open(v['file'], 'w', encoding='utf-8') as f:
+            f.write(''.join(lines))
+    else:
+        # append to the end of file
+        with open(inifile, 'a', encoding='utf-8') as f:
+            f.write('\n%s%s = %s\n' % (commented and '#' or '', item, value))
+
+    return True
+
+
+def genkey(path, password=''):
+    """Generate a ssh key pair.
+    """
+    cmd = shlex.split('ssh-keygen -t rsa')
+    child = pexpect.spawn(cmd[0], cmd[1:])
+    i = child.expect(['Enter file in which to save the key', pexpect.EOF])
+    if i == 1:
+        if child.isalive():
+            child.wait()
+        return False
+
+    child.sendline(path)
+    i = child.expect(['Overwrite', 'Enter passphrase', pexpect.EOF])
+    if i == 0:
+        child.sendline('y')
+        i = child.expect(['Enter passphrase', pexpect.EOF])
+        if i == 1:
+            if child.isalive():
+                child.wait()
+            return False
+    elif i == 2:
+        if child.isalive():
+            child.wait()
+        return False
+
+    child.sendline(password)
+    i = child.expect(['Enter same passphrase', pexpect.EOF])
+    if i == 1:
+        if child.isalive():
+            child.wait()
+        return False
+
+    child.sendline(password)
+    child.expect(pexpect.EOF)
+
+    if child.isalive():
+        return child.wait() == 0
+    return True
+
+
+def chpasswd(path, oldpassword, newpassword):
+    """Change password of a private key.
+    """
+    if len(newpassword) != 0 and not len(newpassword) > 4:
+        return False
+
+    cmd = shlex.split('ssh-keygen -p')
+    child = pexpect.spawn(cmd[0], cmd[1:])
+    i = child.expect(['Enter file in which the key is', pexpect.EOF])
+    if i == 1:
+        if child.isalive():
+            child.wait()
+        return False
+
+    child.sendline(path)
+    i = child.expect(
+        ['Enter old passphrase', 'Enter new passphrase', pexpect.EOF])
+    if i == 0:
+        child.sendline(oldpassword)
+        i = child.expect(
+            ['Enter new passphrase', 'Bad passphrase', pexpect.EOF])
+        if i != 0:
+            if child.isalive():
+                child.wait()
+            return False
+    elif i == 2:
+        if child.isalive():
+            child.wait()
+        return False
+
+    child.sendline(newpassword)
+    i = child.expect(['Enter same passphrase again', pexpect.EOF])
+    if i == 1:
+        if child.isalive():
+            child.wait()
+        return False
+
+    child.sendline(newpassword)
+    child.expect(pexpect.EOF)
+
+    if child.isalive():
+        return child.wait() == 0
+    return True
+
+
+def web_handler(context):
+    action = context.get_argument('action', '')
+
+    if action == 'getsettings':
+        port = cfg_get('Port')
+        enable_pwdauth = cfg_get('PasswordAuthentication') == 'yes'
+        enable_pubkauth = cfg_get('PubkeyAuthentication') == 'yes'
+        subsystem = cfg_get('Subsystem')
+        enable_sftp = subsystem and 'sftp' in subsystem
+        pubkey_path = '/root/.ssh/sshkey_inpanel.pub'
+        prvkey_path = '/root/.ssh/sshkey_inpanel'
+        context.write({'code': 0, 'msg': '获取 SSH 服务配置信息成功！', 'data': {
+           'port': port,
+           'enable_pwdauth': enable_pwdauth,
+           'enable_pubkauth': enable_pubkauth,
+           'enable_sftp': enable_sftp,
+           'pubkey': Path(pubkey_path).is_file() and pubkey_path or '',
+           'prvkey': Path(prvkey_path).is_file() and prvkey_path or '',
+        }})
+
+    elif action == 'savesettings':
+        if context.config.get('runtime', 'mode') == 'demo':
+            context.write({'code': -1, 'msg': '演示模式不允许修改 SSH 服务设置！'})
+            return
+
+        port = context.get_argument('port', '')
+        if port:
+            cfg_set('Port', port)
+        enable_pwdauth = context.get_argument('enable_pwdauth', '')
+        if enable_pwdauth:
+            cfg_set('PasswordAuthentication', enable_pwdauth=='on' and 'yes' or 'no')
+        enable_pubkauth = context.get_argument('enable_pubkauth', '')
+        if enable_pubkauth:
+            if enable_pubkauth == 'on':
+                pubkey_path = context.get_argument('pubkey', '')
+                if not Path(pubkey_path).is_file():
+                    context.write({'code': -1, 'msg': '公钥文件不存在！'})
+                    return
+            cfg_set('PubkeyAuthentication', enable_pubkauth=='on' and 'yes' or 'no')
+            cfg_set('AuthorizedKeysFile', pubkey_path)
+
+        enable_sftp = context.get_argument('enable_sftp', '')
+        if enable_sftp:
+            cfg_set('Subsystem', 'sftp /usr/libexec/openssh/sftp-server', enable_sftp!='on')
+        context.write({'code': 0, 'msg': 'SSH 服务配置保存成功！'})
+
+
+if __name__ == '__main__':
+    import pprint
+    pp = pprint.PrettyPrinter(indent=4)
+
+    # pp.pprint(loadconfig())
+    #print cfg_get('Port')
+    #print cfg_get('Subsystem', detail=True)
+    #print cfg_set('Protocol', '2', commented=False)
+    #print cfg_set('Subsystem', 'sftp\t/usr/libexec/openssh/sftp-server', commented=True)
+
+    #print genkey('/root/.ssh/sshkey_inpanel')
+    #print chpasswd('/root/.ssh/sshkey_inpanel', '', 'aaaaaa')

inpanel/mod/system.py

@@ -0,0 +1,220 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2020-2026 Jackson Dou
+# All rights reserved.
+#
+# InPanel is distributed under the terms of The New BSD License.
+# The full license can be found in 'LICENSE'.
+'''Module for System Detection and Information'''
+
+import os
+import shutil
+from platform import mac_ver, platform, uname, win32_ver
+from typing import Dict, Optional, Tuple
+
+
+def get_os_release() -> Dict[str, str]:
+    """读取 /etc/os-release 文件，返回系统信息字典"""
+    info = {}
+    try:
+        with open("/etc/os-release", "r") as f:
+            for line in f:
+                if "=" in line:
+                    k, v = line.strip().split("=", 1)
+                    info[k] = v.strip('"')
+    except FileNotFoundError:
+        pass
+    return info
+
+
+_os_release = get_os_release()
+
+OS_ID = _os_release.get("ID", "")
+OS_ID_LIKE = _os_release.get("ID_LIKE", "")
+OS_VERSION_ID = _os_release.get("VERSION_ID", "")
+OS_PRETTY_NAME = _os_release.get("PRETTY_NAME", "")
+OS_NAME = _os_release.get("NAME", "")
+
+
+def get_os_family() -> str:
+    """获取操作系统家族：rhel 或 debian 或 darwin 或 windows"""
+    if OS_ID in ("almalinux", "rocky", "centos", "rhel", "fedora", "ol", "scientific", "centos-stream"):
+        return "rhel"
+    if OS_ID in ("debian", "ubuntu", "linuxmint", "pop", "elementary", "kali"):
+        return "debian"
+    if OS_ID_LIKE:
+        if "rhel" in OS_ID_LIKE:
+            return "rhel"
+        if "debian" in OS_ID_LIKE:
+            return "debian"
+    kernel_name, _, _, _, _, _ = uname()
+    if kernel_name == "Darwin":
+        return "darwin"
+    if kernel_name == "Windows":
+        return "windows"
+    return "unknown"
+
+
+def get_os_id() -> str:
+    """获取操作系统 ID"""
+    return OS_ID
+
+
+def get_os_version() -> str:
+    """获取操作系统版本号（完整）"""
+    return OS_VERSION_ID
+
+
+def get_os_version_major() -> int:
+    """获取操作系统主版本号（整数）"""
+    if OS_VERSION_ID:
+        try:
+            return int(OS_VERSION_ID.split(".")[0])
+        except (ValueError, IndexError):
+            pass
+    return 0
+
+
+def get_os_version_full() -> Tuple[int, int]:
+    """获取操作系统版本号（主版本，次版本）"""
+    major = 0
+    minor = 0
+    if OS_VERSION_ID:
+        parts = OS_VERSION_ID.split(".")
+        try:
+            major = int(parts[0])
+            if len(parts) > 1:
+                minor = int(parts[1])
+        except ValueError:
+            pass
+    return (major, minor)
+
+
+def get_os_title() -> str:
+    """获取操作系统全称"""
+    if OS_PRETTY_NAME:
+        return OS_PRETTY_NAME
+    if OS_NAME:
+        if OS_VERSION_ID:
+            return f"{OS_NAME} {OS_VERSION_ID}"
+        return OS_NAME
+    kernel_name, _, _, _, _, _ = uname()
+    if kernel_name == "Darwin":
+        os_version, _, _ = mac_ver()
+        return f"macOS {os_version}"
+    if kernel_name == "Windows":
+        os_version, _, _ = win32_ver()
+        return f"Windows {os_version}"
+    return "Unknown"
+
+
+def get_os_name() -> str:
+    """获取操作系统发行名称"""
+    if OS_ID:
+        return OS_ID.capitalize()
+    kernel_name, _, _, _, _, _ = uname()
+    if kernel_name == "Darwin":
+        return "macOS"
+    if kernel_name == "Windows":
+        return "Windows"
+    return "Unknown"
+
+
+def is_rhel_family() -> bool:
+    """是否为 RHEL 家族系统"""
+    return get_os_family() == "rhel"
+
+
+def is_debian_family() -> bool:
+    """是否为 Debian 家族系统"""
+    return get_os_family() == "debian"
+
+
+def is_darwin() -> bool:
+    """是否为 macOS 系统"""
+    return get_os_family() == "darwin"
+
+
+def is_windows() -> bool:
+    """是否为 Windows 系统"""
+    return get_os_family() == "windows"
+
+
+def compare_version(version_str: str) -> int:
+    """
+    比较当前系统版本与指定版本
+    返回值: -1 (小于), 0 (等于), 1 (大于)
+    """
+    current_major, current_minor = get_os_version_full()
+    parts = version_str.split(".")
+    try:
+        major = int(parts[0])
+        minor = int(parts[1]) if len(parts) > 1 else 0
+    except ValueError:
+        return 0
+    
+    if current_major > major:
+        return 1
+    elif current_major < major:
+        return -1
+    else:
+        if current_minor > minor:
+            return 1
+        elif current_minor < minor:
+            return -1
+        return 0
+
+
+def has_package_manager() -> bool:
+    """检查系统是否有包管理器"""
+    return any([
+        shutil.which("yum"),
+        shutil.which("dnf"),
+        shutil.which("apt"),
+        shutil.which("zypper"),
+        shutil.which("pacman"),
+    ])
+
+
+def get_system_info() -> Dict[str, str]:
+    """返回完整的系统信息字典"""
+    kernel_name, hostname, kernel_release, kernel_version, machine, processor = uname()
+    return {
+        'os_title': get_os_title(),
+        'os_name': get_os_name(),
+        'os_version': get_os_version(),
+        'os_versint': str(get_os_version_major()),
+        'os_platform': platform(),
+        'os_family': get_os_family(),
+        'os_id': get_os_id(),
+        'hostname': hostname,
+        'kernel_name': kernel_name,
+        'kernel_release': kernel_release,
+        'kernel_version': kernel_version,
+        'machine': machine,
+        'processor': processor,
+    }
+
+
+__all__ = [
+    'get_os_release',
+    'OS_ID',
+    'OS_ID_LIKE',
+    'OS_VERSION_ID',
+    'OS_PRETTY_NAME',
+    'OS_NAME',
+    'get_os_family',
+    'get_os_id',
+    'get_os_version',
+    'get_os_version_major',
+    'get_os_version_full',
+    'get_os_title',
+    'get_os_name',
+    'is_rhel_family',
+    'is_debian_family',
+    'is_darwin',
+    'is_windows',
+    'compare_version',
+    'has_package_manager',
+    'get_system_info',
+]