mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2026-04-23 20:10:39 +00:00
MacRimi
297 lines
13 KiB
Python
297 lines
13 KiB
Python
from flask import Blueprint, jsonify, request
|
|
import json
|
|
import os
|
|
import re
|
|
|
|
proxmenux_bp = Blueprint('proxmenux', __name__)
|
|
|
|
# Tool metadata: description, function name in bash script, and version
|
|
# version: current version of the optimization function
|
|
# function: the bash function name that implements this optimization
|
|
TOOL_METADATA = {
|
|
'subscription_banner': {'name': 'Subscription Banner Removal', 'function': 'remove_subscription_banner', 'version': '1.0'},
|
|
'time_sync': {'name': 'Time Synchronization', 'function': 'configure_time_sync', 'version': '1.0'},
|
|
'apt_languages': {'name': 'APT Language Skip', 'function': 'skip_apt_languages', 'version': '1.0'},
|
|
'journald': {'name': 'Journald Optimization', 'function': 'optimize_journald', 'version': '1.1'},
|
|
'logrotate': {'name': 'Logrotate Optimization', 'function': 'optimize_logrotate', 'version': '1.1'},
|
|
'system_limits': {'name': 'System Limits Increase', 'function': 'increase_system_limits', 'version': '1.1'},
|
|
# entropy removed — modern kernels 5.6+ have built-in entropy generation, haveged no longer needed
|
|
'memory_settings': {'name': 'Memory Settings Optimization', 'function': 'optimize_memory_settings', 'version': '1.1'},
|
|
'kernel_panic': {'name': 'Kernel Panic Configuration', 'function': 'configure_kernel_panic', 'version': '1.0'},
|
|
'apt_ipv4': {'name': 'APT IPv4 Force', 'function': 'force_apt_ipv4', 'version': '1.0'},
|
|
'kexec': {'name': 'kexec for quick reboots', 'function': 'enable_kexec', 'version': '1.0'},
|
|
'network_optimization': {'name': 'Network Optimizations', 'function': 'apply_network_optimizations', 'version': '1.0'},
|
|
'bashrc_custom': {'name': 'Bashrc Customization', 'function': 'customize_bashrc', 'version': '1.0'},
|
|
'figurine': {'name': 'Figurine', 'function': 'configure_figurine', 'version': '1.0'},
|
|
'fastfetch': {'name': 'Fastfetch', 'function': 'configure_fastfetch', 'version': '1.0'},
|
|
'log2ram': {'name': 'Log2ram (SSD Protection)', 'function': 'configure_log2ram', 'version': '1.0'},
|
|
'amd_fixes': {'name': 'AMD CPU (Ryzen/EPYC) fixes', 'function': 'apply_amd_fixes', 'version': '1.0'},
|
|
'persistent_network': {'name': 'Setting persistent network interfaces', 'function': 'setup_persistent_network', 'version': '1.0'},
|
|
'vfio_iommu': {'name': 'VFIO/IOMMU Passthrough', 'function': 'enable_vfio_iommu', 'version': '1.0'},
|
|
'lvm_repair': {'name': 'LVM PV Headers Repair', 'function': 'repair_lvm_headers', 'version': '1.0'},
|
|
'repo_cleanup': {'name': 'Repository Cleanup', 'function': 'cleanup_repos', 'version': '1.0'},
|
|
# ── Legacy / Deprecated entries ──
|
|
# These optimizations were applied by previous ProxMenux versions but are
|
|
# no longer needed or have been removed from the current scripts. We still
|
|
# expose their source code for transparency with existing users.
|
|
'entropy': {'name': 'Entropy Generation (haveged)', 'function': 'configure_entropy', 'version': '1.0', 'deprecated': True},
|
|
}
|
|
|
|
# Backward-compatible description mapping (used by get_installed_tools)
|
|
TOOL_DESCRIPTIONS = {k: v['name'] for k, v in TOOL_METADATA.items()}
|
|
|
|
# Source code preserved for deprecated/removed optimization functions.
|
|
# When a function is removed from the active bash scripts (because it's
|
|
# no longer needed, e.g. obsoleted by kernel improvements), keep its code
|
|
# here so users who installed it in the past can still inspect what ran.
|
|
DEPRECATED_SOURCES = {
|
|
'configure_entropy': {
|
|
'script': 'customizable_post_install.sh (legacy)',
|
|
'source': '''# ─────────────────────────────────────────────────────────────────
|
|
# NOTE: This optimization has been REMOVED from current ProxMenux versions.
|
|
# Modern Linux kernels (5.6+, shipped with Proxmox VE 7.x and 8.x) include
|
|
# built-in entropy generation via the Jitter RNG and CRNG, making haveged
|
|
# unnecessary. The function below is preserved here for transparency so
|
|
# users who applied it in the past can see exactly what was installed.
|
|
# New ProxMenux installations no longer include this optimization.
|
|
# ─────────────────────────────────────────────────────────────────
|
|
|
|
configure_entropy() {
|
|
msg_info2 "$(translate "Configuring entropy generation to prevent slowdowns...")"
|
|
|
|
# Install haveged
|
|
msg_info "$(translate "Installing haveged...")"
|
|
/usr/bin/env DEBIAN_FRONTEND=noninteractive apt-get -y -o Dpkg::Options::='--force-confdef' install haveged > /dev/null 2>&1
|
|
msg_ok "$(translate "haveged installed successfully")"
|
|
|
|
# Configure haveged
|
|
msg_info "$(translate "Configuring haveged...")"
|
|
cat <<EOF > /etc/default/haveged
|
|
# -w sets low entropy watermark (in bits)
|
|
DAEMON_ARGS="-w 1024"
|
|
EOF
|
|
|
|
# Reload systemd daemon
|
|
systemctl daemon-reload > /dev/null 2>&1
|
|
|
|
# Enable haveged service
|
|
systemctl enable haveged > /dev/null 2>&1
|
|
msg_ok "$(translate "haveged service enabled successfully")"
|
|
|
|
register_tool "entropy" true
|
|
msg_success "$(translate "Entropy generation configuration completed")"
|
|
}
|
|
''',
|
|
},
|
|
}
|
|
|
|
# Scripts to search for function source code (in order of preference)
|
|
_SCRIPT_PATHS = [
|
|
'/usr/local/share/proxmenux/scripts/post_install/customizable_post_install.sh',
|
|
'/usr/local/share/proxmenux/scripts/post_install/auto_post_install.sh',
|
|
]
|
|
|
|
|
|
def _extract_bash_function(function_name: str) -> dict:
|
|
"""Extract a bash function's source code.
|
|
|
|
Checks DEPRECATED_SOURCES first (for functions removed from active scripts),
|
|
then searches the live bash scripts for `function_name() {` and captures
|
|
everything until the matching closing `}`, respecting brace nesting.
|
|
|
|
Returns {'source': str, 'script': str, 'line_start': int, 'line_end': int}
|
|
or {'source': '', 'error': '...'} on failure.
|
|
"""
|
|
# Check preserved deprecated source code first
|
|
if function_name in DEPRECATED_SOURCES:
|
|
entry = DEPRECATED_SOURCES[function_name]
|
|
source = entry['source']
|
|
return {
|
|
'source': source,
|
|
'script': entry['script'],
|
|
'line_start': 1,
|
|
'line_end': len(source.split('\n')),
|
|
}
|
|
|
|
for script_path in _SCRIPT_PATHS:
|
|
if not os.path.isfile(script_path):
|
|
continue
|
|
try:
|
|
with open(script_path, 'r') as f:
|
|
lines = f.readlines()
|
|
|
|
# Find function start: "function_name() {" or "function_name () {"
|
|
pattern = re.compile(rf'^{re.escape(function_name)}\s*\(\)\s*\{{')
|
|
start_idx = None
|
|
for i, line in enumerate(lines):
|
|
if pattern.match(line):
|
|
start_idx = i
|
|
break
|
|
|
|
if start_idx is None:
|
|
continue # Try next script
|
|
|
|
# Capture until the closing } at indent level 0
|
|
brace_depth = 0
|
|
end_idx = start_idx
|
|
for i in range(start_idx, len(lines)):
|
|
brace_depth += lines[i].count('{') - lines[i].count('}')
|
|
if brace_depth <= 0:
|
|
end_idx = i
|
|
break
|
|
|
|
source = ''.join(lines[start_idx:end_idx + 1])
|
|
script_name = os.path.basename(script_path)
|
|
|
|
return {
|
|
'source': source,
|
|
'script': script_name,
|
|
'line_start': start_idx + 1,
|
|
'line_end': end_idx + 1,
|
|
}
|
|
except Exception:
|
|
continue
|
|
|
|
return {'source': '', 'error': 'Function not found in available scripts'}
|
|
|
|
@proxmenux_bp.route('/api/proxmenux/update-status', methods=['GET'])
|
|
def get_update_status():
|
|
"""Get ProxMenux update availability status from config.json"""
|
|
config_path = '/usr/local/share/proxmenux/config.json'
|
|
|
|
try:
|
|
if not os.path.exists(config_path):
|
|
return jsonify({
|
|
'success': True,
|
|
'update_available': {
|
|
'stable': False,
|
|
'stable_version': '',
|
|
'beta': False,
|
|
'beta_version': ''
|
|
}
|
|
})
|
|
|
|
with open(config_path, 'r') as f:
|
|
config = json.load(f)
|
|
|
|
update_status = config.get('update_available', {
|
|
'stable': False,
|
|
'stable_version': '',
|
|
'beta': False,
|
|
'beta_version': ''
|
|
})
|
|
|
|
return jsonify({
|
|
'success': True,
|
|
'update_available': update_status
|
|
})
|
|
|
|
except Exception as e:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': str(e)
|
|
}), 500
|
|
|
|
|
|
@proxmenux_bp.route('/api/proxmenux/installed-tools', methods=['GET'])
|
|
def get_installed_tools():
|
|
"""Get list of installed ProxMenux tools/optimizations"""
|
|
installed_tools_path = '/usr/local/share/proxmenux/installed_tools.json'
|
|
|
|
try:
|
|
if not os.path.exists(installed_tools_path):
|
|
return jsonify({
|
|
'success': True,
|
|
'installed_tools': [],
|
|
'message': 'No ProxMenux optimizations installed yet'
|
|
})
|
|
|
|
with open(installed_tools_path, 'r') as f:
|
|
data = json.load(f)
|
|
|
|
# Convert to list format with descriptions and version
|
|
tools = []
|
|
for tool_key, enabled in data.items():
|
|
if enabled: # Only include enabled tools
|
|
meta = TOOL_METADATA.get(tool_key, {})
|
|
tools.append({
|
|
'key': tool_key,
|
|
'name': meta.get('name', tool_key.replace('_', ' ').title()),
|
|
'enabled': enabled,
|
|
'version': meta.get('version', '1.0'),
|
|
'has_source': bool(meta.get('function')),
|
|
'deprecated': bool(meta.get('deprecated', False)),
|
|
})
|
|
|
|
# Sort alphabetically by name
|
|
tools.sort(key=lambda x: x['name'])
|
|
|
|
return jsonify({
|
|
'success': True,
|
|
'installed_tools': tools,
|
|
'total_count': len(tools)
|
|
})
|
|
|
|
except json.JSONDecodeError:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': 'Invalid JSON format in installed_tools.json'
|
|
}), 500
|
|
except Exception as e:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': str(e)
|
|
}), 500
|
|
|
|
|
|
@proxmenux_bp.route('/api/proxmenux/tool-source/<tool_key>', methods=['GET'])
|
|
def get_tool_source(tool_key):
|
|
"""Get the bash source code of a specific optimization function.
|
|
|
|
Returns the function body extracted from the post-install scripts,
|
|
so users can see exactly what code was executed on their server.
|
|
"""
|
|
try:
|
|
meta = TOOL_METADATA.get(tool_key)
|
|
if not meta:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': f'Unknown tool: {tool_key}'
|
|
}), 404
|
|
|
|
func_name = meta.get('function')
|
|
if not func_name:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': f'No function mapping for {tool_key}'
|
|
}), 404
|
|
|
|
result = _extract_bash_function(func_name)
|
|
|
|
if not result.get('source'):
|
|
return jsonify({
|
|
'success': False,
|
|
'error': result.get('error', 'Source code not available'),
|
|
'tool': tool_key,
|
|
'function': func_name,
|
|
}), 404
|
|
|
|
return jsonify({
|
|
'success': True,
|
|
'tool': tool_key,
|
|
'name': meta['name'],
|
|
'version': meta.get('version', '1.0'),
|
|
'deprecated': bool(meta.get('deprecated', False)),
|
|
'function': func_name,
|
|
'source': result['source'],
|
|
'script': result['script'],
|
|
'line_start': result['line_start'],
|
|
'line_end': result['line_end'],
|
|
})
|
|
|
|
except Exception as e:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': str(e)
|
|
}), 500
|