package com.ruoyi.cai.util;

import com.ruoyi.common.exception.base.BaseException;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.exec.*;
import org.apache.commons.exec.environment.EnvironmentUtils;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@Slf4j
public class LoginLogByFileUtil {

    public static List<String> getLog(String mobile){
        // 1. 定义命令（带管道的Linux命令，需用/bin/sh -c包裹）
        String commandFormat = "tac /home/server/api/logs/sys-console.log | grep \"%s\" | grep -m 10 \"auth/login\"";
        String command = String.format(commandFormat, mobile);
        CommandLine cmdLine = CommandLine.parse("/bin/sh -c " + command); // 自动处理参数分隔

        // 2. 配置执行器（设置超时、流处理）
        DefaultExecutor executor = new DefaultExecutor();
        executor.setExitValue(0); // 期望的退出码（0表示成功）

        // 设置进程超时（5秒，适配你的快速执行场景）
        ExecuteWatchdog watchdog = new ExecuteWatchdog(5000); // 5000ms = 5s
        executor.setWatchdog(watchdog);

        // 3. 处理输出（ByteArrayOutputStream存储10行结果，或用PumpStreamHandler实时处理）
        // 场景1：存储结果到内存（仅10行，内存无压力）
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        PumpStreamHandler streamHandler = new PumpStreamHandler(outputStream, outputStream); // 合并标准输出/错误流
        executor.setStreamHandler(streamHandler);

        // 场景2：实时处理每行输出（推荐，避免存储）
        // PumpStreamHandler streamHandler = new PumpStreamHandler(new LogOutputStream() {
        //     @Override
        //     protected void processLine(String line, int level) {
        //         // 实时处理每行数据（如输出到前端、写入日志）
        //         System.out.println(line);
        //     }
        // });
        // executor.setStreamHandler(streamHandler);

        try {
            // 4. 执行命令（环境变量使用系统默认，可自定义）
            Map<String, String> env = EnvironmentUtils.getProcEnvironment();
            executor.execute(cmdLine, env);

            // 5. 解析结果（场景1的方式）
            String result = outputStream.toString(StandardCharsets.UTF_8.name());
            String[] split = result.split(System.lineSeparator());
            List<String> lines = Arrays.stream(split).collect(Collectors.toList());
            System.out.println("匹配的行数：" + lines.size());
            lines.forEach(System.out::println);

        } catch (ExecuteException e) {
            if (watchdog.killedProcess()) {
                log.error("命令执行超时，已终止进程");
            } else {
                log.error("命令执行失败，退出码：" + e.getExitValue());
            }
            log.error(e.getMessage(),e);
            throw new BaseException("命令执行失败");
        } catch (IOException e) {
            log.error("IO异常：" + e.getMessage());
            log.error(e.getMessage(),e);
            throw new BaseException("IO异常");
        }
        return Collections.emptyList();
    }
}