-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
50f0c98
commit f64175c
Showing
3 changed files
with
88 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
8 changes: 8 additions & 0 deletions
8
src/main/java/net/arptell/dhcpcord/exceptions/DHCPConnectException.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,8 @@ | ||
package net.arptell.dhcpcord.exceptions; | ||
|
||
public class DHCPConnectException extends Exception { | ||
|
||
public DHCPConnectException(String reason) { | ||
super(reason); | ||
} | ||
} |
70 changes: 70 additions & 0 deletions
70
src/main/java/net/arptell/dhcpcord/handlers/DHCPConnectionHandler.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,70 @@ | ||
package net.arptell.dhcpcord.handlers; | ||
|
||
import net.arptell.dhcpcord.exceptions.*; | ||
|
||
import java.util.HashMap; | ||
import java.io.BufferedReader; | ||
import java.io.IOException; | ||
import java.io.InputStreamReader; | ||
import java.io.PrintWriter; | ||
import java.net.*; | ||
|
||
import net.dv8tion.jda.core.entities.*; | ||
|
||
public class DHCPConnectionHandler { | ||
|
||
private HashMap<Member, String> ipCache = new HashMap<>(); | ||
private Socket conn = null; | ||
private PrintWriter output = null; | ||
private BufferedReader input = null; | ||
|
||
protected DHCPConnectionHandler(String ip, int port) throws DHCPConnectException { | ||
try { | ||
conn = new Socket(ip, port); | ||
output = new PrintWriter(conn.getOutputStream(), true); | ||
input = new BufferedReader(new InputStreamReader(conn.getInputStream())); | ||
} | ||
catch(Exception e) { | ||
throw new DHCPConnectException(e.getMessage()); | ||
} | ||
} | ||
private String makeRequest(String request) { | ||
if(conn.isClosed()) { | ||
throw new IllegalStateException("Already disconnected from the server!"); | ||
} | ||
try { | ||
output.print(request); | ||
return input.readLine(); | ||
} | ||
catch(IOException e) {} | ||
return null; | ||
} | ||
public String getIp(Member member) { | ||
String ip = ipCache.get(member); | ||
if(ip == null) { | ||
ip = makeRequest("GET IP " + member.getGuild().getId() + " " + member.getUser().getId()); | ||
ipCache.put(member, ip); | ||
} | ||
return ip; | ||
} | ||
public String assignIp(Member member) { | ||
String ip = makeRequest("ASSIGN IP " + member.getGuild().getId() + " " + member.getUser().getId()); | ||
ipCache.put(member, ip); | ||
return ip; | ||
} | ||
public boolean setIp(Member member, String ip) { | ||
boolean success = Boolean.parseBoolean(makeRequest("SET IP " + member.getGuild().getId() + " " + member.getUser().getId() + " " + ip)); | ||
if(success) { | ||
ipCache.put(member, ip); | ||
} | ||
return success; | ||
} | ||
public void close() throws IOException{ | ||
if(conn.isClosed()) { | ||
throw new IllegalStateException("Already disconnected from the server!"); | ||
} | ||
input.close(); | ||
output.close(); | ||
conn.close(); | ||
} | ||
} |