package com.example.client.service;

import com.example.client.dto.ColumnDto;
import com.example.client.utils.*;
import com.example.server.DataSync.service.DataSyncService;
import com.example.server.progressTrack.model.DjJdgzNetworkLevel2List;
import com.example.server.progressTrack.service.DjJdgzNetworkLevel2ListService;
import com.mxgraph.view.mxGraph;
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.io.*;
import java.nio.charset.Charset;
import java.util.Date;
import java.util.Enumeration;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;


@Service
public class ImportDataService {
    @Value("${data.imgDir}")
    private String imgPath;
    private File selectedFiles;
    private DefaultTableModel model;
    @Value("${zt.oss.Unzip-path}")
    private String UnzipPath;
    @Autowired
    private DataSyncService dataSyncService;

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

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


        JScrollPane scrolltable = new JScrollPane(table, JScrollPane.VERTICAL_SCROLLBAR_ALWAYS, JScrollPane.HORIZONTAL_SCROLLBAR_ALWAYS);

        scrolltable.setViewportView(table);
        scrolltable.getViewport().setBackground(Color.WHITE);

        JButton browseButton = new JButton("选择文件");
        jPanel.add(browseButton);
        jPanel.add(scrolltable);

        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.getSelectedFile();

                    Integer oldNum = model.getRowCount();
                    String[][] data = new String[oldNum + 1][2];
                    for (int i = 0; i < oldNum; i++) {
                        data[i][0] = model.getValueAt(i, 0).toString();
                    }

                    String suffix = selectedFiles.getName().substring(selectedFiles.getName().lastIndexOf("."));
                    if (suffix.equals(".zip")) {
                        data[oldNum][0] = selectedFiles.getPath();
                    } else {
                        flag = false;

                    }

                    if (!flag) {
                        JOptionPane.showMessageDialog(null, "只能导入压缩包，且不超过2G", "提示", JOptionPane.WARNING_MESSAGE);
                        return;
                    }

                    model.setDataVector(data, columnIdentifiers);
                    table.setModel(model);
                    table.getColumnModel().getColumn(0).setPreferredWidth(600);
                    table.getColumnModel().getColumn(1).setPreferredWidth(100);


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

                } 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 {
                    Boolean flag = true;
                    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++) {
                        if (filepathArr[i].contains(".zip")) {
                            data[i + oldNum][0] = filepathArr[i];
                        } else {
                            flag = false;
                            break;
                        }
                    }
                    if (!flag) {
                        JOptionPane.showMessageDialog(null, "只能导入压缩包文件，且不超过2G\"", "提示", JOptionPane.WARNING_MESSAGE);
                        return flag;
                    }
                    model.setDataVector(data, columnIdentifiers);
                    table.setModel(model);
                    table.getColumnModel().getColumn(0).setPreferredWidth(650);
                    table.getColumnModel().getColumn(1).setPreferredWidth(80);

                    table.getColumnModel().getColumn(0).setCellRenderer(new TableViewRenderer());
                    table.getColumnModel().getColumn(1).setCellRenderer(new TableCellRendererButton());
                    table.getColumnModel().getColumn(1).setCellEditor(new TableCellEditorButton(null, null));
                    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 String UnzipFile(){
        String filePath = model.getValueAt(0, 0).toString();
        String outputDirectory = UnzipPath;
        try {

            File zipFile = new File(filePath);
            if (!zipFile.exists()) {
                throw new FileNotFoundException("ZIP文件不存在：" + filePath);
            }
            if (!zipFile.canRead()) {
                System.err.println("无法读取ZIP文件，请检查文件权限：" + filePath);
                return "false";
            }

            // 创建目标目录
            File outputDir = new File(outputDirectory);
            if (!outputDir.exists()) {
                outputDir.mkdirs();
            }

            // 使用ZipFile读取ZIP文件
            try (ZipFile zip = new ZipFile(zipFile, Charset.forName("GBK"))) {
                Enumeration<? extends ZipEntry> entries = zip.entries();

                while (entries.hasMoreElements()) {
                    ZipEntry entry = entries.nextElement();
                    String entryName = entry.getName();

                    // 创建目标文件或目录
                    File entryFile = new File(outputDirectory, entryName);
                    File entryDir = entryFile.getParentFile();

                    // 如果是目录，创建目录
                    if (!entryDir.exists()) {
                        entryDir.mkdirs();
                    }

                    // 如果是文件，提取文件内容
                    if (!entry.isDirectory()) {
                        try (InputStream is = zip.getInputStream(entry);
                             FileOutputStream fos = new FileOutputStream(entryFile)) {
                            byte[] buffer = new byte[1024];
                            int length;
                            while ((length = is.read(buffer)) >= 0) {
                                fos.write(buffer, 0, length);
                            }
                        }
                    }
                }
            }
            System.out.println("文件解压完成，目标目录：" + outputDirectory);
            dataSyncService.importData();
        } catch (IOException e) {
            e.printStackTrace();
            return "false";
        }
        return "true";
    }
}