package login;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;


public class UserSession {

    private int userID;
    private boolean loggedIn;
    private Connection con;

    public UserSession(){
        userID = 0;
        loggedIn = false;
        con = null;
    }

    public boolean loginUser(int userID) {
        if (this.userID == userID)  {
            return loggedIn;
        }
        return false;
    }

    public boolean isLoggedIn() {
        return loggedIn;
    }

    //get userID from Userdata - if user does not exist or connection
    //isn't working - return <= 0, else return userID
    public int getUserID(String userName, Connection con){

        String userIDquery = "SELECT userID FROM UserData WHERE userName = " + userName + ";";
        int rv = -1;

        try{
            Statement stmt = con.createStatement();
            ResultSet rs = stmt.executeQuery(userIDquery);

            if (rs == null) {
                rv = 0;
            }
            else {
                rv = rs.getInt(1);
            }
            rs.close();
            stmt.close();
        }
        catch(SQLException e){
            System.out.println(e.getMessage());
        }
        return rv;
    }

    public String getUserName() {

        String userNameQuery = "SELECT userName from UserData WHERE userID = " + userID + ";";
        String rv = "";

        try {
            Statement stmt = con.createStatement();
            ResultSet rs = stmt.executeQuery(userNameQuery);

            if (rs == null){
                System.out.println("Error: Probably connection closed.");
            }
            else {
                rv = rs.getString(1);
            }
            rs.close();
            stmt.close();
        }
        catch(SQLException e) {
            System.out.println(e.getMessage());
        }
        return rv;

    }

    public void addNewUser(String username, String password, Connection con) {

        try {

            Statement stmt = con.createStatement();
            ResultSet rs = stmt.executeQuery("SHOW TABLES LIKE 'UserData';");
            if (rs == null) {
                stmt.executeQuery("CREATE TABLE UserData (userName VARCHAR(15), password VARCHAR(20), userID INT PRIMARY_KEY AUTO_INCREMENT);");
            }

            String insertQuery = "INSERT INTO UserData VALUES (" + username + "," + password + ", NULL);";
            stmt.executeQuery(insertQuery);
        }

        catch(SQLException e){
            System.out.println(e.getMessage());
        }
    }

    public void startConnection() {
        try {
            String url = "jdbc:mysql://dbserver.cs.uchicago.edu/laurenellsworth";
            Class.forName("com.mysql.jdbc.Driver");
            con = DriverManager.getConnection(url, "laurenellsworth", "tenletters");
        }
        catch(SQLException e) {
            System.out.println(e.getMessage());
        }
        catch(ClassNotFoundException e){
            System.out.println(e.getMessage());
        }
    }
}