/*
* SONEWS News Server
* see AUTHORS for the list of contributors
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.sonews.daemon.command;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.util.List;
import org.sonews.config.Config;
import org.sonews.daemon.NNTPConnection;
import org.sonews.feed.FeedManager;
import org.sonews.feed.Subscription;
import org.sonews.storage.StorageBackendException;
import org.sonews.storage.StorageManager;
import org.sonews.storage.Group;
import org.sonews.util.Stats;
/**
* The XDAEMON command allows a client to get/set properties of the running
* server daemon. Only locally connected clients are allowed to use this
* command. The restriction to localhost connection can be suppressed by
* overriding the sonews.xdaemon.host bootstrap config property.
*
* @author Christian Lins
* @since sonews/0.5.0
*/
public class XDaemonCommand implements Command {
@Override
public String[] getSupportedCommandStrings() {
return new String[] { "XDAEMON" };
}
@Override
public boolean hasFinished() {
return true;
}
@Override
public String impliedCapability() {
return null;
}
@Override
public boolean isStateful() {
return false;
}
private void channelAdd(String[] commands, NNTPConnection conn)
throws IOException, StorageBackendException {
String groupName = commands[2];
if (StorageManager.current().isGroupExisting(groupName)) {
conn.println("400 group " + groupName + " already existing!");
} else {
StorageManager.current().addGroup(groupName,
Integer.parseInt(commands[3]));
conn.println("200 group " + groupName + " created");
}
}
// TODO: Refactor this method to reduce complexity!
@Override
public void processLine(NNTPConnection conn, String line, byte[] raw)
throws IOException, StorageBackendException {
InetSocketAddress addr = (InetSocketAddress) conn.getSocketChannel()
.socket().getRemoteSocketAddress();
if (addr.getHostName().equals(
Config.inst().get(Config.XDAEMON_HOST, "localhost"))) {
String[] commands = line.split(" ", 4);
if (commands.length == 3 && commands[1].equalsIgnoreCase("LIST")) {
if (commands[2].equalsIgnoreCase("CONFIGKEYS")) {
conn.println("100 list of available config keys follows");
for (String key : Config.AVAILABLE_KEYS) {
conn.println(key);
}
conn.println(".");
} else if (commands[2].equalsIgnoreCase("PEERINGRULES")) {
List<Subscription> pull = StorageManager.current()
.getSubscriptions(FeedManager.TYPE_PULL);
List<Subscription> push = StorageManager.current()
.getSubscriptions(FeedManager.TYPE_PUSH);
conn.println("100 list of peering rules follows");
for (Subscription sub : pull) {
conn.println("PULL " + sub.getHost() + ":"
+ sub.getPort() + " " + sub.getGroup());
}
for (Subscription sub : push) {
conn.println("PUSH " + sub.getHost() + ":"
+ sub.getPort() + " " + sub.getGroup());
}
conn.println(".");
} else {
conn.println("401 unknown sub command");
}
} else if (commands.length == 3
&& commands[1].equalsIgnoreCase("DELETE")) {
StorageManager.current().delete(commands[2]);
conn.println("200 article " + commands[2] + " deleted");
} else if (commands.length == 4
&& commands[1].equalsIgnoreCase("GROUPADD")) {
channelAdd(commands, conn);
} else if (commands.length == 3
&& commands[1].equalsIgnoreCase("GROUPDEL")) {
Group group = StorageManager.current().getGroup(commands[2]);
if (group == null) {
conn.println("400 group not found");
} else {
group.setFlag(Group.DELETED);
group.update();
conn.println("200 group " + commands[2]
+ " marked as deleted");
}
} else if (commands.length == 5
&& commands[1].equalsIgnoreCase("GROUPFLAG")) {
Group group = StorageManager.current().getGroup(commands[2]);
String flagName = commands[4];
if (commands[3].equalsIgnoreCase("SET")) {
if (flagName.equals("MAILINGLIST")) {
group.setFlag(Group.MAILINGLIST);
} else if (flagName.equals("DELETED")) {
group.setFlag(Group.DELETED);
} else if (flagName.equals("READONLY")) {
group.setFlag(Group.READONLY);
}
} else if (commands[3].equalsIgnoreCase("UNSET")) {
if (flagName.equals("MAILINGLIST")) {
group.unsetFlag(Group.MAILINGLIST);
} else if (flagName.equals("DELETED")) {
group.unsetFlag(Group.DELETED);
} else if (flagName.equals("READONLY")) {
group.unsetFlag(Group.READONLY);
}
} else {
conn.println("500 invalid command usage");
}
StorageManager.current().update(group);
} else if (commands.length == 4
&& commands[1].equalsIgnoreCase("SET")) {
String key = commands[2];
String val = commands[3];
Config.inst().set(key, val);
conn.println("200 new config value set");
} else if (commands.length == 3
&& commands[1].equalsIgnoreCase("GET")) {
String key = commands[2];
String val = Config.inst().get(key, null);
if (val != null) {
conn.println("100 config value for " + key + " follows");
conn.println(val);
conn.println(".");
} else {
conn.println("400 config value not set");
}
} else if (commands.length >= 3
&& commands[1].equalsIgnoreCase("LOG")) {
Group group = null;
if (commands.length > 3) {
group = StorageManager.current().getGroup(commands[3]);
}
if (commands[2].equalsIgnoreCase("CONNECTED_CLIENTS")) {
conn.println("100 number of connections follow");
conn.println(Integer.toString(Stats.getInstance()
.connectedClients()));
conn.println(".");
} else if (commands[2].equalsIgnoreCase("POSTED_NEWS")) {
conn.println("100 hourly numbers of posted news yesterday");
for (int n = 0; n < 24; n++) {
conn.println(n
+ " "
+ Stats.getInstance().getYesterdaysEvents(
Stats.POSTED_NEWS, n, group));
}
conn.println(".");
} else if (commands[2].equalsIgnoreCase("GATEWAYED_NEWS")) {
conn.println("100 hourly numbers of gatewayed news yesterday");
for (int n = 0; n < 24; n++) {
conn.println(n
+ " "
+ Stats.getInstance().getYesterdaysEvents(
Stats.GATEWAYED_NEWS, n, group));
}
conn.println(".");
} else if (commands[2].equalsIgnoreCase("TRANSMITTED_NEWS")) {
conn.println("100 hourly numbers of news transmitted to peers yesterday");
for (int n = 0; n < 24; n++) {
conn.println(n
+ " "
+ Stats.getInstance().getYesterdaysEvents(
Stats.FEEDED_NEWS, n, group));
}
conn.println(".");
} else if (commands[2].equalsIgnoreCase("HOSTED_NEWS")) {
conn.println("100 number of overall hosted news");
conn.println(Integer.toString(Stats.getInstance()
.getNumberOfNews()));
conn.println(".");
} else if (commands[2].equalsIgnoreCase("HOSTED_GROUPS")) {
conn.println("100 number of hosted groups");
conn.println(Integer.toString(Stats.getInstance()
.getNumberOfGroups()));
conn.println(".");
} else if (commands[2].equalsIgnoreCase("POSTED_NEWS_PER_HOUR")) {
conn.println("100 posted news per hour");
conn.println(Double.toString(Stats.getInstance()
.postedPerHour(-1)));
conn.println(".");
} else if (commands[2].equalsIgnoreCase("FEEDED_NEWS_PER_HOUR")) {
conn.println("100 feeded news per hour");
conn.println(Double.toString(Stats.getInstance()
.feededPerHour(-1)));
conn.println(".");
} else if (commands[2]
.equalsIgnoreCase("GATEWAYED_NEWS_PER_HOUR")) {
conn.println("100 gatewayed news per hour");
conn.println(Double.toString(Stats.getInstance()
.gatewayedPerHour(-1)));
conn.println(".");
} else {
conn.println("401 unknown sub command");
}
} else if (commands.length >= 3
&& commands[1].equalsIgnoreCase("PLUGIN")) {
conn.println("400 invalid command usage");
} else {
conn.println("400 invalid command usage");
}
} else {
conn.println("501 not allowed");
}
}
}