import os
import sys
import subprocess
import argparse
import json
import time
from datetime import datetime
import platform
import smtplib
from email.mime.text import MimeText
from email.mime.multipart import MimeMultipart

class ServiceStatusChecker:
    def __init__(self):
        self.os_type = platform.system().lower()
        self.services = []
        self.results = []
        self.errors = []
        
    def add_service(self, name, display_name=None, description=None):
        """添加要檢查的服務(wù)"""
        service = {
            'name': name,
            'display_name': display_name or name,
            'description': description or ''
        }
        self.services.append(service)
        
    def load_services_from_config(self, config_file):
        """從配置文件加載服務(wù)列表"""
        try:
            with open(config_file, 'r', encoding='utf-8') as f:
                config = json.load(f)
                services = config.get('services', [])
                for service in services:
                    self.add_service(
                        service['name'],
                        service.get('display_name'),
                        service.get('description')
                    )
            return True
        except Exception as e:
            self.errors.append(f"加載配置文件失敗: {e}")
            return False
            
    def check_windows_service(self, service_name):
        """檢查Windows服務(wù)狀態(tài)"""
        try:
            # 使用sc query命令檢查服務(wù)狀態(tài)
            cmd = ['sc', 'query', service_name]
            result = subprocess.run(cmd, capture_output=True, text=True, timeout=30)
            
            if result.returncode != 0:
                return {
                    'name': service_name,
                    'status': 'unknown',
                    'error': result.stderr.strip()
                }
                
            output = result.stdout.lower()
            
            if 'running' in output:
                status = 'running'
            elif 'stopped' in output:
                status = 'stopped'
            else:
                status = 'unknown'
                
            return {
                'name': service_name,
                'status': status,
                'details': output
            }
        except subprocess.TimeoutExpired:
            return {
                'name': service_name,
                'status': 'timeout',
                'error': '命令執(zhí)行超時(shí)'
            }
        except Exception as e:
            return {
                'name': service_name,
                'status': 'error',
                'error': str(e)
            }
            
    def check_linux_service(self, service_name):
        """檢查L(zhǎng)inux systemd服務(wù)狀態(tài)"""
        try:
            # 使用systemctl is-active命令檢查服務(wù)狀態(tài)
            cmd = ['systemctl', 'is-active', service_name]
            result = subprocess.run(cmd, capture_output=True, text=True, timeout=30)
            
            if result.returncode == 0:
                status = result.stdout.strip()
            else:
                # 如果命令失敗，可能是服務(wù)不存在
                status = 'inactive' if 'inactive' in result.stdout.lower() else 'unknown'
                
            # 獲取服務(wù)詳細(xì)信息
            details = {}
            try:
                cmd = ['systemctl', 'show', service_name, '--no-pager']
                detail_result = subprocess.run(cmd, capture_output=True, text=True, timeout=30)
                if detail_result.returncode == 0:
                    for line in detail_result.stdout.strip().split('\n'):
                        if '=' in line:
                            key, value = line.split('=', 1)
                            details[key] = value
            except:
                pass
                
            return {
                'name': service_name,
                'status': status,
                'details': details
            }
        except subprocess.TimeoutExpired:
            return {
                'name': service_name,
                'status': 'timeout',
                'error': '命令執(zhí)行超時(shí)'
            }
        except Exception as e:
            return {
                'name': service_name,
                'status': 'error',
                'error': str(e)
            }
            
    def check_macos_service(self, service_name):
        """檢查macOS launchd服務(wù)狀態(tài)"""
        try:
            # 使用launchctl list命令檢查服務(wù)狀態(tài)
            cmd = ['launchctl', 'list']
            result = subprocess.run(cmd, capture_output=True, text=True, timeout=30)
            
            if result.returncode != 0:
                return {
                    'name': service_name,
                    'status': 'unknown',
                    'error': result.stderr.strip()
                }
                
            # 解析輸出查找服務(wù)
            lines = result.stdout.strip().split('\n')[1:]  # 跳過標(biāo)題行
            service_found = False
            pid = None
            
            for line in lines:
                parts = line.split()
                if len(parts) >= 3 and service_name in parts[2]:
                    service_found = True
                    pid = parts[0] if parts[0] != '-' else None
                    break
                    
            if service_found:
                status = 'running' if pid else 'loaded'
            else:
                status = 'not_found'
                
            return {
                'name': service_name,
                'status': status,
                'pid': pid
            }
        except subprocess.TimeoutExpired:
            return {
                'name': service_name,
                'status': 'timeout',
                'error': '命令執(zhí)行超時(shí)'
            }
        except Exception as e:
            return {
                'name': service_name,
                'status': 'error',
                'error': str(e)
            }
            
    def check_service(self, service_name):
        """檢查服務(wù)狀態(tài)（跨平臺(tái)）"""
        if self.os_type == 'windows':
            return self.check_windows_service(service_name)
        elif self.os_type == 'linux':
            return self.check_linux_service(service_name)
        elif self.os_type == 'darwin':  # macOS
            return self.check_macos_service(service_name)
        else:
            return {
                'name': service_name,
                'status': 'unsupported',
                'error': f'不支持的操作系統(tǒng): {self.os_type}'
            }
            
    def check_all_services(self):
        """檢查所有服務(wù)的狀態(tài)"""
        print(f"開始檢查服務(wù)狀態(tài) (操作系統(tǒng): {self.os_type})")
        print("=" * 60)
        
        self.results = []
        
        for service in self.services:
            print(f"檢查服務(wù): {service['display_name']} ({service['name']})")
            result = self.check_service(service['name'])
            result['display_name'] = service['display_name']
            result['description'] = service['description']
            result['check_time'] = datetime.now().isoformat()
            self.results.append(result)
            
            # 顯示結(jié)果
            status_display = {
                'running': '? 運(yùn)行中',
                'stopped': '? 已停止',
                'inactive': '??  未激活',
                'loaded': '?? 已加載',
                'not_found': '? 未找到',
                'timeout': '? 超時(shí)',
                'error': '??  錯(cuò)誤',
                'unknown': '? 未知'
            }
            
            status_text = status_display.get(result['status'], result['status'])
            print(f"  狀態(tài): {status_text}")
            
            if 'error' in result:
                print(f"  錯(cuò)誤: {result['error']}")
                
            print()
            
        return self.results
        
    def generate_report(self):
        """生成服務(wù)狀態(tài)報(bào)告"""
        report = []
        report.append("=" * 80)
        report.append("系統(tǒng)服務(wù)狀態(tài)檢查報(bào)告")
        report.append("=" * 80)
        report.append(f"檢查時(shí)間: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
        report.append(f"操作系統(tǒng): {self.os_type}")
        report.append(f"檢查服務(wù)數(shù): {len(self.services)}")
        report.append("")
        
        # 統(tǒng)計(jì)信息
        status_counts = {}
        for result in self.results:
            status = result['status']
            status_counts[status] = status_counts.get(status, 0) + 1
            
        report.append("狀態(tài)統(tǒng)計(jì):")
        for status, count in status_counts.items():
            report.append(f"  {status}: {count}")
        report.append("")
        
        # 詳細(xì)結(jié)果
        report.append("服務(wù)詳細(xì)信息:")
        report.append("-" * 50)
        for result in self.results:
            status_display = {
                'running': '? 運(yùn)行中',
                'stopped': '? 已停止',
                'inactive': '??  未激活',
                'loaded': '?? 已加載',
                'not_found': '? 未找到',
                'timeout': '? 超時(shí)',
                'error': '??  錯(cuò)誤',
                'unknown': '? 未知'
            }
            
            status_text = status_display.get(result['status'], result['status'])
            report.append(f"{result['display_name']} ({result['name']}): {status_text}")
            if result['description']:
                report.append(f"  描述: {result['description']}")
            if 'error' in result:
                report.append(f"  錯(cuò)誤: {result['error']}")
            report.append("")
            
        # 錯(cuò)誤信息
        if self.errors:
            report.append("錯(cuò)誤信息:")
            report.append("-" * 50)
            for error in self.errors:
                report.append(f"  {error}")
            report.append("")
            
        report.append("=" * 80)
        return "\n".join(report)
        
    def save_report(self, filename):
        """保存報(bào)告到文件"""
        try:
            report = self.generate_report()
            with open(filename, 'w', encoding='utf-8') as f:
                f.write(report)
            print(f"報(bào)告已保存到: {filename}")
            return True
        except Exception as e:
            print(f"保存報(bào)告時(shí)出錯(cuò): {e}")
            return False
            
    def save_json_report(self, filename):
        """保存JSON格式報(bào)告"""
        try:
            report_data = {
                'check_time': datetime.now().isoformat(),
                'os_type': self.os_type,
                'services': self.results,
                'errors': self.errors
            }
            
            with open(filename, 'w', encoding='utf-8') as f:
                json.dump(report_data, f, indent=2, ensure_ascii=False)
            print(f"JSON報(bào)告已保存到: {filename}")
            return True
        except Exception as e:
            print(f"保存JSON報(bào)告時(shí)出錯(cuò): {e}")
            return False
            
    def send_alert(self, smtp_config, alert_services=None):
        """發(fā)送告警郵件"""
        try:
            # 確定需要告警的服務(wù)
            if alert_services is None:
                # 默認(rèn)告警停止或異常的服務(wù)
                alert_services = [
                    r for r in self.results 
                    if r['status'] not in ['running', 'loaded']
                ]
                
            if not alert_services:
                print("沒有需要告警的服務(wù)")
                return True
                
            # 構(gòu)建郵件內(nèi)容
            subject = f"服務(wù)狀態(tài)告警 - {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
            body = "以下服務(wù)狀態(tài)異常:\n\n"
            
            for service in alert_services:
                body += f"服務(wù): {service['display_name']} ({service['name']})\n"
                body += f"狀態(tài): {service['status']}\n"
                if 'error' in service:
                    body += f"錯(cuò)誤: {service['error']}\n"
                body += "\n"
                
            # 發(fā)送郵件
            msg = MimeMultipart()
            msg['From'] = smtp_config['from']
            msg['To'] = smtp_config['to']
            msg['Subject'] = subject
            
            msg.attach(MimeText(body, 'plain', 'utf-8'))
            
            server = smtplib.SMTP(smtp_config['server'], smtp_config['port'])
            server.starttls()
            server.login(smtp_config['username'], smtp_config['password'])
            server.send_message(msg)
            server.quit()
            
            print("告警郵件已發(fā)送")
            return True
            
        except Exception as e:
            print(f"發(fā)送告警郵件時(shí)出錯(cuò): {e}")
            return False
            
    def continuous_monitor(self, interval=60, max_checks=None):
        """持續(xù)監(jiān)控服務(wù)狀態(tài)"""
        check_count = 0
        
        try:
            while True:
                print(f"\n[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] 開始第 {check_count + 1} 次檢查")
                self.check_all_services()
                
                # 顯示簡(jiǎn)要狀態(tài)
                running_count = len([r for r in self.results if r['status'] == 'running'])
                print(f"運(yùn)行中服務(wù): {running_count}/{len(self.results)}")
                
                check_count += 1
                
                # 檢查是否達(dá)到最大檢查次數(shù)
                if max_checks and check_count >= max_checks:
                    break
                    
                # 等待下次檢查
                time.sleep(interval)
                
        except KeyboardInterrupt:
            print("\n\n停止監(jiān)控...")

def create_sample_config():
    """創(chuàng)建示例配置文件"""
    config = {
        "services": [
            {
                "name": "ssh",
                "display_name": "SSH服務(wù)",
                "description": "安全Shell服務(wù)"
            },
            {
                "name": "nginx",
                "display_name": "Nginx Web服務(wù)器",
                "description": "高性能HTTP和反向代理服務(wù)器"
            },
            {
                "name": "mysql",
                "display_name": "MySQL數(shù)據(jù)庫",
                "description": "關(guān)系型數(shù)據(jù)庫管理系統(tǒng)"
            }
        ]
    }
    
    try:
        with open('service_config.json', 'w', encoding='utf-8') as f:
            json.dump(config, f, indent=2, ensure_ascii=False)
        print("示例配置文件已創(chuàng)建: service_config.json")
        return True
    except Exception as e:
        print(f"創(chuàng)建示例配置文件時(shí)出錯(cuò): {e}")
        return False

def main():
    parser = argparse.ArgumentParser(description="系統(tǒng)服務(wù)狀態(tài)檢查工具")
    parser.add_argument("-c", "--config", help="配置文件路徑")
    parser.add_argument("-s", "--service", action='append', help="要檢查的服務(wù)名稱（可多次使用）")
    parser.add_argument("-o", "--output", help="保存報(bào)告到文件")
    parser.add_argument("-j", "--json", help="保存JSON報(bào)告到文件")
    parser.add_argument("-m", "--monitor", type=int, help="持續(xù)監(jiān)控間隔（秒）")
    parser.add_argument("--max-checks", type=int, help="最大檢查次數(shù)")
    parser.add_argument("--create-config", action="store_true", help="創(chuàng)建示例配置文件")
    parser.add_argument("--smtp-server", help="SMTP服務(wù)器地址")
    parser.add_argument("--smtp-port", type=int, default=587, help="SMTP服務(wù)器端口")
    parser.add_argument("--smtp-username", help="SMTP用戶名")
    parser.add_argument("--smtp-password", help="SMTP密碼")
    parser.add_argument("--smtp-from", help="發(fā)件人郵箱")
    parser.add_argument("--smtp-to", help="收件人郵箱")
    
    args = parser.parse_args()
    
    # 創(chuàng)建示例配置文件
    if args.create_config:
        create_sample_config()
        return
        
    try:
        checker = ServiceStatusChecker()
        
        # 加載服務(wù)列表
        if args.config:
            if not checker.load_services_from_config(args.config):
                print("加載配置文件失敗")
                sys.exit(1)
        elif args.service:
            for service_name in args.service:
                checker.add_service(service_name)
        else:
            print("請(qǐng)指定配置文件或服務(wù)名稱")
            sys.exit(1)
            
        # 檢查服務(wù)狀態(tài)
        if args.monitor:
            # 持續(xù)監(jiān)控模式
            checker.continuous_monitor(args.monitor, args.max_checks)
        else:
            # 單次檢查模式
            checker.check_all_services()
            
            # 生成報(bào)告
            if args.output:
                checker.save_report(args.output)
            elif args.json:
                checker.save_json_report(args.json)
            else:
                print(checker.generate_report())
                
            # 發(fā)送告警郵件
            if args.smtp_server and args.smtp_username and args.smtp_password:
                smtp_config = {
                    'server': args.smtp_server,
                    'port': args.smtp_port,
                    'username': args.smtp_username,
                    'password': args.smtp_password,
                    'from': args.smtp_from or args.smtp_username,
                    'to': args.smtp_to
                }
                checker.send_alert(smtp_config)
                
    except KeyboardInterrupt:
        print("\n\n用戶中斷操作")
    except Exception as e:
        print(f"程序執(zhí)行出錯(cuò): {e}")
        sys.exit(1)

if __name__ == "__main__":
    main()

# 創(chuàng)建示例配置文件
python service_checker.py --create-config

# 使用配置文件檢查服務(wù)
python service_checker.py -c service_config.json

# 檢查單個(gè)服務(wù)
python service_checker.py -s nginx

# 檢查多個(gè)服務(wù)
python service_checker.py -s nginx -s mysql -s ssh

# 保存報(bào)告到文件
python service_checker.py -c service_config.json -o report.txt

# 保存JSON報(bào)告
python service_checker.py -c service_config.json -j report.json

# 持續(xù)監(jiān)控服務(wù)狀態(tài)（每30秒檢查一次）
python service_checker.py -c service_config.json -m 30

# 持續(xù)監(jiān)控最多10次
python service_checker.py -c service_config.json -m 30 --max-checks 10

# 發(fā)送告警郵件
python service_checker.py -c service_config.json \
  --smtp-server smtp.gmail.com \
  --smtp-username your_email@gmail.com \
  --smtp-password your_password \
  --smtp-to admin@example.com

{
  "services": [
    {
      "name": "ssh",
      "display_name": "SSH服務(wù)",
      "description": "安全Shell服務(wù)"
    },
    {
      "name": "nginx",
      "display_name": "Nginx Web服務(wù)器",
      "description": "高性能HTTP和反向代理服務(wù)器"
    },
    {
      "name": "mysql",
      "display_name": "MySQL數(shù)據(jù)庫",
      "description": "關(guān)系型數(shù)據(jù)庫管理系統(tǒng)"
    }
  ]
}

python service_checker.py -c service_config.json

python service_checker.py -c service_config.json -m 60 --max-checks 10 -o status.log

python service_checker.py -c service_config.json \
  --smtp-server smtp.gmail.com \
  --smtp-username admin@example.com \
  --smtp-password app_password \
  --smtp-to ops-team@example.com