Dépôt officiel du code source de l'ERP OpenConcerto

/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

 *

 * Copyright 2011-2019 OpenConcerto, by ILM Informatique. All rights reserved.

 *

 * The contents of this file are subject to the terms of the GNU General Public License Version 3

 * only ("GPL"). You may not use this file except in compliance with the License. You can obtain a

 * copy of the License at http://www.gnu.org/licenses/gpl-3.0.html See the License for the specific

 * language governing permissions and limitations under the License.

 *

 * When distributing the software, include this License Header Notice in each file.

 */

 package org.openconcerto.ui.login;

import org.openconcerto.ui.DefaultGridBagConstraints;

import org.openconcerto.ui.ReloadPanel;

import org.openconcerto.ui.SwingThrottle;

import org.openconcerto.ui.component.combo.ISearchableCombo;

import org.openconcerto.ui.valuewrapper.EmptyValueWrapper;

import org.openconcerto.ui.valuewrapper.ValueWrapperFactory;

import org.openconcerto.utils.Base64;

import org.openconcerto.utils.JImage;

import org.openconcerto.utils.i18n.TM;

import org.openconcerto.utils.i18n.TranslationManager;

import org.openconcerto.utils.jsonrpc.JSONRPCClient;

import org.openconcerto.utils.text.SimpleDocumentListener;

import java.awt.Color;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.Insets;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.beans.PropertyChangeEvent;

import java.beans.PropertyChangeListener;

import java.io.IOException;

import java.security.MessageDigest;

import java.security.NoSuchAlgorithmException;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Collections;

import java.util.List;

import java.util.Locale;

import java.util.ResourceBundle;

import java.util.concurrent.ExecutorService;

import java.util.concurrent.Executors;

import javax.swing.JButton;

import javax.swing.JCheckBox;

import javax.swing.JCheckBoxMenuItem;

import javax.swing.JFrame;

import javax.swing.JLabel;

import javax.swing.JPanel;

import javax.swing.JPasswordField;

import javax.swing.JPopupMenu;

import javax.swing.JSeparator;

import javax.swing.JTextField;

import javax.swing.SwingConstants;

import javax.swing.SwingUtilities;

import javax.swing.event.DocumentEvent;

import net.minidev.json.JSONArray;

import net.minidev.json.JSONObject;

public class LoginFrame extends JFrame implements ActionListener, ConnectionStateListener {

    public static final int CONNECTION_NOT_INITIATED = -1;

    public static final int CONNECTION_INPROGRESS = 0;

    public static final int CONNECTION_OK = 1;

    public static final int CONNECTION_TIMEOUT = 2;

    public static final int CONNECTION_REFUSED = 3;

    public static final int CONNECTION_BROKEN = 4;

    private final LoginProperties loginProperties;

    // Login

    private EmptyValueWrapper<String> textLogin;

    // Password

    private JPasswordField textPassWord;

    private String encryptedPassword;

    private String clearPassword;

    private boolean allowStoredPass = true;

    // Company

    private ISearchableCombo<Company> comboCompany;

    private ReloadPanel reloadPanel;

    private CompanyIListModel model;

    // 1/ open an empty interface, rotate reloadpanel

    // 2/ fill with stored login/password/ defaultcompany (id+name)

    // 3a/ when connection to server is ready, stop reloadpanel

    // 3b/ when login+pass are entered, retrieve allowed companies and enable

    // connect when

    // connection to server is ready

    private final JCheckBox saveCheckBox = new JCheckBox(getTM().translate("loginPanel.storePass"));

    private final JButton buttonConnect = new JButton(getTM().translate("loginPanel.loginAction"));

    private final JLabel loginLabel = new JLabel(getTM().translate("loginPanel.loginLabel"));

    private final JLabel passwordLabel = new JLabel(getTM().translate("loginPanel.passLabel"));

    private final JLabel companyLabel = new JLabel(getTM().translate("loginPanel.companyLabel"));

    private String localeBaseName = null;

    private final List<Locale> localesToDisplay = new ArrayList<Locale>();

    private final JButton langButton = new JButton(Locale.ROOT.getLanguage());

    private String company;

    private final JSONRPCClient client;

    private int connectionState = CONNECTION_NOT_INITIATED;

    private final ExecutorService executorService = Executors.newFixedThreadPool(1);

    private final String serverUrl;

    private String context;

    private int idCompany;

    private SwingThrottle tCheckValidity;

    public LoginFrame(String serverUrl, final JSONRPCClient client, String context) throws IOException {

        this.client = client;

        this.serverUrl = serverUrl;

        this.context = context;

        this.setContentPane(createLoginPanel());

        // load stored info

        System.err.println("LoginFrame.LoginFrame()" + serverUrl);

        this.loginProperties = new LoginProperties(serverUrl);

        this.loginProperties.load();

        final Company storedDefaultCompany = new Company(this.loginProperties.getLastCompanyId(), this.loginProperties.getLastCompanyName());

        this.init(this.loginProperties.getLastLoginName(), this.loginProperties.getEncryptedStoredPassword(), storedDefaultCompany);

    }

    public synchronized int getConnectionState() {

        return this.connectionState;

    }

    public synchronized void setConnectionState(int connectionState) {

        this.connectionState = connectionState;

    }

    private TM getTM() {

        return org.openconcerto.utils.i18n.TM.getInstance();

    }

    private void init(String storedLogin, String storedEncryptedPassword, Company storedDefaultCompany) {

        this.textLogin.setValue(storedLogin);

        if (this.saveCheckBox != null && storedEncryptedPassword != null && storedEncryptedPassword.length() > 0) {

            this.saveCheckBox.setSelected(true);

        }

        // to show the user its password has been retrieved

        if (storedEncryptedPassword != null) {

            final char[] s = new char[8];

            Arrays.fill(s, ' ');

            this.textPassWord.setText(new String(s));

            this.clearPassword = null;

        } else

            this.clearPassword = "";

        this.model = new CompanyIListModel();

        this.comboCompany.initCache(this.model);

        this.tCheckValidity = new SwingThrottle(800, new Runnable() {

            @Override

            public void run() {

                checkValidity();

            }

        });

    }

