package com.example.client.utils;

import com.example.server.progressTrack.model.DjJdgzNetworkLevel3;
import com.example.server.progressTrack.model.DjJdgzTrackRecord;
import com.example.server.sysOss.model.SysOss;
import com.example.server.sysOss.service.OssEncryptService;
import com.example.server.sysOss.service.SysOssService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import javax.swing.*;
import javax.swing.table.DefaultTableModel;
import java.awt.*;
import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.Transferable;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.io.*;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

@Service
public class UploadFile {
    private File[] selectedFiles;
    @Autowired
    private SysOssService sysOssService;
    @Autowired
    private OssEncryptService ossEncryptService;
    private List<SysOss> ossList;
    private SysOss templet;

    @Value("${zt.oss.local-path}")
    private String localPath;
    private static final String key = "zhpt-key#%W";

    public JPanel uploadFile(JFrame frame, Integer width, Integer heigth, SysOss sysOssTemplet) {
        JPanel jPanel = new JPanel();
        jPanel.setPreferredSize(new Dimension(width, heigth));

        templet = sysOssTemplet;
        if (sysOssTemplet.getBusiId() != null) {
            ossList = sysOssService.getList(sysOssTemplet.getBusiId(), sysOssTemplet.getBusiType());
        } else {
            ossList = new ArrayList<>();
        }

        JTable table = new JTable();
        table.setRowHeight(10);
        DefaultTableModel model = new DefaultTableModel();
        String[] columnIdentifiers = {"文件地址", "操作"};//表头

        JScrollPane scrolltable = new JScrollPane(table, JScrollPane.VERTICAL_SCROLLBAR_ALWAYS, JScrollPane.HORIZONTAL_SCROLLBAR_ALWAYS);
        scrolltable.setPreferredSize(new Dimension(width, heigth - 100));
        scrolltable.setViewportView(table);
        scrolltable.getViewport().setBackground(Color.WHITE);


        JButton browseButton = new JButton("选择文件");
        jPanel.add(browseButton);
        jPanel.add(scrolltable);
        if (ossList != null && ossList.size() > 0) {
            String[][] data = new String[ossList.size()][2];
            for (int i = 0; i < ossList.size(); i++) {
                data[i][0] = ossList.get(i).getName();
            }

            model.setDataVector(data, columnIdentifiers);
            table.setModel(model);
            table.getColumnModel().getColumn(0).setPreferredWidth(width - 120);
            table.getColumnModel().getColumn(1).setPreferredWidth(120);

            table.getColumnModel().getColumn(0).setCellRenderer(new TableViewRenderer());
            table.getColumnModel().getColumn(1).setCellRenderer(new TableCellRendererButton());
            table.getColumnModel().getColumn(1).setCellEditor(new TableCellEditorButton(ossList, sysOssService));

        }

        final boolean[] historyCalled = {false};
        table.addMouseListener(new MouseAdapter() {
            @Override
            public void mouseClicked(MouseEvent e) {
                if (historyCalled[0]) return;
                // 获取点击的行和列
                int row = table.rowAtPoint(e.getPoint());
                int column = table.columnAtPoint(e.getPoint());

                // 检查是否点击了有效单元格
                if (row >= 0 && column >= 0) {
                    if (column == 0) { // 列索引从0开始
                        String path = ossList.get(row).getPath();
                        String name = ossList.get(row).getName();
                        File file = new File(localPath + path);
                        InputStream is = null;

                        try {
                            is = ossEncryptService.decryptStream(file, ossEncryptService.getKey(key));

                            // 创建文件选择器并设置初始文件名
                            JFileChooser fileChooser = new JFileChooser();
                            fileChooser.setSelectedFile(new File(name)); // 设置默认文件名

                            int result = fileChooser.showSaveDialog(frame);

                            if (result == JFileChooser.APPROVE_OPTION) {
                                File selectedFile = fileChooser.getSelectedFile();

                                // 将 InputStream 的内容写入到 selectedFile
                                try (InputStream input = is;
                                     OutputStream output = new FileOutputStream(selectedFile)) {

                                    byte[] buffer = new byte[8192]; // 缓冲区大小
                                    int bytesRead;
                                    while ((bytesRead = input.read(buffer)) != -1) {
                                        output.write(buffer, 0, bytesRead);
                                    }

                                    // 文件保存成功
                                    JOptionPane.showMessageDialog(frame, "文件下载成功");
                                } catch (IOException ex) {
                                    ex.printStackTrace();
                                    JOptionPane.showMessageDialog(frame, "文件保存失败: " + ex.getMessage());
                                }
                            } else {
                                JOptionPane.showMessageDialog(frame, "文件下载取消");
                            }
                        } catch (Exception exception) {
                            exception.printStackTrace();
                        } finally {
                            // 确保 InputStream 被关闭，避免资源泄露
                            if (is != null) {
                                try {
                                    is.close();
                                } catch (IOException ec) {
                                    ec.printStackTrace();
                                }
                            }
                        }
                    }
                }
            }
        });

        browseButton.addActionListener(new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent e) {
                Boolean flag = true;
                // 保存当前的外观设置
                LookAndFeel savedLookAndFeel = UIManager.getLookAndFeel();

                // 设置文件选择器外观
                try {
                    UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());
                } catch (Exception ex) {
                    ex.printStackTrace();
                }

                JFileChooser fileChooser = new JFileChooser();

                fileChooser.setDialogTitle("选择文件");
                fileChooser.setPreferredSize(new Dimension(800, 500));


                fileChooser.setMultiSelectionEnabled(true);

                //恢复外观，避免改变所有组件外观
                try {
                    UIManager.setLookAndFeel(savedLookAndFeel);
                } catch (UnsupportedLookAndFeelException ed) {
                    ed.printStackTrace();
                }

                int result = fileChooser.showOpenDialog(frame);
                if (result == JFileChooser.FILES_ONLY) {
                    selectedFiles = fileChooser.getSelectedFiles();

                    Integer oldNum = model.getRowCount();
                    String[][] data = new String[oldNum + selectedFiles.length][2];
                    for (int i = 0; i < oldNum; i++) {
                        data[i][0] = model.getValueAt(i, 0).toString();
                    }
                    for (int i = 0; i < selectedFiles.length; i++) {
                        data[i + oldNum][0] = selectedFiles[i].getPath();
                        String suffix = selectedFiles[i].getName().substring(selectedFiles[i].getName().lastIndexOf("."));
                        SysOss oss = null;
                        try {
                            oss = templet.clone();
                        } catch (CloneNotSupportedException cloneNotSupportedException) {
                            cloneNotSupportedException.printStackTrace();
                        }
                        oss.setBusiType(suffix);
                        oss.setPath(selectedFiles[i].getPath());
                        oss.setName(selectedFiles[i].getName());
                        ossList.add(oss);
                    }

                    model.setDataVector(data, columnIdentifiers);
                    table.setModel(model);
                    table.getColumnModel().getColumn(0).setPreferredWidth(width - 120);
                    table.getColumnModel().getColumn(1).setPreferredWidth(120);


                    table.getColumnModel().getColumn(0).setCellRenderer(new TableViewRenderer());
                    table.getColumnModel().getColumn(1).setCellRenderer(new TableCellRendererButton());
                    table.getColumnModel().getColumn(1).setCellEditor(new TableCellEditorButton(ossList, sysOssService));

                } else {
                    Object[] options = {"OK ", "CANCEL "};
                    JOptionPane.showOptionDialog(null, "选择的文件不正确 ", "提示", JOptionPane.DEFAULT_OPTION,
                            JOptionPane.WARNING_MESSAGE, null, options, options[0]);
                }
            }
        });


        scrolltable.setTransferHandler(new TransferHandler() {
            private static final long serialVersionUID = 1L;

            @Override
            public boolean importData(JComponent comp, Transferable t) {
                try {
                    Object o = t.getTransferData(DataFlavor.javaFileListFlavor);
                    String filepath = o.toString();
                    if (filepath.startsWith("[")) {
                        filepath = filepath.substring(1);
                    }
                    if (filepath.endsWith("]")) {
                        filepath = filepath.substring(0, filepath.length() - 1);
                    }
                    String[] filepathArr = filepath.split(", ");
                    Integer oldNum = model.getRowCount();
                    String[][] data = new String[oldNum + filepathArr.length][2];
                    for (int i = 0; i < oldNum; i++) {
                        data[i][0] = model.getValueAt(i, 0).toString();
                    }
                    for (int i = 0; i < filepathArr.length; i++) {
                        data[i + oldNum][0] = filepathArr[i];
                        SysOss oss = null;
                        try {
                            oss = templet.clone();
                        } catch (CloneNotSupportedException cloneNotSupportedException) {
                            cloneNotSupportedException.printStackTrace();
                        }
                        int lastSlashIndex = filepathArr[i].lastIndexOf('/');
                        String fullFileName = "";
                        if (lastSlashIndex != -1) {
                            fullFileName = filepathArr[i].substring(lastSlashIndex + 1);
                        }
                        oss.setPath(filepathArr[i]);
                        oss.setName(fullFileName);
                        ossList.add(oss);
                    }
                    model.setDataVector(data, columnIdentifiers);
                    table.setModel(model);
                    table.getColumnModel().getColumn(0).setPreferredWidth(width - 120);
                    table.getColumnModel().getColumn(1).setPreferredWidth(120);

                    table.getColumnModel().getColumn(0).setCellRenderer(new TableViewRenderer());
                    table.getColumnModel().getColumn(1).setCellRenderer(new TableCellRendererButton());
                    table.getColumnModel().getColumn(1).setCellEditor(new TableCellEditorButton(ossList, sysOssService));
                    return true;
                } catch (Exception e) {
                    e.printStackTrace();
                }
                return false;
            }

            @Override
            public boolean canImport(JComponent comp, DataFlavor[] flavors) {
                for (int i = 0; i < flavors.length; i++) {
                    if (DataFlavor.javaFileListFlavor.equals(flavors[i])) {
                        return true;
                    }
                }
                return false;
            }
        });
        return jPanel;
    }

    public void save(Long id) {
        if (ossList != null && ossList.size() > 0) {
            for (SysOss oss : ossList) {
                if (oss.getBusiId()==null){
                    oss.setBusiId(id);
                }
                if (oss.getId() == null) {
                    sysOssService.upload(oss);
                }
            }
        }
    }
}