001 /**
002 *
003 * Licensed to the Apache Software Foundation (ASF) under one or more
004 * contributor license agreements. See the NOTICE file distributed with
005 * this work for additional information regarding copyright ownership.
006 * The ASF licenses this file to You under the Apache License, Version 2.0
007 * (the "License"); you may not use this file except in compliance with
008 * the License. You may obtain a copy of the License at
009 *
010 * http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018 package org.apache.camel.component.irc;
019
020 import org.apache.camel.CamelContext;
021 import org.apache.camel.RuntimeCamelException;
022 import org.apache.camel.impl.DefaultComponent;
023 import org.apache.camel.util.IntrospectionSupport;
024 import org.apache.commons.logging.Log;
025 import org.apache.commons.logging.LogFactory;
026 import org.schwering.irc.lib.IRCConnection;
027
028 import java.net.URI;
029 import java.util.HashMap;
030 import java.util.Map;
031
032 /**
033 * Defines the <a href="http://activemq.apache.org/camel/irc.html">IRC Component</a>
034 *
035 * @version $Revision:$
036 */
037 public class IrcComponent extends DefaultComponent<IrcExchange> {
038 private static final transient Log log = LogFactory.getLog(IrcComponent.class);
039 private IrcConfiguration configuration;
040 private final Map<String, IRCConnection> connectionCache = new HashMap<String, IRCConnection>();
041
042 public static IrcComponent ircComponent() {
043 return new IrcComponent();
044 }
045
046 public IrcComponent() {
047 configuration = new IrcConfiguration();
048 }
049
050 public IrcComponent(IrcConfiguration configuration) {
051 this.configuration = configuration;
052 }
053
054 public IrcComponent(CamelContext context) {
055 super(context);
056 configuration = new IrcConfiguration();
057 }
058
059 protected IrcEndpoint createEndpoint(String uri, String remaining, Map parameters) throws Exception {
060 IrcConfiguration config = getConfiguration().copy();
061 config.configure(new URI(uri));
062
063 // lets make sure we copy the configuration as each endpoint can customize its own version
064 final IrcEndpoint endpoint = new IrcEndpoint(uri, this, config);
065
066 IntrospectionSupport.setProperties(endpoint.getConfiguration(), parameters);
067 return endpoint;
068 }
069
070 public IrcConfiguration getConfiguration() {
071 return configuration;
072 }
073
074 public void setConfiguration(IrcConfiguration configuration) {
075 this.configuration = configuration;
076 }
077
078 public synchronized IRCConnection getIRCConnection(IrcConfiguration configuration) {
079 final IRCConnection connection;
080 if (connectionCache.containsKey(configuration.getCacheKey())) {
081 if (log.isDebugEnabled()) {
082 log.debug("Returning Cached Connection to " + configuration.getHostname() + " " + configuration.getTarget());
083 }
084 connection = connectionCache.get(configuration.getCacheKey());
085 }
086 else {
087 connection = createConnection(configuration);
088 connectionCache.put(configuration.getCacheKey(), connection);
089 }
090 return connection;
091 }
092
093 protected IRCConnection createConnection(IrcConfiguration configuration) {
094 log.debug("Creating Connection to " + configuration.getHostname() + " destination: " + configuration.getTarget()
095 + " nick: " + configuration.getNickname() + " user: " + configuration.getUsername());
096
097 final IRCConnection conn = new IRCConnection(configuration.getHostname(), configuration.getPorts(), configuration.getPassword(), configuration.getNickname(), configuration.getUsername(), configuration.getRealname());
098 conn.setEncoding("UTF-8");
099 // conn.setDaemon(true);
100 conn.setColors(configuration.isColors());
101 conn.setPong(true);
102
103 try {
104 conn.connect();
105 }
106 catch (Exception e) {
107 log.error("Failed to connect: " + e, e);
108
109 // TODO use checked exceptions?
110 throw new RuntimeCamelException(e);
111 }
112 return conn;
113 }
114
115 public void closeConnection(String key, IRCConnection connection) {
116 try {
117 connection.doQuit();
118 connection.close();
119 }
120 catch (Exception e) {
121 e.printStackTrace();
122 }
123 }
124
125 @Override
126 protected synchronized void doStop() throws Exception {
127 // lets use a copy so we can clear the connections eagerly in case of exceptions
128 Map<String, IRCConnection> map = new HashMap<String, IRCConnection>(connectionCache);
129 connectionCache.clear();
130 for (Map.Entry<String, IRCConnection> entry : map.entrySet()) {
131 closeConnection(entry.getKey(), entry.getValue());
132 }
133 super.doStop();
134 }
135 }