/*
 *  Copyright (C) 2010-2021 JPEXS
 * 
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 * 
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 * 
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.jpexs.decompiler.flash.gui;

import com.jpexs.browsers.cache.CacheEntry;
import com.jpexs.browsers.cache.CacheImplementation;
import com.jpexs.browsers.cache.CacheReader;
import com.jpexs.decompiler.flash.SWFSourceInfo;
import com.jpexs.decompiler.flash.configuration.Configuration;
import com.jpexs.helpers.Helper;
import com.jpexs.helpers.ReReadableInputStream;
import java.awt.BorderLayout;
import java.awt.Container;
import java.awt.FlowLayout;
import java.awt.Image;
import java.awt.event.ActionEvent;
import java.awt.event.KeyEvent;
import java.awt.event.KeyListener;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Vector;
import javax.swing.BoxLayout;
import javax.swing.JButton;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JList;
import javax.swing.JPanel;
import javax.swing.JProgressBar;
import javax.swing.JScrollPane;
import javax.swing.JTextField;
import javax.swing.SwingWorker;
import javax.swing.filechooser.FileFilter;

/**
 *
 * @author JPEXS
 */
public class LoadFromCacheFrame extends AppFrame {

    private final JList<CacheEntry> list;

    private final JTextField searchField;

    private List<CacheImplementation> caches;

    private List<CacheEntry> entries;

    private final JProgressBar progressBar;

    private final JButton saveButton;

    private final JButton refreshButton;

    private final JButton openButton;

    public LoadFromCacheFrame() {
        setSize(900, 600);
        View.setWindowIcon(this);
        View.centerScreen(this);
        setTitle(translate("dialog.title"));
        setDefaultCloseOperation(HIDE_ON_CLOSE);
        Container cnt = getContentPane();
        list = new JList<>();
        list.addMouseListener(new MouseAdapter() {
            @Override
            public void mouseClicked(MouseEvent e) {
                if (e.getClickCount() > 1) {
                    openSWF();
                }
            }
        });
        searchField = new JTextField();
        searchField.addKeyListener(new KeyListener() {
            @Override
            public void keyTyped(KeyEvent e) {
                filter();
            }

            @Override
            public void keyPressed(KeyEvent e) {
            }

            @Override
            public void keyReleased(KeyEvent e) {
            }
        });
        cnt.setLayout(new BorderLayout());
        cnt.add(searchField, BorderLayout.NORTH);
        cnt.add(new JScrollPane(list), BorderLayout.CENTER);

        JPanel bottomPanel = new JPanel();
        bottomPanel.setLayout(new BoxLayout(bottomPanel, BoxLayout.Y_AXIS));

        JPanel buttonsPanel = new JPanel(new FlowLayout());

        openButton = new JButton(translate("button.open"));
        openButton.addActionListener(this::openButtonActionPerformed);
        buttonsPanel.add(openButton);

        saveButton = new JButton(translate("button.save"));
        saveButton.addActionListener(this::saveButtonActionPerformed);
        buttonsPanel.add(saveButton);

        refreshButton = new JButton(translate("button.refresh"));
        refreshButton.addActionListener(this::refreshButtonActionPerformed);
        buttonsPanel.add(refreshButton);

        JPanel browsersPanel = new JPanel(new FlowLayout());
        browsersPanel.add(new JLabel(translate("supported.browsers")));
        JLabel chromeLabel = new JLabel("Google Chrome", View.getIcon("chrome16"), JLabel.CENTER);
        JLabel firefoxLabel = new JLabel("Mozilla Firefox*", View.getIcon("firefox16"), JLabel.CENTER);
        browsersPanel.add(chromeLabel);
        browsersPanel.add(firefoxLabel);
        buttonsPanel.setAlignmentX(0.5f);

        progressBar = new JProgressBar();
        progressBar.setAlignmentX(0.5f);
        progressBar.setIndeterminate(true);
        bottomPanel.add(progressBar);
        bottomPanel.add(buttonsPanel);
        browsersPanel.setAlignmentX(0.5f);
        bottomPanel.add(browsersPanel);
        JLabel infoLabel = new JLabel(translate("info.closed"));
        infoLabel.setAlignmentX(0.5f);
        bottomPanel.add(infoLabel);
        cnt.add(bottomPanel, BorderLayout.SOUTH);
        progressBar.setVisible(false);
        openButton.setEnabled(false);
        saveButton.setEnabled(false);

        List<Image> images = new ArrayList<>();
        images.add(View.loadImage("loadcache16"));
        images.add(View.loadImage("loadcache32"));
        setIconImages(images);
        refresh();
    }

