package jabadot;

import java.util.*;
import java.sql.*;
import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
import javax.swing.table.*;

/** A User Database Administrator program
 * This does NOT use the UserDB interface as it needs
 * to be able to do ANYTHING to the database,
 * to go beyond, and to repair any errors introduced
 * by bugs in the UserDB code and/or queries. :-)
 *
 * If using InstantDB, therefore, you MUST NOT RUN THIS PROGRAM
 * while users have access to the system, or the database will
 * get worse instead of better!
 */
public class JDAdmin extends JFrame {

    /** the list of users */
    protected ArrayList userList = new ArrayList(  );
    /** The database connection */
    protected Connection conn;
    /** A Statement for listing users */
    protected PreparedStatement listUsersStatement;
    /** A Statement for deleting users */
    protected PreparedStatement deleteUserStatement;
    /** A Statement for resetting passwords for forgetful users */
    protected PreparedStatement setPasswordStatement;

    /** The main table */
    protected JTable theTable;

    /** Main program -- driver */
    public static void main(String av[]) throws Exception {
        JDAdmin aFrame = new JDAdmin(  );
        aFrame.populate(  );
        // aFrame.pack(  );
        aFrame.setSize(600,450);
        aFrame.setVisible(true);
    }

    /** Constructor */
    public JDAdmin(  ) throws SQLException {
        super("JabaDotAdmin");

        // INIT THE DB 
        // Do this before the GUI, since JDBC does more-delayed
        // type checking than Swing...
        
        String dbDriver = JDConstants.getProperty("jabadot.userdb.driver");
        try {
            Class.forName(dbDriver);
        } catch (ClassNotFoundException ex) {
            JOptionPane.showMessageDialog(this,
                "JDBC Driver Failure:
" + ex, "Error",
                JOptionPane.ERROR_MESSAGE);
        }
        conn = DriverManager.getConnection(
            JDConstants.getProperty("jabadot.userdb.url"));
        listUsersStatement = conn.prepareStatement("select * from userdb");
        deleteUserStatement = 
            conn.prepareStatement("delete from userdb where name = ?");
        setPasswordStatement = conn.prepareStatement(
            "update userdb SET password = ? where name = ?");

        // INIT THE GUI
        Container cp = getContentPane(  );
        cp.setLayout(new BorderLayout(  ));
        cp.add(new JScrollPane(theTable = new JTable(new MyTableModel(  ))),
            BorderLayout.CENTER);
        JPanel bp = new JPanel(  );
        JButton x;
        bp.add(x = new JButton("Delete"));
        x.addActionListener(new ActionListener(  ) {
            public void actionPerformed(ActionEvent ex) {
                    int r = theTable.getSelectedRow(  );
                    if (r == -1) {
                        JOptionPane.showMessageDialog(JDAdmin.this,
                            "Please select a user to delete", "Error", 
                            JOptionPane.ERROR_MESSAGE);
                        return;
                    }
                    int i = JOptionPane.showConfirmDialog(JDAdmin.this,
                            "Really delete user?", "Confirm", 
                            JOptionPane.YES_NO_OPTION);
                    switch(i) {
                        case 0:
                            try {
                                delete(r);
                            } catch (SQLException e) {
                                JOptionPane.showMessageDialog(JDAdmin.this,
                                "SQL Error:
" + e, "Error", 
                                JOptionPane.ERROR_MESSAGE);
                            }
                            break;
                        case 1:
                            // nothing to do.
                            break;
                        default:
                            System.err.println("showConfirm: unex ret " + i);
                    }
            }
        });
        bp.add(x = new JButton("List"));
        x.addActionListener(new ActionListener(  ) {
            public void actionPerformed(ActionEvent ex) {
                try {
                    populate(  );
                } catch (SQLException e) {
                    JOptionPane.showMessageDialog(JDAdmin.this,
                    "SQL Error:
" + e, "Error", 
                    JOptionPane.ERROR_MESSAGE);
                }
            }
        });
        bp.add(x = new JButton("Exit"));
        x.addActionListener(new ActionListener(  ) {
            public void actionPerformed(ActionEvent ex) {
                System.exit(0);
            }
        });
    
        cp.add(bp, BorderLayout.SOUTH);

    }

