mirror of https://github.com/deavminet/skippy.git
281 lines
5.6 KiB
D
281 lines
5.6 KiB
D
import std.stdio;
|
|
import std.socket;
|
|
import libdnet.dclient : DClient;
|
|
import std.string : cmp, split, strip;
|
|
import std.conv : to;
|
|
import notifications;
|
|
import std.file;
|
|
import std.json;
|
|
|
|
JSONValue config;
|
|
|
|
/* Current conneciton */
|
|
DClient dclient;
|
|
|
|
NotificationWatcher dnotifications;
|
|
|
|
Mode currentMode;
|
|
|
|
void main()
|
|
{
|
|
/* If the configuration file exists */
|
|
if(exists("config.example")) /* TODO: Change */
|
|
{
|
|
/* Load the config */
|
|
loadConfig("config.example");
|
|
}
|
|
/* If the configuration file doesn't exist */
|
|
else
|
|
{
|
|
/* Set default config */
|
|
defaultConfig();
|
|
}
|
|
|
|
/* Start the REPL */
|
|
commandLine();
|
|
}
|
|
|
|
void clientAuth(string username, string password)
|
|
{
|
|
if(dclient.auth(username, password))
|
|
{
|
|
writeln("Auth good");
|
|
}
|
|
else
|
|
{
|
|
writeln("Auth bad");
|
|
}
|
|
}
|
|
|
|
public enum Mode
|
|
{
|
|
CHANNEL,
|
|
USER,
|
|
SERVER
|
|
}
|
|
|
|
void commandLine()
|
|
{
|
|
|
|
|
|
/* The current command */
|
|
string commandLine;
|
|
|
|
string currentChannel;
|
|
|
|
while(true)
|
|
{
|
|
/* Read in a command line */
|
|
write(currentChannel~"~> ");
|
|
commandLine = readln();
|
|
|
|
if(cmp(strip(commandLine), "") == 0)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
string[] elements = split(commandLine);
|
|
string command = elements[0];
|
|
|
|
/* If the command is `exit` */
|
|
if(cmp(command, "exit") == 0)
|
|
{
|
|
break;
|
|
}
|
|
/* If the command is `connect` */
|
|
else if(cmp(command, "connect") == 0)
|
|
{
|
|
string address;
|
|
string port;
|
|
Address addr;
|
|
bool isConfigConnect;
|
|
|
|
/* If there is only one argument then it is a server name */
|
|
if(elements.length == 2)
|
|
{
|
|
string serverName = elements[1];
|
|
|
|
try
|
|
{
|
|
/* Get the address and port */
|
|
JSONValue serverInfo = config["servers"][serverName];
|
|
address = serverInfo["address"].str();
|
|
port = serverInfo["port"].str();
|
|
|
|
isConfigConnect = true;
|
|
}
|
|
catch(JSONException e)
|
|
{
|
|
writeln("Could not find server: "~to!(string)(e));
|
|
continue;
|
|
}
|
|
}
|
|
/* Then it must be `<address> <port>` */
|
|
else if(elements.length == 3)
|
|
{
|
|
address = elements[1];
|
|
port = elements[2];
|
|
}
|
|
/* Syntax error */
|
|
else
|
|
{
|
|
writeln("Syntax error");
|
|
continue;
|
|
}
|
|
|
|
/* TODO: How many are rtuend and which to use ? */
|
|
addr = getAddress(address, to!(ushort)(port))[0];
|
|
writeln("Connecting to "~to!(string)(addr)~"...");
|
|
dclient = new DClient(addr);
|
|
dnotifications = new NotificationWatcher(dclient);
|
|
writeln("Connected!");
|
|
|
|
/* Get the message of the day */
|
|
serverGreet();
|
|
|
|
if(isConfigConnect)
|
|
{
|
|
string server = elements[1];
|
|
|
|
string username = config["servers"][server]["auth"]["username"].str();
|
|
string password = config["servers"][server]["auth"]["password"].str();
|
|
|
|
/* Authenticate the user */
|
|
clientAuth(username, password);
|
|
|
|
/* Auto join config */
|
|
configAutoJoin(server);
|
|
}
|
|
}
|
|
/* If the command is `auth` */
|
|
else if(cmp(command, "auth") == 0)
|
|
{
|
|
string username = elements[1];
|
|
string password = elements[2];
|
|
|
|
/* Authenticate the user */
|
|
clientAuth(username, password);
|
|
}
|
|
/* If the command is `list` */
|
|
else if(cmp(command, "list") == 0 || cmp(command, "l") == 0)
|
|
{
|
|
string[] channels = dclient.list();
|
|
writeln("Channels ("~to!(string)(channels.length)~" total)\n");
|
|
foreach(string channel; channels)
|
|
{
|
|
write("\t"~channel~" (");
|
|
ulong memberCount = dclient.getMemberCount(channel);
|
|
writeln(to!(string)(memberCount)~")");
|
|
}
|
|
}
|
|
/* If the command is `names` */
|
|
else if(cmp(command, "names") == 0)
|
|
{
|
|
string[] members = dclient.getMembers(currentChannel);
|
|
writeln(members);
|
|
}
|
|
/* If the command is `join` */
|
|
else if(cmp(command, "join") == 0)
|
|
{
|
|
string[] channels = elements[1..elements.length];
|
|
|
|
foreach(string channel; channels)
|
|
{
|
|
if(dclient.join(channel))
|
|
{
|
|
writeln("Already present in channel "~channel);
|
|
}
|
|
}
|
|
|
|
currentChannel = elements[elements.length-1];
|
|
}
|
|
/* If the command is `query` */
|
|
else if(cmp(command, "query") == 0)
|
|
{
|
|
/* Get the username to query */
|
|
string user = elements[1];
|
|
|
|
/* Set current "channel" to username */
|
|
currentChannel = user;
|
|
|
|
/* Set mode to USER */
|
|
currentMode = Mode.USER;
|
|
}
|
|
else
|
|
{
|
|
/* If the current mode is `channel` then send to channel */
|
|
if(currentMode == Mode.CHANNEL)
|
|
{
|
|
/* Send the current commandLine to the channel */
|
|
dclient.sendMessage(false, currentChannel, strip(commandLine));
|
|
}
|
|
/* If the current mode is `user` then send to the user */
|
|
else if(currentMode == Mode.USER)
|
|
{
|
|
dclient.sendMessage(true, currentChannel, strip(commandLine));
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
if(dclient)
|
|
{
|
|
/* TODO: Fix in tristanable */
|
|
dclient.close();
|
|
}
|
|
|
|
}
|
|
|
|
void serverGreet()
|
|
{
|
|
string motd = dclient.getMotd();
|
|
writeln("!~> "~motd);
|
|
}
|
|
|
|
void configAutoJoin(string server)
|
|
{
|
|
foreach(JSONValue value; config["servers"][server]["channels"].array())
|
|
{
|
|
string channel = value.str();
|
|
//import core.thread;
|
|
//Thread.getThis().sleep(dur!("seconds")(1));
|
|
if(dclient.join(channel))
|
|
{
|
|
writeln("Already present in channel "~channel);
|
|
}
|
|
}
|
|
}
|
|
|
|
void defaultConfig()
|
|
{
|
|
/* Server block */
|
|
JSONValue serverBlock;
|
|
|
|
/* TODO: Remove test servers? */
|
|
JSONValue dserv;
|
|
dserv["address"] = "127.0.0.1";
|
|
dserv["port"] = "7777";
|
|
// JSONValue[] joins = []
|
|
// dserv["joins"] =
|
|
serverBlock["dserv"] = dserv;
|
|
|
|
config["servers"] = serverBlock;
|
|
}
|
|
|
|
void loadConfig(string configPath)
|
|
{
|
|
/* Open the provided configuration file */
|
|
File file;
|
|
file.open(configPath);
|
|
|
|
/* Read the configuration file */
|
|
byte[] buffer;
|
|
buffer.length = file.size();
|
|
buffer = file.rawRead(buffer);
|
|
|
|
/* Close the file */
|
|
file.close();
|
|
|
|
/* Parse the JSON of the configuration file */
|
|
config = parseJSON(cast(string)buffer);
|
|
} |