/**
*額外加了個可以添加圖書的功能，
*不過有exception之后用戶要重新輸入所有內容，有點累贅。
*/
package?com.imooc.librarysystem;

import?com.imooc.librarysystem.exception.BookExistException;
import?com.imooc.librarysystem.exception.BookNotExistException;
import?com.imooc.librarysystem.exception.WrongCommandException;

import?java.util.ArrayList;
import?java.util.List;
import?java.util.Scanner;

public?class?Application?{
????public?static?List<Book>?bookList?=?new?ArrayList<Book>();

????public?List<String>?addBookInfo(Scanner?scanner){
????????List<String>?info?=?new?ArrayList<String>();
????????System.out.print("Please?insert?the?book?name:?");
????????String?name?=?scanner.next();
????????System.out.print("Please?insert?the?book?code:?");
????????String?code?=?scanner.next();
????????info.add(name);
????????info.add(code);
????????return?info;
????}

????public?List<String>?searchBookInfo(Scanner?scanner)?throws?WrongCommandException?{
????????System.out.print("Search?by?1==>Book?name;?2==>?Book?code?:?");
????????String?command?=?scanner.next();
????????List<String>?info?=?new?ArrayList<String>();
????????switch?(command){
????????????case?"1":
????????????????System.out.print("Please?insert?the?book?name:?");
????????????????info.add(command);
????????????????info.add(scanner.next());
????????????????break;
????????????case?"2":
????????????????System.out.print("Please?insert?the?book?code:?");
????????????????info.add(command);
????????????????info.add(scanner.next());
????????????????break;
????????????default:
????????????????throw?new?WrongCommandException("You?have?inserted?a?wrong?command.");
????????}
????????return?info;
????}

????public?Book?addBook(String?name,?String?code)?throws?BookExistException,?WrongCommandException{
????????for(Book?book:bookList){
????????????if(book.getName().equals(name)){
????????????????throw?new?BookExistException("Already?exist?a?book?named:?"+name);
????????????}else?if(book.getCode().equals(code)){
????????????????throw?new?BookExistException("The?book?code?is?already?used.");
????????????}
????????}
????????try?{
????????????Integer.parseInt(code);
????????????Book?book?=?new?Book();
????????????book.setName(name);
????????????book.setCode(code);
????????????return?book;
????????}?catch(NumberFormatException?e){
????????????throw?new?WrongCommandException("Book?code?should?be?numerical.");
????????}
????}

????public?Book?searchBookByName(String?name)?throws?BookNotExistException{
????????for(Book?book:bookList){
????????????if(book.getName().equals(name)){
????????????????return?book;
????????????}
????????}
????????throw?new?BookNotExistException("There?is?no?book?named:?"+name);
????}

????public?Book?searchBookByCode(String?code)?throws?BookNotExistException,WrongCommandException{
????????try{
????????????Integer.parseInt(code);
????????????for(Book?book:bookList){
????????????????if(book.getCode().equals(code)){
????????????????????return?book;
????????????????}
????????????}
????????????throw?new?BookNotExistException("There?is?no?book?has?code?"+code);
????????}catch?(NumberFormatException?e){
????????????throw?new?WrongCommandException("Book?code?should?be?numerical.");
????????}
????}

????public?static?void?main(String[]?args)?{
????????Application?app?=?new?Application();
????????Scanner?scanner?=?new?Scanner(System.in);
????????Boolean?quit?=?false;
????????Book?book;
????????while(!quit){
????????????System.out.print("Press?1==>Add?book;?2==>Search?book?;?0==>Quit?:");
????????????String?choice?=?scanner.next();
????????????switch?(choice)?{
????????????????case?"1":
????????????????????List<String>?info?=?app.addBookInfo(scanner);
????????????????????try?{
????????????????????????book?=?app.addBook(info.get(0),?info.get(1));
????????????????????????bookList.add(book);
????????????????????????System.out.println("You?have?added?a?book:?"?+?info.get(0));
????????????????????}?catch?(BookExistException?e)?{
????????????????????????System.out.println(e.getMessage());
????????????????????????System.out.println("Please?reinsert?a?book?info.");
????????????????????}?catch?(WrongCommandException?e){
????????????????????????System.out.println(e.getMessage());
????????????????????????System.out.println("Please?reinsert?a?book?info.");
????????????????????}?catch?(Exception?e){
????????????????????????e.printStackTrace();
????????????????????}
????????????????????break;
????????????????case?"2":
????????????????????try?{
????????????????????????List<String>?info2?=?app.searchBookInfo(scanner);
????????????????????????if?(info2.get(0).equals("1")){
????????????????????????????book?=?app.searchBookByName(info2.get(1));
????????????????????????}else{
????????????????????????????book?=?app.searchBookByCode(info2.get(1));
????????????????????????}
????????????????????????System.out.println("Congratulations!?Found?book?:"+book.getName());
????????????????????}?catch?(WrongCommandException?e)?{
????????????????????????System.out.println(e.getMessage());
????????????????????????System.out.println("Please?insert?a?correct?command.");
????????????????????}?catch?(BookNotExistException?e){
????????????????????????System.out.println(e.getMessage());
????????????????????????System.out.println("Sorry.");
????????????????????}?catch?(Exception?e){
????????????????????????e.printStackTrace();
????????????????????}
????????????????????break;
????????????????case?"0":
????????????????????System.out.println("Thank?you?for?using?the?book?system.");
????????????????????quit?=?true;
????????????????????break;
????????????????default:
????????????????????System.out.println("You?have?inserted?a?wrong?command.");
????????????????????break;
????????????}
????????}
????}
}

package?com.imooc.librarysystem.exception;

public?class?BookExistException?extends?Exception{
????public?BookExistException(String?message)?{
????????super(message);
????}
}

package?com.imooc.librarysystem.exception;

public?class?BookNotExistException?extends?Exception{
????public?BookNotExistException(String?message)?{
????????super(message);
????}
}

package?com.imooc.librarysystem.exception;

public?class?WrongCommandException?extends?Exception{

????public?WrongCommandException(String?message)?{
????????super(message);
????}
}

package?com.imooc.librarysystem;

public?class?Book?{
????private?String?name;
????private?String?code;

????public?String?getName()?{
????????return?name;
????}

????public?void?setName(String?name)?{
????????this.name?=?name;
????}

????public?String?getCode()?{
????????return?code;
????}

????public?void?setCode(String?code)?{
????????this.code?=?code;
????}
}