It works! Reading in a file, writing to the same file or a different existing file, glancing, perusing, adding and removing cards by number

1 files changed, 178 insertions, 20 deletions

diff --git a/src/eu/equalparts/cardbase/standalone/CardBaseCLI.java b/src/eu/equalparts/cardbase/standalone/CardBaseCLI.java
index 2806a9e..d0503ac 100644
--- a/src/eu/equalparts/cardbase/standalone/CardBaseCLI.java
+++ b/src/eu/equalparts/cardbase/standalone/CardBaseCLI.java
@@ -4,22 +4,32 @@ import java.io.BufferedReader;
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStreamReader;
+import java.net.MalformedURLException;
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.Scanner;
 
 import com.fasterxml.jackson.core.JsonParseException;
 import com.fasterxml.jackson.databind.JsonMappingException;
 
+import eu.equalparts.cardbase.data.Card;
 import eu.equalparts.cardbase.data.CardBaseManager;
+import eu.equalparts.cardbase.data.CardSet;
 import eu.equalparts.cardbase.data.MetaCardSet;
+import eu.equalparts.cardbase.query.IO;
 
 /**
  * This provides a lightweight CLI for interacting with cardbase files. 
  * 
  */
 public class CardBaseCLI {
-	
-	private static String selectedSet = "";
+
+	private static CardSet selectedSet = null;
+	private static HashMap<String, CardSet> setCache = new HashMap<String, CardSet>();
 	private static CardBaseManager cbm;
 	private static boolean exit = false;
+	private static String help = "No help file was found";
+	private static File cardBaseFile;
 
 	/**
 	 * Execute the interface.
@@ -33,18 +43,35 @@ public class CardBaseCLI {
 		try {
 			// construct the cardbase
 			if (args.length > 0) {
-				System.out.println("Building cardbase from " + args[0]);
-				cbm = new CardBaseManager(new File(args[0]));
+				cardBaseFile = new File(args[0]);
+				if (cardBaseFile.exists() && cardBaseFile.isFile() && cardBaseFile.canRead()) {
+					System.out.println("Loading cardbase from " + args[0]);
+					cbm = new CardBaseManager(cardBaseFile);
+				} else {
+					System.out.println(args[0] + " appears to be invalid");
+					System.exit(0);
+				}
 			} else {
-				System.out.println("No cardbase file was provided, initialising a clean cardbase");
-				cbm = new CardBaseManager();
+				//System.out.println("No cardbase file was provided, initialising a clean cardbase");
+				System.out.println("Loading testbase for debugging purposes");
+				cardBaseFile = new File("testbase");
+				cbm = new CardBaseManager(cardBaseFile);
 			}
 
 			// initialise necessary components
 			System.out.println("Fetching card sets from upstream");
 			BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
+			System.out.println("Loading externals");
+			File helpFile = new File("help");
+			if (helpFile.exists() && helpFile.canRead()) {
+				help = new Scanner(helpFile).useDelimiter("\\Z").next();
+			} else {
+				System.out.println("Help file is not available, I hope you know how to use the program!");
+			}
+			
 			while (!exit) {
+				System.out.print(selectedSet == null ? "> " : selectedSet.code + " > ");
 				String rawInput = br.readLine().trim().toLowerCase();
 				String[] commands = rawInput.split("[ \t]+");
 
@@ -64,21 +91,51 @@ public class CardBaseCLI {
 
 	}
 
-	private static void parse(String[] commands) {
+	private static void parse(String[] commands) throws JsonParseException, JsonMappingException, MalformedURLException, IOException {
 		if (commands.length > 0) {
 			switch (commands[0]) {
 			/*
-			 * Exit procedures
+			 * Show help
 			 */
-			case "exit":
-				exit = true;
+			case "help":
+				System.out.println(help);
 				break;
+			
 			/*
-			 * Show help (externalised?) 
+			 * Write current CardBase to file
 			 */
-			case "help":
-				System.out.println("google it");
+			case "write":
+				if (commands.length > 1) {
+					File output = new File(commands[1]);
+					if (output.exists()) {
+						if(output.isFile()) {
+							if (output.canWrite()) {
+								IO.writeCardBase(output, cbm.cardBase);
+								cardBaseFile = output;
+								System.out.println("Cardbase saved to " + output.getAbsolutePath());
+							} else {
+								System.out.println(commands[1] + " cannot be written to, nothing was written");
+							}
+						} else {
+							System.out.println(commands[1] + " is not a file, nothing was written");
+						}
+					} else {
+						System.out.println(commands[1] + " does not exist, nothing was written");
+					}
+				} else if (cardBaseFile != null) {
+					System.out.println("Writing to " + cardBaseFile.getAbsolutePath());
+				} else {
+					System.out.println("Please provide a file name");
+				}
 				break;
+
+			/*
+			 * Exit procedures
+			 */
+			case "exit":
+				exit = true;
+				break;
+
 			/*
 			 * Print a list of valid set codes
 			 */
@@ -87,31 +144,132 @@ public class CardBaseCLI {
 					System.out.println(mcs);
 				}
 				break;
+
 			/*
 			 * Select a set, any card numbers provided will be fetched from that set
 			 */
 			case "set":
 				for (MetaCardSet mcs : cbm.getAllMetaSets()) {
-					if (mcs.getCode().equalsIgnoreCase(commands[1])) {
-						System.out.println("Selected set: " + mcs.getName());
-						selectedSet = mcs.getCode();
+					if (mcs.code.equalsIgnoreCase(commands[1])) {
+						// if the set is cached, use that
+						if (setCache.containsKey(mcs.code)) {
+							selectedSet = setCache.get(mcs.code);
+						} else {
+							selectedSet = IO.getCardSet(mcs.code);
+							setCache.put(mcs.code, selectedSet);
+						}
+						System.out.println("Selected set: " + mcs.name);
 						return;
 					}
 				}
 				System.out.println(commands[1] + " does not correspond to any set (use \"sets\" to see all valid set codes)");
 				break;
+				
 			/*
-			 * Write current CardBase to file
+			 * Print a brief list of the complete cardbase.
 			 */
-			case "write":
+			case "glance":
+				for (Iterator<Card> i = cbm.cardIterator(); i.hasNext();) {
+					printGlance(i.next());
+				}
+				break;
+			
+			/*
+			 * Print a detailed information of a single card or the whole cardbase.
+			 */
+			case "peruse":
+				if (commands.length > 1) {
+					if (selectedSet != null) {
+						Card card = cbm.getCard(selectedSet.code, commands[1]);
+						if (card != null) {
+							printPerusal(card);
+						} else {
+							System.out.println("Card not in cardbase");
+						}
+					} else {
+						System.out.println("Please select a set before perusing a specific card");
+					}
+				} else {
+					for (Iterator<Card> i = cbm.cardIterator(); i.hasNext();) {
+						printPerusal(i.next());
+					}
+				}
+				break;
+				
+			/*
+			 * Remove one or more cards
+			 */
+			case "remove":
+				if (selectedSet != null) {
+					if (commands.length > 1) {
+						Card remove = selectedSet.getCardByNumber(commands[1]);
+						if (remove != null) {
+							Integer count = 1;
+							if (commands.length > 2 && commands[2].matches("[0-9]+")) {
+								count = Integer.valueOf(commands[2]);
+								if (count <= 0) {
+									System.out.println("Can't remove " + count + " cards");
+									return;
+								}
+							}
+							cbm.removeCard(remove, count);
+							System.out.println("Removed " + count + "x " + remove.name);
+						} else {
+							System.out.println(commands[1] + " does not correspond to a card in " + selectedSet.name);
+						}
+					} else {
+						System.out.println("Please specify a card number to remove");
+					}
+				} else {
+					System.out.println("Select a set before removing cards.");
+				}
 				break;
+				
 			/*
-			 * Try to parse a card number and get it from the set
+			 * Add one or more cards
 			 */
 			default:
+				if (selectedSet != null) {
+					Card newCard = selectedSet.getCardByNumber(commands[0]);
+					if (newCard != null) {
+						Integer count = 1;
+						if (commands.length > 1 && commands[1].matches("[0-9]+")) {
+							count = Integer.valueOf(commands[1]);
+							if (count <= 0) {
+								System.out.println("Can't add " + count + " cards");
+								return;
+							}
+						}
+						newCard.setCode = selectedSet.code;
+						cbm.addCard(newCard, count);
+						System.out.println("Added " + count + "x " + newCard.name);
+					} else {
+						System.out.println(commands[0] + " does not correspond to a card in " + selectedSet.name);
+					}
+				} else {
+					System.out.println("Select a set before adding cards.");
+				}
 				break;
 			}
 		}
 	}
-
+	
+	private static void printPerusal(Card card) {
+		printGlance(card);
+		if (card.type != null) System.out.println("\t" + card.type);
+		if (card.manaCost != null) System.out.println("\tCost: " + card.manaCost);
+		if (card.power != null && card.toughness != null) System.out.println("\t" + card.power + "/" + card.toughness);
+		if (card.loyalty != null) System.out.println("\tLoyalty: " + card.loyalty);
+		
+		if (card.text != null) System.out.println("\t" + card.text.replaceAll("\n", "\n\t"));
+		if (card.flavor != null) System.out.println("\t" + card.flavor.replaceAll("\n", "\n\t"));
+		
+		if (card.rarity != null) System.out.println("\t" + card.rarity);
+		if (card.multiverseid != null) System.out.println("\tMID: " + card.multiverseid);
+		if (card.artist != null) System.out.println("\tIllus. " + card.artist);
+	}
+	
+	private static void printGlance(Card card) {
+		System.out.println(String.format("%1$-4d %2$s (%3$s, %4$s)", card.count, card.name, card.setCode, card.number));
+	}
 }