Lannamokia_vhdmount/VHDSelectServer/server.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
VHD选择服务器 - Python版本
跨平台的VHD关键词管理服务器，带有Web GUI界面
"""

import json
import os
import sys
from datetime import datetime
from http.server import HTTPServer, BaseHTTPRequestHandler
from urllib.parse import urlparse, parse_qs
import threading
import webbrowser
from pathlib import Path

class VHDSelectHandler(BaseHTTPRequestHandler):
    def __init__(self, *args, **kwargs):
        self.config_file = Path(__file__).parent / 'vhd-config.json'
        super().__init__(*args, **kwargs)
    
    def load_config(self):
        """加载VHD配置"""
        try:
            if self.config_file.exists():
                with open(self.config_file, 'r', encoding='utf-8') as f:
                    config = json.load(f)
                    return config.get('vhdKeyword', 'SDEZ')
        except Exception as e:
            print(f"配置加载失败: {e}")
        return 'SDEZ'
    
    def save_config(self, vhd_keyword):
        """保存VHD配置"""
        try:
            config = {'vhdKeyword': vhd_keyword}
            with open(self.config_file, 'w', encoding='utf-8') as f:
                json.dump(config, f, ensure_ascii=False, indent=2)
            return True
        except Exception as e:
            print(f"配置保存失败: {e}")
            return False
    
    def send_json_response(self, data, status_code=200):
        """发送JSON响应"""
        self.send_response(status_code)
        self.send_header('Content-Type', 'application/json; charset=utf-8')
        self.send_header('Access-Control-Allow-Origin', '*')
        self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
        self.send_header('Access-Control-Allow-Headers', 'Content-Type')
        self.end_headers()
        
        json_str = json.dumps(data, ensure_ascii=False, indent=2)
        self.wfile.write(json_str.encode('utf-8'))
    
    def send_html_file(self, file_path):
        """发送HTML文件"""
        try:
            with open(file_path, 'r', encoding='utf-8') as f:
                content = f.read()
            
            self.send_response(200)
            self.send_header('Content-Type', 'text/html; charset=utf-8')
            self.end_headers()
            self.wfile.write(content.encode('utf-8'))
        except FileNotFoundError:
            self.send_error(404, '文件未找到')
        except Exception as e:
            self.send_error(500, f'服务器错误: {e}')
    
    def do_OPTIONS(self):
        """处理CORS预检请求"""
        self.send_response(200)
        self.send_header('Access-Control-Allow-Origin', '*')
        self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
        self.send_header('Access-Control-Allow-Headers', 'Content-Type')
        self.end_headers()
    
    def do_GET(self):
        """处理GET请求"""
        parsed_path = urlparse(self.path)
        path = parsed_path.path
        
        print(f"[{datetime.now().isoformat()}] GET {path}")
        
        if path == '/':
            # 主页
            html_file = Path(__file__).parent / 'public' / 'index.html'
            self.send_html_file(html_file)
            
        elif path == '/api/boot-image-select':
            # 获取当前VHD关键词
            current_keyword = self.load_config()
            response = {
                'success': True,
                'BootImageSelected': current_keyword,
                'timestamp': datetime.now().isoformat()
            }
            self.send_json_response(response)
            
        elif path == '/api/status':
            # 服务器状态
            current_keyword = self.load_config()
            response = {
                'success': True,
                'status': 'running',
                'BootImageSelected': current_keyword,
                'timestamp': datetime.now().isoformat(),
                'version': '1.0.0 (Python)'
            }
            self.send_json_response(response)
            
        else:
            # 404错误
            response = {
                'success': False,
                'error': '页面未找到'
            }
            self.send_json_response(response, 404)
    
    def do_POST(self):
        """处理POST请求"""
        parsed_path = urlparse(self.path)
        path = parsed_path.path
        
        print(f"[{datetime.now().isoformat()}] POST {path}")
        
        if path == '/api/set-vhd':
            try:
                # 读取请求体
                content_length = int(self.headers.get('Content-Length', 0))
                post_data = self.rfile.read(content_length)
                
                # 解析JSON数据
                data = json.loads(post_data.decode('utf-8'))
                vhd_keyword = data.get('BootImageSelected', '').strip().upper()
                
                if not vhd_keyword:
                    response = {
                        'success': False,
                        'error': 'VHD关键词不能为空'
                    }
                    self.send_json_response(response, 400)
                    return
                
                # 保存配置
                if self.save_config(vhd_keyword):
                    response = {
                        'success': True,
                        'BootImageSelected': vhd_keyword,
                        'message': 'VHD关键词更新成功'
                    }
                    print(f"VHD关键词已更新为: {vhd_keyword}")
                    self.send_json_response(response)
                else:
                    response = {
                        'success': False,
                        'error': '配置保存失败'
                    }
                    self.send_json_response(response, 500)
                    
            except json.JSONDecodeError:
                response = {
                    'success': False,
                    'error': '无效的JSON数据'
                }
                self.send_json_response(response, 400)
            except Exception as e:
                response = {
                    'success': False,
                    'error': f'服务器错误: {str(e)}'
                }
                self.send_json_response(response, 500)
        else:
            # 404错误
            response = {
                'success': False,
                'error': '页面未找到'
            }
            self.send_json_response(response, 404)
    
    def log_message(self, format, *args):
        """自定义日志格式"""
        pass  # 禁用默认日志，使用自定义日志

def start_server(port=8080):
    """启动服务器"""
    server_address = ('', port)
    httpd = HTTPServer(server_address, VHDSelectHandler)
    
    print('=' * 50)
    print('🚀 VHD选择服务器已启动 (Python版本)')
    print(f'📡 服务器地址: http://localhost:{port}')
    print(f'🔧 API地址: http://localhost:{port}/api/boot-image-select')
    print(f'📊 状态页面: http://localhost:{port}/api/status')
    
    # 加载当前配置
    config_file = Path(__file__).parent / 'vhd-config.json'
    try:
        if config_file.exists():
            with open(config_file, 'r', encoding='utf-8') as f:
                config = json.load(f)
                current_keyword = config.get('vhdKeyword', 'SDEZ')
        else:
            current_keyword = 'SDEZ'
    except:
        current_keyword = 'SDEZ'
    
    print(f'🎯 当前VHD关键词: {current_keyword}')
    print('=' * 50)
    print('💡 提示: 按 Ctrl+C 停止服务器')
    print()
    
    # 自动打开浏览器
    def open_browser():
        import time
        time.sleep(1)  # 等待服务器启动
        try:
            webbrowser.open(f'http://localhost:{port}')
        except:
            pass
    
    browser_thread = threading.Thread(target=open_browser)
    browser_thread.daemon = True
    browser_thread.start()
    
    try:
        httpd.serve_forever()
    except KeyboardInterrupt:
        print('\n正在关闭服务器...')
        httpd.shutdown()
        print('服务器已关闭')

if __name__ == '__main__':
    # 检查命令行参数
    port = 8080
    if len(sys.argv) > 1:
        try:
            port = int(sys.argv[1])
        except ValueError:
            print('错误: 端口号必须是数字')
            sys.exit(1)
    
    start_server(port)