package com.cloud.device.utils;

import com.cloud.common.model.FileInfos;
import com.cloud.common.utils.FastDFSUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import java.io.*;
import java.util.List;

@Component
@Slf4j
public class CreatePhUtil {

    //user 用户模块 接口 端口
    @Value("${fdfsImgUrl.ffmpeg}")
    private String ffmpegUrl = "ffmpeg.exe";

    //user 用户模块 接口 端口
    @Value("${fdfsImgUrl.ffmpegSaveJpgUrl}")
    private String saveJpgUrl = "test.jpg";

    @Autowired
    private FastDFSUtil fastDFSUtil;
    //public static final String FFMPEG_PATH = "E:/ffmpeg/ffmpeg.exe";
    public String processImg(String veido_path) throws IOException {

        String window = System.getProperties().getProperty("os.name");
        log.info("===========os.name:" + window);
        log.info("===========file.separator:" + System.getProperties().getProperty("file.separator"));
        String ffmpeg_path = ffmpegUrl;
        if (StringUtils.isBlank(ffmpeg_path)) {
            if (window.contains("Windows")) ffmpeg_path = "ffmpeg.exe";
            else ffmpeg_path = "ffmpeg";
        }
//     ffmpeg -i rtsp://admin:a1234567@192.168.1.215:554/h264/ch1/main/av_stream
//     -vf select='eq(pict_type\,I)',setpts='N/(25*TB)'
//             -f image2 -s 1920*1080 -y C:\Users\25135\Desktop\test233.jpg
        log.info("veido_path=" + veido_path);
        List<String> commands = new java.util.ArrayList<String>();
        commands.add(ffmpeg_path);
        commands.add("-i");
        commands.add(veido_path);
        commands.add("-y");
        commands.add("-f");
        commands.add("image2");
//  commands.add("-ss");	commands.add("0.001");//这个参数是设置截取视频多少秒时的画面
        commands.add("-t");
        commands.add("0.001");  // -vframes 1
  /*commands.add("-vframes");
  commands.add("1");*/
        commands.add("-vf");
        commands.add("select='eq(pict_type\\,I)',setpts='N/(25*TB)'");
        commands.add("-s");
        commands.add("1920*1080");
        commands.add(saveJpgUrl);
        Process process = null;
        try {
            ProcessBuilder builder = new ProcessBuilder();
            builder.command(commands);
//   builder.redirectErrorStream(true);
            process = builder.start();
        } catch (Exception e) {
            e.printStackTrace();
        }
        // 使用这种方式会在瞬间大量消耗CPU和内存等系统资源，所以这里我们需要对流进行处理
        InputStream errorStream = process.getErrorStream();
        InputStreamReader inputStreamReader = new InputStreamReader(errorStream);
        BufferedReader br = new BufferedReader(inputStreamReader);
        String line = "";
        while ((line = br.readLine()) != null) {
        }
        log.info("流返回：" + line);
        if (line == null) {
            try {
                FileInputStream fis = new FileInputStream(saveJpgUrl);
                byte[] bytes = readStream(fis);
                FileInfos fileInfos = fastDFSUtil.uploadByByte(bytes, "jpg");
                return fileInfos.getPath();
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                if (br != null) {
                    br.close();
                }
                if (inputStreamReader != null) {
                    inputStreamReader.close();
                }
                if (errorStream != null) {
                    errorStream.close();
                }
            }
        }
        return null;
    }

    public static void main(String[] args) {
        try {
            String s = new CreatePhUtil().processImg("rtsp://admin:a1234567@192.168.1.215:554/h264/ch1/main/av_stream");
            log.info("图片地址：" + s);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static byte[] readStream(InputStream inStream) throws Exception {
        byte[] buffer = new byte[1024];
        int len = -1;
        ByteArrayOutputStream outStream = new ByteArrayOutputStream();
        while ((len = inStream.read(buffer)) != -1) {
            outStream.write(buffer, 0, len);
        }
        byte[] data = outStream.toByteArray();
        outStream.close();
        inStream.close();
        return data;
    }

}