public static void main(String[] args) {
Team baseballTeam = new Team(); // Creates a new Team object
// The following print statement prints the menu
System.out.println(
"A) Add Player "
+ "\n"
+ "G) Get Player"
+ "\n"
+ "L) Get Leader in a stat"
+ "\n"
+ "R) Remove Player"
+ "\n"
+ "P) Print All Players"
+ "\n"
+ "S) Size"
+ "\n"
+ "H) Update Hits"
+ "\n"
+ "E) Update Errors"
+ "\n"
+ "Q) Quit ");
Scanner scan = new Scanner(System.in);
String menuOption = ""; // Creates a variable to hold user input
while (!menuOption.equalsIgnoreCase(
"Q")) // Loop will continue until user inputs "Q" for their menu option
{
System.out.print("\n" + "Select a menu option: "); // Prompts user for menu option
menuOption = scan.nextLine();
/**
* If the selected option is "A", prompt user for name, hits, errors and position. Checks to
* see if hits and errors are greater than or equal to zero. If condition not met, prompts
* user for new input. Attempt to create a player using input. Throws an exception when team
* is full or position is not within valid range.
*/
if (menuOption.equalsIgnoreCase("A")) {
System.out.print("\n" + "Enter the player's name: ");
String name = scan.nextLine();
System.out.print("Enter the number of hits: ");
int hits = scan.nextInt();
while (hits < 0) {
System.out.print("Please enter a value greater than or equal to 0." + "\n");
System.out.print("Enter the number of hits: ");
hits = scan.nextInt();
scan.nextLine();
}
System.out.print("Enter the number of errors: ");
int errors = scan.nextInt();
while (errors < 0) {
System.out.println("Please enter a value greater than or equal to 0." + "\n");
System.out.print("Enter the number of errors: ");
errors = scan.nextInt();
scan.nextLine();
}
System.out.print("Enter the position: ");
int position = scan.nextInt();
scan.nextLine();
try {
baseballTeam.addPlayer(new Player(name, hits, errors), position);
System.out.println("Player added: " + baseballTeam.getPlayer(position));
} catch (FullTeamException e) {
System.out.println(e.getMessage());
} catch (IllegalArgumentException e) {
System.out.println(e.getMessage());
}
}
/**
* Prints the player in the position requested, prompts user for position. Throws exception
* when position not within valid range.
*/
if (menuOption.equalsIgnoreCase("G")) {
System.out.print("Enter the position of the Player requested:");
int position = scan.nextInt();
scan.nextLine();
try {
System.out.println(baseballTeam.getPlayer(position));
} catch (IllegalArgumentException e) {
System.out.println(e.getMessage());
}
}
/**
* Prints the player who has the best score in a given statistic, prompts user for statistic
* Throws exception when user requests stat other than "hits" or "errors" Throws exception
* when there are no players in the Team.
*/
if (menuOption.equalsIgnoreCase("L")) {
System.out.print("Enter the stat:");
String stat = scan.nextLine();
try {
if (stat.equalsIgnoreCase("hits"))
System.out.println("Leader in hits: " + baseballTeam.getLeader(stat).toString());
else System.out.println("Leader in errors: " + baseballTeam.getLeader(stat).toString());
} catch (IllegalArgumentException e) {
System.out.println(e.getMessage());
} catch (NullPointerException e) {
System.out.println("There are currently no players in the team.");
}
}
/**
* Removes a player from a given position. Prompts user for the position. Throws exception
* when position requested is not in valid range.
*/
if (menuOption.equalsIgnoreCase("R")) {
System.out.print("Enter the position: ");
int position = scan.nextInt();
scan.nextLine();
try {
if (baseballTeam.getPlayer(position) == null)
System.out.println("No player exists at that position");
else {
baseballTeam.removePlayer(position);
System.out.println("Player removed at position " + position);
}
} catch (IllegalArgumentException e) {
System.out.println(e.getMessage());
}
}
/** Prints all the players in the team and their stats in a nice format. */
if (menuOption.equalsIgnoreCase("P")) {
System.out.print("\n");
baseballTeam.PrintAllPlayers();
}
/** Prints the number of players currently in the team. */
if (menuOption.equalsIgnoreCase("S")) {
System.out.println("There are " + baseballTeam.size() + " player(s) in the current Team");
}
/**
* Updates the number of hits of a player. Prompts user for player's name. Throws exception
* when player nonexistent.
*/
if (menuOption.equalsIgnoreCase("H")) {
System.out.print("\n" + "Enter the player to update: ");
String name = scan.nextLine();
System.out.print("Enter the new number of hits:");
int hits = scan.nextInt();
scan.nextLine();
while (hits < 0) {
System.out.print("Please enter a value greater than or equal to 0");
System.out.print("\n" + "Enter the new number of hits:");
hits = scan.nextInt();
scan.nextLine();
}
try {
baseballTeam.getPlayerByName(name).setHits(hits);
System.out.println("Updated " + name + " hits");
} catch (NullPointerException e) {
System.out.println("The player is not currently in the Team");
}
}
/**
* Updates the number of errors of a player. Prompts user for player's name. Throws exception
* when player nonexistent.
*/
if (menuOption.equalsIgnoreCase("E")) {
System.out.print("\n" + "Enter the player to update: ");
String name = scan.nextLine();
System.out.print("Enter the new number of errors:");
int errors = scan.nextInt();
scan.nextLine();
while (errors < 0) {
System.out.print("Please enter a value greater than or equal to 0");
System.out.print("\n" + "Enter the new number of errors:");
errors = scan.nextInt();
scan.nextLine();
}
try {
baseballTeam.getPlayerByName(name).setErrors(errors);
System.out.println("Updated " + name + " errors");
} catch (NullPointerException e) {
System.out.println("The player is not currently in the Team");
}
}
}
scan.close();
System.out.println("Program terminated.");
}
/** @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse response) */
protected void doGet(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
// TODO Auto-generated method stub
System.out.println("entered submit servlet");
request.setAttribute("errordash", "");
HttpSession session = request.getSession(false);
if (session == null) {
request.setAttribute("error", "Please login to continue");
request.getRequestDispatcher("Login").forward(request, response);
return;
}
String formType = null;
if (request.getParameterMap().containsKey("formType")) {
System.out.println("formtype was set");
formType = request.getParameter("formType");
} else {
request.getRequestDispatcher("Login").forward(request, response);
return;
}
System.out.println(formType);
switch (formType) {
case "addUser":
User.addUser(
request.getParameter("name"),
request.getParameter("email"),
request.getParameter("password"),
request.getParameter("phone"),
request.getParameter("address"));
request.setAttribute("error", "Registered successfully! Please login");
request.getRequestDispatcher("Login").forward(request, response);
return;
case "createSubtask":
String[] temp = request.getParameterValues("assignedTo");
ArrayList<Integer> userList = new ArrayList<Integer>();
for (int i = 0; i < temp.length; ++i) {
userList.add(Integer.parseInt(temp[i]));
}
java.util.Date utilDate = new java.util.Date();
java.sql.Date sqlDate = new java.sql.Date(utilDate.getTime());
if (sqlDate.compareTo(Date.valueOf(request.getParameter("deadline"))) > 0) {
request.setAttribute("errordash", "The deadline can't be in the past");
request.getRequestDispatcher("Login").forward(request, response);
return;
}
Task.createSubTask(
userList,
request.getParameter("title"),
Date.valueOf(request.getParameter("deadline")),
Integer.parseInt(request.getParameter("task_id")),
Integer.parseInt(session.getAttribute("user").toString()),
request.getParameter("description"));
request.getRequestDispatcher("Login").forward(request, response);
return;
case "extendDeadline":
utilDate = new java.util.Date();
sqlDate = new java.sql.Date(utilDate.getTime());
if (sqlDate.compareTo(Date.valueOf(request.getParameter("deadline"))) > 0) {
request.setAttribute("errordash", "The deadline can't be in the past");
request.getRequestDispatcher("Login").forward(request, response);
return;
}
Task.extendDeadline(
Integer.parseInt(request.getParameter("task_id")),
Date.valueOf(request.getParameter("deadline")));
request.getRequestDispatcher("Login").forward(request, response);
return;
case "markAsDone":
Task.markAsDone(
Integer.parseInt(request.getParameter("task_id")), request.getParameter("remarks"));
request.getRequestDispatcher("Login").forward(request, response);
return;
case "createTask":
String[] temp4 = request.getParameterValues("assignedTo");
ArrayList<Integer> userList4 = new ArrayList<Integer>();
for (int i = 0; i < temp4.length; ++i) {
userList4.add(Integer.parseInt(temp4[i]));
}
if (!Team.checkMembers(userList4, Integer.parseInt(request.getParameter("team_id")))) {
request.setAttribute("errordash", "The Users selected must be of the selected Team");
request.getRequestDispatcher("Login").forward(request, response);
return;
}
utilDate = new java.util.Date();
sqlDate = new java.sql.Date(utilDate.getTime());
if (sqlDate.compareTo(Date.valueOf(request.getParameter("deadline"))) > 0) {
request.setAttribute("errordash", "The deadline can't be in the past");
request.getRequestDispatcher("Login").forward(request, response);
return;
}
Task.createTask(
userList4,
request.getParameter("title"),
Date.valueOf(request.getParameter("deadline")),
Integer.parseInt(session.getAttribute("user").toString()),
request.getParameter("description"),
Integer.parseInt(request.getParameter("team_id")));
request.getRequestDispatcher("Login").forward(request, response);
return;
case "deleteUser":
String[] temp1 = request.getParameterValues("userID");
for (int i = 0; i < temp1.length; ++i) {
User.deleteUser(Integer.parseInt(temp1[i]));
}
request.getRequestDispatcher("Login").forward(request, response);
return;
case "makeTeam":
Team.makeTeam(
request.getParameter("team_name"), Integer.parseInt(request.getParameter("leader_id")));
request.getRequestDispatcher("Login").forward(request, response);
return;
case "changeTeamLeader":
ArrayList<Integer> tempList = new ArrayList<Integer>();
tempList.add(Integer.parseInt(request.getParameter("leader_id")));
if (!Team.checkMembers(tempList, Integer.parseInt(request.getParameter("team_id")))) {
request.setAttribute(
"errordash", "The new leader selected must belong to the selected Team");
request.getRequestDispatcher("Login").forward(request, response);
return;
}
Team.changeLeader(
Integer.parseInt(request.getParameter("team_id")),
Integer.parseInt(request.getParameter("leader_id")));
request.getRequestDispatcher("Login").forward(request, response);
return;
case "removeMember":
ArrayList<Integer> tempList1 = new ArrayList<Integer>();
tempList1.add(Integer.parseInt(request.getParameter("user_id")));
if (!Team.checkMembers(tempList1, Integer.parseInt(request.getParameter("team_id")))) {
request.setAttribute("errordash", "The User selected must belong to the selected Team");
request.getRequestDispatcher("Login").forward(request, response);
return;
}
if (Integer.parseInt(request.getParameter("user_id"))
== Team.getLeader(Integer.parseInt(request.getParameter("team_id")))) {
request.setAttribute(
"errordash",
"The User selected is leader of the selected Team, Please change the leader before removing");
request.getRequestDispatcher("Login").forward(request, response);
return;
}
Team.deleteMember(
Integer.parseInt(request.getParameter("team_id")),
Integer.parseInt(request.getParameter("user_id")));
request.getRequestDispatcher("Login").forward(request, response);
return;
case "exitTeam":
if (Integer.parseInt(session.getAttribute("user").toString())
== Team.getLeader(Integer.parseInt(request.getParameter("team_id")))) {
request.setAttribute(
"errordash",
"You are leader of the selected Team, Please assign a new leader before exiting");
request.getRequestDispatcher("Login").forward(request, response);
return;
}
Team.deleteMember(
Integer.parseInt(request.getParameter("team_id")),
Integer.parseInt(session.getAttribute("user").toString()));
request.getRequestDispatcher("Login").forward(request, response);
return;
case "deleteTeam":
String[] temp2 = request.getParameterValues("team_id");
for (int i = 0; i < temp2.length; ++i) {
Team.deleteTeam(Integer.parseInt(temp2[i]));
}
request.getRequestDispatcher("Login").forward(request, response);
return;
case "addMemberToTeam":
String[] temp3 = request.getParameterValues("user_id");
for (int i = 0; i < temp3.length; ++i) {
Team.addMember(
Integer.parseInt(temp3[i]), Integer.parseInt(request.getParameter("team_id")));
}
request.getRequestDispatcher("Login").forward(request, response);
return;
case "fileUpload":
// System.out.println("file upload");
// System.out.println(request.getParameter("file").toString());
Part filePart = request.getPart("file");
// System.out.println("got the part");
String fileName = getSubmittedFileName(filePart);
// System.out.println(fileName);
InputStream fileContent = filePart.getInputStream();
int resource_id =
Resource.addResource(
fileName, fileContent, Integer.parseInt(request.getParameter("task_id")));
// String sRootPath = new File("").getAbsolutePath() + "/" + fileName;
// FileOutputStream os = new FileOutputStream(System.getProperty("user.dir")+"/"+1);
// System.out.println(sRootPath);
// int b = 0;
// while ((b = fileContent.read()) != -1)
// {
// os.write(b);
// }
fileContent.close();
request.getRequestDispatcher("Login").forward(request, response);
return;
case "manageFiles":
switch (request.getParameter("resourceSubmit")) {
case "download":
if (request.getParameter("resource_id") == null) {
break;
}
InputStream is =
Resource.getResource(Integer.parseInt(request.getParameter("resource_id")));
String fileName1 =
Resource.getResourceName(Integer.parseInt(request.getParameter("resource_id")));
System.out.println("filename1 " + fileName1);
int fileLength = is.available();
// System.out.println(is.available());
// System.out.println(resource_id);
ServletContext context = getServletContext();
// sets MIME type for the file download
String mimeType = context.getMimeType(fileName1);
if (mimeType == null) {
mimeType = "application/octet-stream";
}
// set content properties and header attributes for the response
response.setContentType(mimeType);
response.setContentLength(fileLength);
String headerKey = "Content-Disposition";
String headerValue = String.format("attachment; filename=\"%s\"", fileName1);
response.setHeader(headerKey, headerValue);
// writes the file to the client
OutputStream outStream = response.getOutputStream();
byte[] buffer = new byte[BUFFER_SIZE];
int bytesRead = -1;
while ((bytesRead = is.read(buffer)) != -1) {
outStream.write(buffer, 0, bytesRead);
// System.out.println(buffer+"buffer");
}
// inputStream.close();
// is.close();
outStream.close();
is.close();
request.getRequestDispatcher("Login"); // .forward(request, response);
return;
case "share":
if (request.getParameter("share_task_id") == null
|| request.getParameter("resource_id") == null) {
break;
}
Resource.addResourceAssign(
Integer.parseInt(request.getParameter("resource_id")),
Integer.parseInt(request.getParameter("share_task_id")));
request.getRequestDispatcher("Login").forward(request, response);
return;
case "remove":
if (request.getParameter("resource_id") == null) {
break;
}
System.out.println("in remove resource");
Resource.removeResourceAssign(
Integer.parseInt(request.getParameter("resource_id")),
Integer.parseInt(request.getParameter("task_id")));
request.getRequestDispatcher("Login").forward(request, response);
return;
}
;
}
System.out.println("reached the end, to be forwarded to login servlet!");
request.getRequestDispatcher("Login").forward(request, response);
return;
}