Mercurialsonews-drupal / file revision
summary | shortlog | changelog | graph | tags | bookmarks | branches | files | changeset | file | latest | revisions | annotate | diff | raw | help
src/org/sonews/daemon/command/XDaemonCommand.java
author cli
Sat, 10 Sep 2011 18:18:05 +0200
changeset 45 7e24949b87b0
parent 39 73b21e9f3958
child 46 28870db3b9fd
permissions -rwxr-xr-x
HSQLDB backend support completed, but untested. 
     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 == 5 && commands[1].equalsIgnoreCase("GROUPFLAG")) {

   131 				Group group = StorageManager.current().getGroup(commands[2]);

   132 				String flagName = commands[4];

   133 				if(commands[3].equalsIgnoreCase("SET")) {

   134 					

   135 				} else if(commands[3].equalsIgnoreCase("UNSET")) {

   136 					

   137 				} else {

   138 					conn.println("500 invalid command usage");

   139 				}

   140 			} else if (commands.length == 4 && commands[1].equalsIgnoreCase("SET")) {

   141 				String key = commands[2];

   142 				String val = commands[3];

   143 				Config.inst().set(key, val);

   144 				conn.println("200 new config value set");

   145 			} else if (commands.length == 3 && commands[1].equalsIgnoreCase("GET")) {

   146 				String key = commands[2];

   147 				String val = Config.inst().get(key, null);

   148 				if (val != null) {

   149 					conn.println("100 config value for " + key + " follows");

   150 					conn.println(val);

   151 					conn.println(".");

   152 				} else {

   153 					conn.println("400 config value not set");

   154 				}

   155 			} else if (commands.length >= 3 && commands[1].equalsIgnoreCase("LOG")) {

   156 				Group group = null;

   157 				if (commands.length > 3) {

   158 					group = (Group) Channel.getByName(commands[3]);

   159 				}

   160 

   161 				if (commands[2].equalsIgnoreCase("CONNECTED_CLIENTS")) {

   162 					conn.println("100 number of connections follow");

   163 					conn.println(Integer.toString(Stats.getInstance().connectedClients()));

   164 					conn.println(".");

   165 				} else if (commands[2].equalsIgnoreCase("POSTED_NEWS")) {

   166 					conn.println("100 hourly numbers of posted news yesterday");

   167 					for (int n = 0; n < 24; n++) {

   168 						conn.println(n + " " + Stats.getInstance().getYesterdaysEvents(Stats.POSTED_NEWS, n, group));

   169 					}

   170 					conn.println(".");

   171 				} else if (commands[2].equalsIgnoreCase("GATEWAYED_NEWS")) {

   172 					conn.println("100 hourly numbers of gatewayed news yesterday");

   173 					for (int n = 0; n < 24; n++) {

   174 						conn.println(n + " " + Stats.getInstance().getYesterdaysEvents(Stats.GATEWAYED_NEWS, n, group));

   175 					}

   176 					conn.println(".");

   177 				} else if (commands[2].equalsIgnoreCase("TRANSMITTED_NEWS")) {

   178 					conn.println("100 hourly numbers of news transmitted to peers yesterday");

   179 					for (int n = 0; n < 24; n++) {

   180 						conn.println(n + " " + Stats.getInstance().getYesterdaysEvents(Stats.FEEDED_NEWS, n, group));

   181 					}

   182 					conn.println(".");

   183 				} else if (commands[2].equalsIgnoreCase("HOSTED_NEWS")) {

   184 					conn.println("100 number of overall hosted news");

   185 					conn.println(Integer.toString(Stats.getInstance().getNumberOfNews()));

   186 					conn.println(".");

   187 				} else if (commands[2].equalsIgnoreCase("HOSTED_GROUPS")) {

   188 					conn.println("100 number of hosted groups");

   189 					conn.println(Integer.toString(Stats.getInstance().getNumberOfGroups()));

   190 					conn.println(".");

   191 				} else if (commands[2].equalsIgnoreCase("POSTED_NEWS_PER_HOUR")) {

   192 					conn.println("100 posted news per hour");

   193 					conn.println(Double.toString(Stats.getInstance().postedPerHour(-1)));

   194 					conn.println(".");

   195 				} else if (commands[2].equalsIgnoreCase("FEEDED_NEWS_PER_HOUR")) {

   196 					conn.println("100 feeded news per hour");

   197 					conn.println(Double.toString(Stats.getInstance().feededPerHour(-1)));

   198 					conn.println(".");

   199 				} else if (commands[2].equalsIgnoreCase("GATEWAYED_NEWS_PER_HOUR")) {

   200 					conn.println("100 gatewayed news per hour");

   201 					conn.println(Double.toString(Stats.getInstance().gatewayedPerHour(-1)));

   202 					conn.println(".");

   203 				} else {

   204 					conn.println("401 unknown sub command");

   205 				}

   206 			} else if (commands.length >= 3 && commands[1].equalsIgnoreCase("PLUGIN")) {

   207 				conn.println("400 invalid command usage");

   208 			} else {

   209 				conn.println("400 invalid command usage");

   210 			}

   211 		} else {

   212 			conn.println("501 not allowed");

   213 		}

   214 	}

   215 }
sonews-drupal