    private void refresh() {
        progressBar.setVisible(true);
        openButton.setEnabled(false);
        saveButton.setEnabled(false);
        refreshButton.setEnabled(false);
        new SwingWorker<Object, Object>() {
            @Override
            protected Object doInBackground() throws Exception {
                if (caches == null) {
                    caches = new ArrayList<>();
                    for (String b : CacheReader.availableBrowsers()) {
                        caches.add(CacheReader.getBrowserCache(b));
                    }
                } else {
                    for (CacheImplementation c : caches) {
                        c.refresh();
                    }
                }
                entries = new ArrayList<>();
                for (CacheImplementation c : caches) {
                    List<CacheEntry> list = c.getEntries();
                    if (list != null) {
                        for (CacheEntry en : c.getEntries()) {
                            String contentType = en.getHeader("Content-Type");
                            if ("application/x-shockwave-flash".equals(contentType)) {
                                entries.add(en);
                            }
                        }
                    }
                }
                filter();

                return null;
            }

            @Override
            protected void done() {
                openButton.setEnabled(true);
                saveButton.setEnabled(true);
                refreshButton.setEnabled(true);
                progressBar.setVisible(false);
            }
        }.execute();

    }

    private void filter() {
        String search = searchField.getText();
        List<CacheEntry> filtered = new ArrayList<>();
        for (CacheEntry en : entries) {
            if (search.isEmpty() || en.getRequestURL().contains(search)) {
                filtered.add(en);
            }
        }
        list.setListData(new Vector<>(filtered));
    }

    private static String entryToFileName(CacheEntry en) {
        String ret = en.getRequestURL();
        //Strip parameters
        if (ret.contains("?")) {
            ret = ret.substring(0, ret.indexOf('?'));
        }
        //Strip path
        if (ret.contains("/")) {
            ret = ret.substring(ret.lastIndexOf('/') + 1);
        }
        return ret;
    }

    private void openSWF() {
        CacheEntry en = list.getSelectedValue();
        if (en != null) {
            ReReadableInputStream str = new ReReadableInputStream(en.getResponseDataStream());
            SWFSourceInfo sourceInfo = new SWFSourceInfo(str, null, entryToFileName(en));
            Main.openFile(sourceInfo);
        }
    }

    private void refreshButtonActionPerformed(ActionEvent evt) {
        refresh();
    }

    private void openButtonActionPerformed(ActionEvent evt) {
        openSWF();
    }

    private void saveButtonActionPerformed(ActionEvent evt) {
        List<CacheEntry> selected = list.getSelectedValuesList();
        if (!selected.isEmpty()) {
            JFileChooser fc = new JFileChooser();
            fc.setCurrentDirectory(new File(Configuration.lastSaveDir.get()));
            if (selected.size() > 1) {
                fc.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
            } else {
                fc.setSelectedFile(new File(Configuration.lastSaveDir.get(), entryToFileName(selected.get(0))));
                fc.setFileFilter(new FileFilter() {
                    @Override
                    public boolean accept(File f) {
                        return (f.getName().endsWith(".swf")) || (f.isDirectory());
                    }

                    @Override
                    public String getDescription() {
                        return AppStrings.translate("filter.swf");
                    }
                });
            }
            fc.setAcceptAllFileFilterUsed(false);
            JFrame f = new JFrame();
            View.setWindowIcon(f);
            if (fc.showSaveDialog(f) == JFileChooser.APPROVE_OPTION) {
                File file = Helper.fixDialogFile(fc.getSelectedFile());
                try {
                    if (selected.size() == 1) {
                        Helper.saveStream(selected.get(0).getResponseDataStream(), file);
                    } else {
                        for (CacheEntry sel : selected) {
                            Helper.saveStream(sel.getResponseDataStream(), new File(file, entryToFileName(sel)));
                        }
                    }
                    Configuration.lastSaveDir.set(file.getParentFile().getAbsolutePath());
                } catch (IOException ex) {
                    View.showMessageDialog(null, translate("error.file.write"));
                }
            }
        }
    }
}