Driver.java

import quizlib.QuizDB;
import java.util.*;
import java.io.Console;
import java.io.IOException;

public class Driver
{
	public static void user_options(User u, String type)
	{
		Scanner sc = new Scanner(System.in);
		if(type.equals("teacher"))
		{
			System.out.println();
			u.display();
			while(true)
			{
				System.out.println("\nTeacher Menu\n------------\n1. set a test\n2. view marks\n3. exit");
				System.out.print("\nEnter choice: ");
				int choice = sc.nextInt();
				switch(choice)
				{
					case 1:
					{
						((Teacher) u).setTest();
					}
					break;

					case 2:
					{
						System.out.println("Available Tests:");
						QuizDB qobj = new QuizDB();
						qobj.runQuery("select * from test");
						qobj.resultDisplay();
						
						System.out.println("\nEnter test id to view marks");
						String id_test = sc.next();
						((Teacher) u).viewMarks(id_test);
					}
					break;
				}
				System.out.println("\nDo you wish to log out?\n  'y' for yes\n  any other key for menu:");
				char exit_choice = sc.next().charAt(0);
				if(exit_choice=='y' || exit_choice=='Y')
					break;
			}
		}
		else if(type.equals("student"))
		{
			System.out.println();
			u.display();
			while(true)
			{
				System.out.println("\nStudent Menu\n------------\n1. attempt a test\n2. view own marks\n3. exit");
				System.out.print("\nEnter choice: ");
				int choice = sc.nextInt();
				switch(choice)
				{
					case 1:
					{
						System.out.println("Available Tests:");
						QuizDB qobj = new QuizDB();
						qobj.runQuery("select * from test");
						qobj.resultDisplay();
						
						System.out.println("Enter test id to attempt");
						String id_test = sc.next();
						((Student) u).attemptTest(id_test);
					}
					break;

					case 2:
					{
						System.out.println("Available Tests:");
						QuizDB qobj = new QuizDB();
						qobj.runQuery("select * from test");
						qobj.resultDisplay();
						
						System.out.println("\nEnter test id to view your marks");
						String id_test = sc.next();
						((Student) u).viewMarks(id_test);
					}
					break;
				}
				System.out.println("\nDo you wish to log out?\n  'y' for yes\n  any other key for menu:");
				char exit_choice = sc.next().charAt(0);
				if(exit_choice=='y' || exit_choice=='Y')
					break;
			}
		}
		else
		{
			System.out.println("No menu for admin.");
		}
	}
	public static void main(String args[])
	{
		System.out.println("############### Quiz Platform ###############");
		Scanner sc = new Scanner(System.in);
		
		while(true)
		{
			User u = null;
			int uid = 1;
			String pwd = "dummy_pwd";
			while (true)
			{
				System.out.println("\nLog in menu\n 1 --> Log in\n 2 --> Sign up");
				int init_choice = sc.nextInt();
				boolean done = false;
				switch (init_choice) {
					case 1: {
						System.out.println("Enter User Id: ");
						uid = sc.nextInt();
						
						Console con = System.console();
						System.out.println("Enter Password: ");
						pwd = sc.nextLine(); //to prevent reading previous newline character.
						pwd = sc.nextLine();
						/*char[] pass_char = con.readPassword();
						pwd = new String(pass_char);*/
						done = User.validate(uid, pwd);
					}
						break;

					case 2: {
						User.signup();
					}
						break;
				}
				if(done)
					break;
			}
			QuizDB qobj = new QuizDB();
			qobj.runQuery("select * from user_list where user_id="+uid);
			ArrayList<ArrayList<String>> result = qobj.getResult();
			ArrayList<String> user_det = result.get(1);
			String type = user_det.get(3);
			if (type.equals("student"))
			{
				u = new Student(Integer.parseInt(user_det.get(0)), user_det.get(1), user_det.get(2));
				user_options(u,"student");
			}
			else if(type.equals("teacher"))
			{
				u = new Teacher(Integer.parseInt(user_det.get(0)), user_det.get(1), user_det.get(2), user_det.get(4));
				user_options(u,"teacher");
			}
			else
			{
				System.out.println("The type of user is admin. No need to login.");
			}
			
			System.out.println("Do you wish to close the program?\n  'y' for yes\n  any other key for log in menu:");
			char exit_choice = sc.next().charAt(0);
			if(exit_choice=='y' || exit_choice=='Y')
				break;
		}
		System.out.println("\n EXITING PROGRAM . . .");
	}
}