Gregory Rudolph
4 years ago
4 changed files with 143 additions and 22 deletions
@ -0,0 +1,123 @@ |
|||||||
|
import java.sql.*; |
||||||
|
import java.util.*; |
||||||
|
|
||||||
|
public class SQLDriver { |
||||||
|
protected Connection c = null; |
||||||
|
|
||||||
|
public boolean tryConnect() { |
||||||
|
try { |
||||||
|
Class.forName("org.sqlite.JDBC"); |
||||||
|
c = DriverManager.getConnection("jdbc:sqlite:cards.db"); |
||||||
|
} catch (Exception e) { |
||||||
|
System.err.println(e.getClass().getName() + ": " + e.getMessage()); |
||||||
|
return false; |
||||||
|
} |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
public Card getRandomCard() { |
||||||
|
try { |
||||||
|
c = DriverManager.getConnection("jdbc:sqlite:cards.db"); |
||||||
|
c.setAutoCommit(false); |
||||||
|
|
||||||
|
Statement stmt = c.createStatement(); |
||||||
|
ResultSet rs = stmt.executeQuery("SELECT * FROM CARDS ORDER BY RANDOM() LIMIT 1;"); |
||||||
|
String cardName = ""; |
||||||
|
int manaCost = 0; |
||||||
|
int power = 0; |
||||||
|
int toughness = 0; |
||||||
|
String description = ""; |
||||||
|
String image = ""; |
||||||
|
String creatureType = ""; |
||||||
|
|
||||||
|
while (rs.next()) { |
||||||
|
cardName = rs.getString("CARDNAME"); |
||||||
|
manaCost = rs.getInt("MANACOST"); |
||||||
|
power = rs.getInt("POWER"); |
||||||
|
toughness = rs.getInt("TOUGHNESS"); |
||||||
|
description = rs.getString("DESCRIPTION"); |
||||||
|
image = rs.getString("IMAGE"); |
||||||
|
creatureType = rs.getString("CREATURETYPE"); |
||||||
|
|
||||||
|
} |
||||||
|
rs.close(); |
||||||
|
stmt.close(); |
||||||
|
c.close(); |
||||||
|
Card card = new Card(cardName, manaCost, power, toughness, description, image, creatureType); |
||||||
|
return card; |
||||||
|
} catch (Exception e) { |
||||||
|
System.err.println(e.getClass().getName() + ": " + e.getMessage()); |
||||||
|
System.exit(0); |
||||||
|
} |
||||||
|
System.err.println("Returning null."); |
||||||
|
return null; |
||||||
|
} |
||||||
|
|
||||||
|
public boolean insertCard(Card card) { |
||||||
|
Statement stmt = null; |
||||||
|
try { |
||||||
|
Class.forName("org.sqlite.JDBC"); |
||||||
|
c = DriverManager.getConnection("jdbc:sqlite:cards.db"); |
||||||
|
c.setAutoCommit(false); |
||||||
|
String baseStmt = "INSERT INTO CARDS (CARDNAME,MANACOST,POWER,TOUGHNESS,DESCRIPTION,IMAGE,CREATURETYPE) VALUES "; |
||||||
|
stmt = c.createStatement(); |
||||||
|
String sql = "(\'" + card.cardName + "\'," + card.manaCost + "," + card.power + "," + card.toughness + ", \'" |
||||||
|
+ card.description + "\',\'" + card.image + "\',\'" + card.creatureType + "\');"; |
||||||
|
stmt.executeUpdate(baseStmt + sql); |
||||||
|
stmt.close(); |
||||||
|
c.commit(); |
||||||
|
c.close(); |
||||||
|
} catch (Exception e) { |
||||||
|
System.err.println(e.getClass().getName() + ": " + e.getMessage()); |
||||||
|
return false; |
||||||
|
} |
||||||
|
return true; |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
public boolean setupCards() { |
||||||
|
ArrayList<Card> topdeck = new ArrayList<Card>(); |
||||||
|
topdeck.add(new Card("Devouring Dragon", 5, 5, 6, "He does to people what Greg does to booty.", "", "Dragon")); |
||||||
|
topdeck.add(new Card("Dragon Worshipper", 1, 1, 1, "He spends his days praying to dragons.", "", "Human")); |
||||||
|
topdeck.add(new Card("Dragon Whelp", 2, 3, 2, "He will get there.", "", "Dragon")); |
||||||
|
topdeck.add(new Card("Dragon Egg", 1, 0, 2, "Close to hatching!", "", "Egg")); |
||||||
|
topdeck.add(new Card("Lingering Flame", 3, 4, 3, "He lives among the dragons.", "", "Elemental")); |
||||||
|
topdeck.add( |
||||||
|
new Card("Dragonguard Sentry", 1, 1, 2, "Devoted to the protection of the dragonflight.", "", "Human")); |
||||||
|
topdeck.add(new Card("Withering Flamewitch", 2, 4, 2, "Her flame bites deep.", "", "Human")); |
||||||
|
topdeck.add(new Card("Final Examination", 6, 9000, 0, "Oh dear, Greg.", "", "Parchment")); |
||||||
|
topdeck.add(new Card("Drunken Dragon", 3, 4, 1, "He is trying.", "", "Dragon")); |
||||||
|
topdeck.add(new Card("Isle of Power", -1, 0, 0, "Invoke this for mana.", "", "Land")); |
||||||
|
for (Card card : topdeck) { |
||||||
|
if (!insertCard(card)) { |
||||||
|
return false; |
||||||
|
} |
||||||
|
} |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
public boolean setupTable() { |
||||||
|
Statement stmt = null; |
||||||
|
try { |
||||||
|
Class.forName("org.sqlite.JDBC"); |
||||||
|
c = DriverManager.getConnection("jdbc:sqlite:cards.db"); |
||||||
|
stmt = c.createStatement(); |
||||||
|
String sql = "CREATE TABLE CARDS " + "(ID INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL," |
||||||
|
+ " CARDNAME TEXT NOT NULL, " + " MANACOST INT NOT NULL, " |
||||||
|
+ " POWER INT NOT NULL, " + " TOUGHNESS INT NOT NULL, " |
||||||
|
+ " DESCRIPTION TEXT NOT NULL, " + " IMAGE TEXT NOT NULL, " |
||||||
|
+ " CREATURETYPE TEXT NOT NULL)"; |
||||||
|
stmt.executeUpdate(sql); |
||||||
|
stmt.close(); |
||||||
|
c.close(); |
||||||
|
if (!setupCards()) { |
||||||
|
System.exit(-1); |
||||||
|
} |
||||||
|
} catch (Exception e) { |
||||||
|
if (!e.getMessage().contains("(table CARDS already exists)")) |
||||||
|
System.err.println(e.getClass().getName() + ": " + e.getMessage()); |
||||||
|
return false; |
||||||
|
} |
||||||
|
return true; |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue