gdf

Author: Minecon724

src/main/java/pl/minecon724/bitbot/BitBot.java

@@ -11,28 +11,47 @@
 import net.dv8tion.jda.api.JDABuilder;
 import net.dv8tion.jda.api.interactions.commands.OptionType;
 import net.dv8tion.jda.api.interactions.commands.build.CommandData;
+import net.dv8tion.jda.api.interactions.commands.build.SubcommandData;
 import pl.minecon724.bitbot.commands.Commands;
+import pl.minecon724.bitbot.commands.GiveawayCommand;
 import pl.minecon724.bitbot.listeners.EventReady;
 import pl.minecon724.bitbot.utils.EcoManager;
 import pl.minecon724.bitbot.utils.StatusChange;
 
 public class BitBot {
 	static EcoManager ecoMan;
+	static GiveawayCommand giveaway;
 
 	public static void main(String[] args) throws LoginException, ClassNotFoundException, SQLException {
+		giveaway = new GiveawayCommand(args[2]);
 		//ecoMan = new EcoManager(args[1]);
 		JDA jda = JDABuilder.createDefault(args[0]).build();
 		jda.addEventListener(new EventReady());
-		jda.addEventListener(new Commands());
+		jda.addEventListener(new Commands(giveaway));
 		jda.upsertCommand("help", "Show all commands");
 		CommandData exchangeCmd = new CommandData("exchange", "Exchange assets");
 		exchangeCmd.addOption(OptionType.STRING, "From", "Asset you want to sell");
 		exchangeCmd.addOption(OptionType.STRING, "To", "Asset you want to buy");
 		exchangeCmd.addOption(OptionType.INTEGER, "Quantity", "Quantity of the asset you want to SELL");
 		jda.upsertCommand(exchangeCmd);
 		CommandData marketCmd = new CommandData("market", "View market");
-		marketCmd.addOption(OptionType.STRING, "Asset", "Detailed asset info");
+		marketCmd.addOption(OptionType.STRING, "Asset", "Detailed asset info", false);
+		jda.upsertCommand(marketCmd);
+		jda.upsertCommand("money", "Check account balance");
+		CommandData giveawayCmd = new CommandData("giveaway", "Giveaway commands");
+		SubcommandData gStart = new SubcommandData("start", "Start a giveaway");
+		gStart.addOption(OptionType.INTEGER, "winners", "winner count");
+		gStart.addOption(OptionType.INTEGER, "minutes", "d");
+		gStart.addOption(OptionType.STRING, "prize", "g");
+		giveawayCmd.addSubcommands(gStart);
+		SubcommandData gReroll = new SubcommandData("reroll", "Reroll winner");
+		gReroll.addOption(OptionType.INTEGER, "message id", "Giveaway message id");
+		giveawayCmd.addSubcommands(gReroll);
+		SubcommandData gEnd = new SubcommandData("end", "End giveaway");
+		gEnd.addOption(OptionType.INTEGER, "message id", "Giveaway message id");
+		giveawayCmd.addSubcommands(gEnd);
 		final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
 		scheduler.scheduleAtFixedRate(new StatusChange(jda), 0l, 1l, TimeUnit.MINUTES);
+		scheduler.scheduleAtFixedRate(giveaway.new refresh(), 0l, 1l, TimeUnit.MINUTES);
 	}
 }

src/main/java/pl/minecon724/bitbot/commands/Commands.java

@@ -6,19 +6,25 @@
 public class Commands extends ListenerAdapter {
 	HelpCommand helpcmd = new HelpCommand();
 	MarketCommand marketCmd = new MarketCommand();
+	GiveawayCommand giveawayCmd;
+	public Commands(GiveawayCommand gCmd) {
+		this.giveawayCmd = gCmd;
+	}
 
 	public void onSlashCmd(SlashCommandEvent e) {
 		switch (e.getName().toLowerCase()) {
 		case "help":
 			e.reply(helpcmd.process());
 			break;
-		case "balance": case "bal": case "money":
+		case "money":
 			break;
-		case "exchange": case "buy": case "sell":
+		case "exchange":
 			break;
 		case "market":
 			e.replyEmbeds(marketCmd.process(e.getUser()));
 			break;
+		case "giveaway":
+			break;
 		}
 	}
 }

src/main/java/pl/minecon724/bitbot/commands/GiveawayCommand.java

@@ -0,0 +1,77 @@
+package pl.minecon724.bitbot.commands;
+
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.List;
+
+import pl.minecon724.bitbot.utils.Giveaway;
+
+public class GiveawayCommand {
+	String dbUrl;
+	Connection con;
+	List<Giveaway> queue;
+	public GiveawayCommand(String dbUrl) throws ClassNotFoundException, SQLException {
+		Class.forName("com.mysql.jdbc.Driver");
+		con = DriverManager.getConnection(dbUrl);
+		System.out.println("[Giveaway] Connected to database");
+	}
+	
+	public void end(Giveaway gw) {
+		// TODO
+		return;
+	}
+	
+	public void start(Giveaway gw) {
+		// TODO
+		return;
+	}
+	
+	public void reroll(Giveaway gw) {
+		// TODO
+		return;
+	}
+	
+	public void setupDatabase(Connection con, boolean reset) throws SQLException {
+		Statement stt;
+		if (reset) {
+			stt = con.createStatement();
+			stt.execute("drop table giveaways");
+			stt.close();
+		}
+		stt = con.createStatement();
+		stt.execute("create table giveaways (message int not null, prize text not null, ending int not null, winners int not null, ended tinyint(1) not null");
+		stt.close();
+	}
+	
+	public class refresh implements Runnable {
+		@Override
+		public void run() {
+			Statement stt;
+			ResultSet rs;
+			try {
+				stt = con.createStatement();
+				stt.execute("select * from giveaways");
+				rs = stt.getResultSet();
+				long time = System.currentTimeMillis() / 1000;
+				while (rs.next()) {
+					long ending = rs.getLong("ending");
+					if (time+60 >= ending) {
+						Giveaway gw = new Giveaway(rs.getInt("message"), rs.getString("prize"), ending, rs.getInt("winners"));
+						queue.add(gw);
+					}
+					if (time >= ending) {
+						Giveaway gw = new Giveaway(rs.getInt("message"), rs.getString("prize"), ending, rs.getInt("winners"));
+						end(gw);
+					}
+				}
+				rs.close();
+				stt.close();
+			} catch (SQLException e) {
+				e.printStackTrace();
+			}
+		}
+	}
+}

src/main/java/pl/minecon724/bitbot/utils/Giveaway.java

@@ -0,0 +1,39 @@
+package pl.minecon724.bitbot.utils;
+
+public class Giveaway {
+	int message;
+	String prize;
+	long ending;
+	int winners;
+	boolean ended;
+	public Giveaway(int message, String prize, long ending, int winners) {
+		this.message = message;
+		this.prize = prize;
+		this.ending = ending;
+		this.winners = winners;
+	}
+	public String getPrize() {
+		return prize;
+	}
+	public void setPrize(String prize) {
+		this.prize = prize;
+	}
+	public long getEnding() {
+		return ending;
+	}
+	public void setEnding(long ending) {
+		this.ending = ending;
+	}
+	public int getWinners() {
+		return winners;
+	}
+	public void setWinners(int winners) {
+		this.winners = winners;
+	}
+	public boolean isEnded() {
+		return ended;
+	}
+	public void setEnded(boolean ended) {
+		this.ended = ended;
+	}
+}