    /** Get the current list of users from the database
     * into the ArrayList, so the display will be up-to-date
     * after any major change.
     */
    public void populate(  ) throws SQLException {
        ResultSet rs = listUsersStatement.executeQuery(  );
        userList.clear(  );
        while (rs.next(  )) {
            String nick = rs.getString(1);
            // System.out.println("Adding " + nick);
            User u = new User(nick, rs.getString(UserDB.PASSWORD),
                rs.getString(UserDB.FULLNAME),
                rs.getString(UserDB.EMAIL), 
                rs.getString(UserDB.CITY),
                rs.getString(UserDB.PROVINCE),
                rs.getString(UserDB.COUNTRY),
                rs.getInt(UserDB.PRIVS));
            userList.add(u);
        }
        rs.close(  );
        theTable.repaint(  );
    }

    /** Delete the given user, by row number 
     * (row number in the display == index into the ArrayList).
     * Use a JDBC PreparedStatement; if it succeeds, then also
     * remove the user object from the ArrayList.
     */
    public void delete(int x) throws SQLException {
        User u = (User)userList.get(x);
        String nick = u.getName(  );
        deleteUserStatement.setString(1, nick);
        int n;
        switch (n = deleteUserStatement.executeUpdate(  )) {
            case 0:
                // no match!
                JOptionPane.showMessageDialog(this,
                    "No match for user " + nick, "Error",
                    JOptionPane.ERROR_MESSAGE);
                break;
            case 1:
                // OK
                JOptionPane.showMessageDialog(this,
                    "User " + nick + " deleted.", "Done",
                    JOptionPane.INFORMATION_MESSAGE);
                userList.remove(x);
                break;
            default:
                // Ulp! Deleted too many! -- n
                JOptionPane.showMessageDialog(this,
                    "Oops, we deleted " + n + " users!!", "Error",
                    JOptionPane.ERROR_MESSAGE);
        }
        theTable.repaint(  );
    }

    // class extends TableModel...
    class MyTableModel extends AbstractTableModel {

        /** Returns the number of items in the list. */
        public int getRowCount(  )  {
            return userList.size(  );
        }

        /** Return the width of the table */
        public int getColumnCount(  ) {
            return 8;
        }

        /** Get the name of a given column */
        public String getColumnName(int i) {
            switch(i) {
            case UserDB.NAME-1:        return "Nickname";
            case UserDB.PASSWORD-1:    return "Password";
            case UserDB.FULLNAME-1:    return "Full Name";
            case UserDB.EMAIL-1:    return "Email";
            case UserDB.CITY-1:        return "City";
            case UserDB.PROVINCE-1:    return "Province";
            case UserDB.COUNTRY-1:    return "Country";
            case UserDB.PRIVS-1:    return "Privs";
            default: return "??";
            }
        }

        /** Returns a data value for the cell at columnIndex and rowIndex.
         * MUST BE IN SAME ORDER as setValueAt(  );
         */
        public Object getValueAt(int row, int col)  {
            User u = (User) userList.get(row);
            switch (col) {
            case UserDB.NAME-1:     return u.getName(  );
            case UserDB.PASSWORD-1: return u.getPassword(  );
            case UserDB.FULLNAME-1: return u.getFullName(  );
            case UserDB.EMAIL-1:     return u.getEmail(  );
            case UserDB.CITY-1:     return u.getCity(  );
            case UserDB.PROVINCE-1:    return u.getProv(  );
            case UserDB.COUNTRY-1:    return u.getCountry(  );
            case UserDB.PRIVS-1:    return new Integer(u.getPrivs(  ));
            default: return null;
            }
        }

        /** Set a value in a cell. MUSE BE IN SAME ORDER AS getValueAt. */
        public void setValueAt(Object val, int row, int col)  {
            User u = (User) userList.get(row);
            switch (col) {
            // DB Schemas start at one, Java columns at zero.
            case UserDB.PASSWORD-1:
                String newPass = (String)val;        // Get new value
                try {
                    setPasswordStatement.setString(1, newPass);        // ready,
                    setPasswordStatement.setString(2, u.getName(  ));    // steady,
                    setPasswordStatement.executeUpdate(  );        // and update!
                } catch (SQLException ex) {
                    JOptionPane.showMessageDialog(null,
                        "SQL Error:
" + ex.toString(  ), "SQL Error",
                        JOptionPane.ERROR_MESSAGE);
                    break;
                }
                u.setPassword(newPass);    // bypassed if DB update failed
                break;

            // Only password cells are editable.
            default:
                JOptionPane.showMessageDialog(null,
                    "setValueAt" + val.getClass(  ) + "," + val, "Logic error",
                    JOptionPane.ERROR_MESSAGE);
                break;
            }
        }

        /** Only password cells are editable. */
        public boolean isCellEditable(int rowIndex, int columnIndex) {
            return columnIndex == UserDB.PASSWORD-1;
        }
    }
}