src/org/sonews/daemon/command/XDaemonCommand.java
author cli
Sun Aug 29 18:17:37 2010 +0200 (2010-08-29)
changeset 37 74139325d305
parent 35 ed84c8bdd87b
child 39 73b21e9f3958
permissions -rw-r--r--
Switch intent style to Original K&R / Linux / Kernel.
     1 /*
     2  *   SONEWS News Server
     3  *   see AUTHORS for the list of contributors
     4  *
     5  *   This program is free software: you can redistribute it and/or modify
     6  *   it under the terms of the GNU General Public License as published by
     7  *   the Free Software Foundation, either version 3 of the License, or
     8  *   (at your option) any later version.
     9  *
    10  *   This program is distributed in the hope that it will be useful,
    11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    13  *   GNU General Public License for more details.
    14  *
    15  *   You should have received a copy of the GNU General Public License
    16  *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
    17  */
    18 
    19 package org.sonews.daemon.command;
    20 
    21 import java.io.IOException;
    22 import java.net.InetSocketAddress;
    23 import java.util.List;
    24 import org.sonews.config.Config;
    25 import org.sonews.daemon.NNTPConnection;
    26 import org.sonews.storage.StorageBackendException;
    27 import org.sonews.storage.StorageManager;
    28 import org.sonews.feed.FeedManager;
    29 import org.sonews.feed.Subscription;
    30 import org.sonews.storage.Channel;
    31 import org.sonews.storage.Group;
    32 import org.sonews.util.Stats;
    33 
    34 /**
    35  * The XDAEMON command allows a client to get/set properties of the
    36  * running server daemon. Only locally connected clients are allowed to
    37  * use this command.
    38  * The restriction to localhost connection can be suppressed by overriding
    39  * the sonews.xdaemon.host bootstrap config property.
    40  * @author Christian Lins
    41  * @since sonews/0.5.0
    42  */
    43 public class XDaemonCommand implements Command
    44 {
    45 
    46 	@Override
    47 	public String[] getSupportedCommandStrings()
    48 	{
    49 		return new String[] {"XDAEMON"};
    50 	}
    51 
    52 	@Override
    53 	public boolean hasFinished()
    54 	{
    55 		return true;
    56 	}
    57 
    58 	@Override
    59 	public String impliedCapability()
    60 	{
    61 		return null;
    62 	}
    63 
    64 	@Override
    65 	public boolean isStateful()
    66 	{
    67 		return false;
    68 	}
    69 
    70 	private void channelAdd(String[] commands, NNTPConnection conn)
    71 		throws IOException, StorageBackendException
    72 	{
    73 		String groupName = commands[2];
    74 		if (StorageManager.current().isGroupExisting(groupName)) {
    75 			conn.println("400 group " + groupName + " already existing!");
    76 		} else {
    77 			StorageManager.current().addGroup(groupName, Integer.parseInt(commands[3]));
    78 			conn.println("200 group " + groupName + " created");
    79 		}
    80 	}
    81 
    82 	// TODO: Refactor this method to reduce complexity!
    83 	@Override
    84 	public void processLine(NNTPConnection conn, String line, byte[] raw)
    85 		throws IOException, StorageBackendException
    86 	{
    87 		InetSocketAddress addr = (InetSocketAddress) conn.getSocketChannel().socket().getRemoteSocketAddress();
    88 		if (addr.getHostName().equals(
    89 			Config.inst().get(Config.XDAEMON_HOST, "localhost"))) {
    90 			String[] commands = line.split(" ", 4);
    91 			if (commands.length == 3 && commands[1].equalsIgnoreCase("LIST")) {
    92 				if (commands[2].equalsIgnoreCase("CONFIGKEYS")) {
    93 					conn.println("100 list of available config keys follows");
    94 					for (String key : Config.AVAILABLE_KEYS) {
    95 						conn.println(key);
    96 					}
    97 					conn.println(".");
    98 				} else if (commands[2].equalsIgnoreCase("PEERINGRULES")) {
    99 					List<Subscription> pull =
   100 						StorageManager.current().getSubscriptions(FeedManager.TYPE_PULL);
   101 					List<Subscription> push =
   102 						StorageManager.current().getSubscriptions(FeedManager.TYPE_PUSH);
   103 					conn.println("100 list of peering rules follows");
   104 					for (Subscription sub : pull) {
   105 						conn.println("PULL " + sub.getHost() + ":" + sub.getPort()
   106 							+ " " + sub.getGroup());
   107 					}
   108 					for (Subscription sub : push) {
   109 						conn.println("PUSH " + sub.getHost() + ":" + sub.getPort()
   110 							+ " " + sub.getGroup());
   111 					}
   112 					conn.println(".");
   113 				} else {
   114 					conn.println("401 unknown sub command");
   115 				}
   116 			} else if (commands.length == 3 && commands[1].equalsIgnoreCase("DELETE")) {
   117 				StorageManager.current().delete(commands[2]);
   118 				conn.println("200 article " + commands[2] + " deleted");
   119 			} else if (commands.length == 4 && commands[1].equalsIgnoreCase("GROUPADD")) {
   120 				channelAdd(commands, conn);
   121 			} else if (commands.length == 3 && commands[1].equalsIgnoreCase("GROUPDEL")) {
   122 				Group group = StorageManager.current().getGroup(commands[2]);
   123 				if (group == null) {
   124 					conn.println("400 group not found");
   125 				} else {
   126 					group.setFlag(Group.DELETED);
   127 					group.update();
   128 					conn.println("200 group " + commands[2] + " marked as deleted");
   129 				}
   130 			} else if (commands.length == 4 && commands[1].equalsIgnoreCase("SET")) {
   131 				String key = commands[2];
   132 				String val = commands[3];
   133 				Config.inst().set(key, val);
   134 				conn.println("200 new config value set");
   135 			} else if (commands.length == 3 && commands[1].equalsIgnoreCase("GET")) {
   136 				String key = commands[2];
   137 				String val = Config.inst().get(key, null);
   138 				if (val != null) {
   139 					conn.println("100 config value for " + key + " follows");
   140 					conn.println(val);
   141 					conn.println(".");
   142 				} else {
   143 					conn.println("400 config value not set");
   144 				}
   145 			} else if (commands.length >= 3 && commands[1].equalsIgnoreCase("LOG")) {
   146 				Group group = null;
   147 				if (commands.length > 3) {
   148 					group = (Group) Channel.getByName(commands[3]);
   149 				}
   150 
   151 				if (commands[2].equalsIgnoreCase("CONNECTED_CLIENTS")) {
   152 					conn.println("100 number of connections follow");
   153 					conn.println(Integer.toString(Stats.getInstance().connectedClients()));
   154 					conn.println(".");
   155 				} else if (commands[2].equalsIgnoreCase("POSTED_NEWS")) {
   156 					conn.println("100 hourly numbers of posted news yesterday");
   157 					for (int n = 0; n < 24; n++) {
   158 						conn.println(n + " " + Stats.getInstance().getYesterdaysEvents(Stats.POSTED_NEWS, n, group));
   159 					}
   160 					conn.println(".");
   161 				} else if (commands[2].equalsIgnoreCase("GATEWAYED_NEWS")) {
   162 					conn.println("100 hourly numbers of gatewayed news yesterday");
   163 					for (int n = 0; n < 24; n++) {
   164 						conn.println(n + " " + Stats.getInstance().getYesterdaysEvents(Stats.GATEWAYED_NEWS, n, group));
   165 					}
   166 					conn.println(".");
   167 				} else if (commands[2].equalsIgnoreCase("TRANSMITTED_NEWS")) {
   168 					conn.println("100 hourly numbers of news transmitted to peers yesterday");
   169 					for (int n = 0; n < 24; n++) {
   170 						conn.println(n + " " + Stats.getInstance().getYesterdaysEvents(Stats.FEEDED_NEWS, n, group));
   171 					}
   172 					conn.println(".");
   173 				} else if (commands[2].equalsIgnoreCase("HOSTED_NEWS")) {
   174 					conn.println("100 number of overall hosted news");
   175 					conn.println(Integer.toString(Stats.getInstance().getNumberOfNews()));
   176 					conn.println(".");
   177 				} else if (commands[2].equalsIgnoreCase("HOSTED_GROUPS")) {
   178 					conn.println("100 number of hosted groups");
   179 					conn.println(Integer.toString(Stats.getInstance().getNumberOfGroups()));
   180 					conn.println(".");
   181 				} else if (commands[2].equalsIgnoreCase("POSTED_NEWS_PER_HOUR")) {
   182 					conn.println("100 posted news per hour");
   183 					conn.println(Double.toString(Stats.getInstance().postedPerHour(-1)));
   184 					conn.println(".");
   185 				} else if (commands[2].equalsIgnoreCase("FEEDED_NEWS_PER_HOUR")) {
   186 					conn.println("100 feeded news per hour");
   187 					conn.println(Double.toString(Stats.getInstance().feededPerHour(-1)));
   188 					conn.println(".");
   189 				} else if (commands[2].equalsIgnoreCase("GATEWAYED_NEWS_PER_HOUR")) {
   190 					conn.println("100 gatewayed news per hour");
   191 					conn.println(Double.toString(Stats.getInstance().gatewayedPerHour(-1)));
   192 					conn.println(".");
   193 				} else {
   194 					conn.println("401 unknown sub command");
   195 				}
   196 			} else if (commands.length >= 3 && commands[1].equalsIgnoreCase("PLUGIN")) {
   197 			} else {
   198 				conn.println("400 invalid command usage");
   199 			}
   200 		} else {
   201 			conn.println("501 not allowed");
   202 		}
   203 	}
   204 }