    private JPanel createLoginPanel() {

        JPanel panel = new JPanel();

        panel.setLayout(new GridBagLayout());

        final GridBagConstraints c = new GridBagConstraints();

        c.gridheight = 1;

        c.gridwidth = 2;

        c.gridx = 0;

        c.gridy = 0;

        c.weightx = 1;

        c.weighty = 0;

        c.insets = new Insets(0, 0, 0, 0);

        c.fill = GridBagConstraints.HORIZONTAL;

        // Logo

        JImage imageLogo = new JImage(LoginFrame.class.getResource("OpenConcerto_login.png"));

        imageLogo.setBackground(Color.WHITE);

        imageLogo.check();

        panel.add(imageLogo, c);

        c.gridy++;

        c.gridwidth = GridBagConstraints.REMAINDER;

        panel.add(new JSeparator(SwingConstants.HORIZONTAL), c);

        // Login

        c.insets = new Insets(2, 2, 1, 2);

        c.gridy++;

        c.gridwidth = 1;

        c.weightx = 0;

        this.loginLabel.setHorizontalAlignment(SwingConstants.RIGHT);

        panel.add(this.loginLabel, c);

        this.textLogin = new EmptyValueWrapper<String>(ValueWrapperFactory.create(new JTextField(), String.class));

        c.gridx++;

        c.weightx = 1;

        panel.add(this.textLogin.getComp(), c);

        ((JTextField) this.textLogin.getComp()).addActionListener(this);

        this.textLogin.addValueListener(new PropertyChangeListener() {

            public void propertyChange(final PropertyChangeEvent evt) {

                LoginFrame.this.tCheckValidity.execute();

            }

        });

        // Password

        c.gridy++;

        c.gridx = 0;

        c.weightx = 0;

        System.err.println("LoginFrame.createLoginPanel()22c");

        this.passwordLabel.setHorizontalAlignment(SwingConstants.RIGHT);

        panel.add(this.passwordLabel, c);

        this.textPassWord = new JPasswordField();

        c.gridx++;

        c.weightx = 1;

        panel.add(this.textPassWord, c);

        this.textPassWord.getDocument().addDocumentListener(new SimpleDocumentListener() {

            public void update(final DocumentEvent e) {

                LoginFrame.this.clearPassword = String.valueOf(LoginFrame.this.textPassWord.getPassword());

                LoginFrame.this.tCheckValidity.execute();

            }

        });

        this.textPassWord.addActionListener(this);

        System.err.println("LoginFrame.createLoginPanel()22d");

        // Societe

        c.gridy++;

        c.gridx = 0;

        c.weightx = 0;

        this.companyLabel.setHorizontalAlignment(SwingConstants.RIGHT);

        panel.add(this.companyLabel, c);

        this.comboCompany = new ISearchableCombo<Company>(true);

        this.comboCompany.setEnabled(false);

        c.gridx++;

        panel.add(this.comboCompany, c);

        // Button

        final JPanel panelButton = new JPanel();

        panelButton.setOpaque(false);

        panelButton.setLayout(new GridBagLayout());

        final GridBagConstraints c2 = new DefaultGridBagConstraints();

        c2.weightx = 1;

        if (this.allowStoredPass) {

            this.saveCheckBox.setOpaque(false);

            panelButton.add(this.saveCheckBox, c2);

            c2.weightx = 0;

        }

        c2.gridx++;

        // Language selector

        //

        this.langButton.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                final JPopupMenu menu = new JPopupMenu();

                final Locale locale = Locale.getDefault();

                for (final Locale l : LoginFrame.this.localesToDisplay) {

                    final JCheckBoxMenuItem menuItem = new JCheckBoxMenuItem(l.getDisplayName(l));

                    if (l.equals(locale)) {

                        menuItem.setSelected(true);

                    }

                    menu.add(menuItem);

                    menuItem.addActionListener(new ActionListener() {

                        @Override

                        public void actionPerformed(ActionEvent e) {

                            setUILanguage(l);

                        }

                    });

                }

                menu.show(LoginFrame.this.langButton, 0, 0);

            }

        });

        this.langButton.setOpaque(false);

        this.langButton.setBorderPainted(false);

        this.langButton.setContentAreaFilled(false);

        this.langButton.setBorder(null);

        this.langButton.setFocusable(false);

        this.langButton.setVisible(false);

        panelButton.add(this.langButton, c2);

        c2.gridx++;

        this.reloadPanel = new ReloadPanel();

        this.reloadPanel.setOpaque(false);

        this.reloadPanel.setMode(ReloadPanel.MODE_EMPTY);

        panelButton.add(this.reloadPanel, c2);

        c2.gridx++;

        c2.weightx = 0;

        this.buttonConnect.setEnabled(false);

        this.buttonConnect.setOpaque(false);

        panelButton.add(this.buttonConnect, c2);

        c.gridy++;

        c.gridx = 0;

        c.weightx = 0;

        c.gridwidth = GridBagConstraints.REMAINDER;

        c.weighty = 1;

        panel.add(panelButton, c);

        this.buttonConnect.addActionListener(this);

        initLocalization("org.openconcerto.ui.login.Messages",

                Arrays.asList(Locale.FRANCE, Locale.CANADA_FRENCH, new Locale("fr", "CH"), new Locale("fr", "BE"), Locale.UK, Locale.CANADA, Locale.US, Locale.GERMANY, new Locale("de", "CH")));

        setUILanguage(Locale.getDefault());

        return panel;

    }

    private void checkValidity() {

        String login = this.textLogin.getValue();

        this.buttonConnect.setEnabled(login != null && !login.trim().isEmpty() && this.comboCompany.getSelectedItem() != null);

        // this.buttonConnect.setEnabled(this.connectionAllowed == null &&

        // this.areFieldsValidated());

        // this.buttonConnect.setToolTipText(this.connectionAllowed);

        System.err.println("LoginFrame.checkValidity()");

        String ePassword = this.encryptedPassword;

        if (ePassword == null) {

            if (this.clearPassword == null) {

                this.clearPassword = "";

            }

            ePassword = encodePassword(this.clearPassword);

        }

        // Empty login is not allowed

        if (login == null || login.trim().isEmpty()) {

            return;

        }

        this.reloadPanel.setMode(ReloadPanel.MODE_ROTATE);

        try {

            sendAllowedCompanyRequest(login, ePassword);

        } catch (Exception e) {

            // Connection to server impossible

            this.reloadPanel.setMode(ReloadPanel.MODE_BLINK);

            e.printStackTrace();

        }

    }

    @Override

    public void actionPerformed(ActionEvent e) {

        if (this.comboCompany.getSelectedItem() == null) {

            return;

        }

        // Connect

        this.setEnabled(false);

        this.reloadPanel.setMode(ReloadPanel.MODE_ROTATE);

        String ePassword = this.encryptedPassword;

        if (ePassword == null) {

            if (this.clearPassword == null) {

                this.clearPassword = "";

            }

            ePassword = encodePassword(this.clearPassword);

        }

        this.company = this.comboCompany.getSelectedItem().getName();

        sendLoginRequest(this.textLogin.getValue(), ePassword, this.comboCompany.getSelectedItem().getId(), this.context);

    }

    @Override

    public void stateChanged(final int oldState, final int newState) {

        if (SwingUtilities.isEventDispatchThread()) {

            throw new IllegalAccessError("Must not be called in EventDispatchThread");

        }

        SwingUtilities.invokeLater(new Runnable() {

            @Override

            public void run() {

                if (newState == CONNECTION_REFUSED || newState == CONNECTION_TIMEOUT) {

                    LoginFrame.this.reloadPanel.setMode(ReloadPanel.MODE_BLINK);

                } else if (newState == CONNECTION_OK) {

                    LoginFrame.this.reloadPanel.setMode(ReloadPanel.MODE_EMPTY);

                } else {

                    LoginFrame.this.reloadPanel.setMode(ReloadPanel.MODE_BLINK);

                    // Connection to server lost

                    // FIXME: autoreconnect after 10s

                }

            }

        });

    }

    @Override

    public void setAllowedCompanies(final List<Company> list) {

        if (SwingUtilities.isEventDispatchThread()) {

            throw new IllegalAccessError("Must not be called in EventDispatchThread");

        }

        SwingUtilities.invokeLater(new Runnable() {

            @Override

            public void run() {

                LoginFrame.this.model.updateFrom(list);

                if (!list.isEmpty()) {

                    LoginFrame.this.comboCompany.setEnabled(true);

                    for (Company company : list) {

                        // Reselect from saved company

                        if (company.getId() == LoginFrame.this.loginProperties.getLastCompanyId()) {

                            LoginFrame.this.comboCompany.setValue(company);

                            break;

                        }

                    }

                    if (LoginFrame.this.comboCompany.getSelectedItem() == null) {

                        LoginFrame.this.comboCompany.setSelectedIndex(0);

                    }

                    LoginFrame.this.buttonConnect.setEnabled(!LoginFrame.this.textLogin.getValue().trim().isEmpty() && LoginFrame.this.comboCompany.getSelectedItem() != null);

                } else {

                    LoginFrame.this.buttonConnect.setEnabled(false);

                }

                LoginFrame.this.reloadPanel.setMode(ReloadPanel.MODE_EMPTY);

            }

        });

    }

    @Override

    public void setEnabled(boolean b) {

        if (b) {

            this.reloadPanel.setMode(ReloadPanel.MODE_EMPTY);

        }

        this.textLogin.getComp().setEnabled(b);

        this.textPassWord.setEnabled(b);

        this.comboCompany.setEnabled(b);

        this.saveCheckBox.setEnabled(b);

        this.buttonConnect.setEnabled(b);

    }

    private final void initLocalization(final String baseName, final List<Locale> toDisplay) {

        if (baseName == null)

            throw new NullPointerException("Null baseName");

        if (this.localeBaseName != null)

            throw new IllegalStateException("Already inited to " + this.localeBaseName);

        this.localeBaseName = baseName;

        this.localesToDisplay.addAll(toDisplay);

        // this.setUILanguage(UserProps.getInstance().getLocale());

        TM.getInstance();

    }

    private void setUILanguage(Locale locale) {

        System.err.println("LoginFrame.setUILanguage():" + this.localeBaseName + " locale:" + locale);

        final ResourceBundle bundle = ResourceBundle.getBundle(this.localeBaseName, locale, TranslationManager.getControl());

        this.loginLabel.setText(bundle.getString("loginLabel"));

        this.passwordLabel.setText(bundle.getString("passwordLabel"));

        this.companyLabel.setText(bundle.getString("companyLabel"));

        this.saveCheckBox.setText(bundle.getString("saveCheckBox"));

        this.buttonConnect.setText(bundle.getString("buttonConnect"));

        this.langButton.setText(locale.getLanguage());

        this.langButton.setVisible(true);

        Locale.setDefault(locale);

    }

    private static MessageDigest md;

    private static synchronized MessageDigest getMessageDigest() {

        if (md == null)

            try {

                md = MessageDigest.getInstance("SHA-1");

            } catch (NoSuchAlgorithmException e) {

                // should be standard

                throw new IllegalStateException("no SHA1", e);

            }

        return md;

    }

    public static String encodePassword(String clearPassword) {

        final byte[] s;

        synchronized (getMessageDigest()) {

            getMessageDigest().reset();

            getMessageDigest().update(clearPassword.getBytes());

            s = getMessageDigest().digest();

        }

        return Base64.encodeBytes(s);

    }

    public String getCompany() {

        return this.company;

    }

    public void saveAndDispose() throws IOException {

        this.loginProperties.setLastCompanyId(this.comboCompany.getSelectedItem().getId());

        this.loginProperties.setLastCompanyName(this.comboCompany.getSelectedItem().getName());

        this.loginProperties.setLastLoginName(this.textLogin.getValue());

        this.loginProperties.store();

        dispose();

    }

    private void sendAllowedCompanyRequest(String login, String passwordHash) {

        this.executorService.execute(new Runnable() {

            @Override

            public void run() {

                LoginFrame.this.client.setCredentials(login, passwordHash);

                final JSONObject params = new JSONObject();

                params.put("login", login);

                try {

                    JSONObject result = (JSONObject) LoginFrame.this.client.rpcCall(LoginFrame.this.serverUrl, "getAllowedCompanies", params);

                    JSONArray array = (JSONArray) result.get("companies");

                    List<Company> list = new ArrayList<>();

                    for (Object o : array) {

                        final Company c = new Company();

                        c.fromJSON((JSONObject) o);

                        list.add(c);

                    }

                    setAllowedCompanies(list);

                } catch (IOException e) {

                    setAllowedCompanies(Collections.emptyList());

                    e.printStackTrace();

                }

            }

        });

    }

    /**

     * @param encodedPassword : Base64 encodded hash of the password

     */

    private void sendLoginRequest(String login, String encodedPassword, int idCompany, final String context) {

        this.executorService.execute(new Runnable() {

            @Override

            public void run() {

                stateChanged(getConnectionState(), CONNECTION_INPROGRESS);

                LoginFrame.this.client.setCredentials(login, encodedPassword);

                final JSONObject params = new JSONObject();

                params.put("company-id", idCompany);

                params.put("context", context);

                try {

                    JSONObject result = (JSONObject) LoginFrame.this.client.rpcCall(LoginFrame.this.serverUrl, "login", params);

                    if (result.getAsString("status").equals("granted")) {

                        LoginFrame.this.idCompany = idCompany;

                        stateChanged(getConnectionState(), CONNECTION_OK);

                    } else {

                        stateChanged(getConnectionState(), CONNECTION_REFUSED);

                    }

                } catch (IOException e) {

                    stateChanged(getConnectionState(), CONNECTION_BROKEN);

                }

            }

        });

    }

    public int getCompanyId() {

        return this.idCompany;

    }

}