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.Exchange; 021 import org.apache.camel.RuntimeCamelException; 022 import org.apache.camel.impl.DefaultProducer; 023 import org.apache.commons.logging.Log; 024 import org.apache.commons.logging.LogFactory; 025 import org.schwering.irc.lib.IRCConnection; 026 import org.schwering.irc.lib.IRCEventListener; 027 028 public class IrcProducer extends DefaultProducer<IrcExchange> { 029 private static final transient Log log = LogFactory.getLog(IrcProducer.class); 030 private IRCConnection connection; 031 private IrcEndpoint endpoint; 032 private IRCEventListener ircErrorLogger; 033 034 public IrcProducer(IrcEndpoint endpoint, IRCConnection connection) { 035 super(endpoint); 036 this.endpoint = endpoint; 037 this.connection = connection; 038 } 039 040 public void process(Exchange exchange) throws Exception { 041 try { 042 final String msg = exchange.getIn().getBody(String.class); 043 if (isMessageACommand(msg)) { 044 connection.send(msg); 045 } 046 else { 047 final String target = endpoint.getConfiguration().getTarget(); 048 049 if (log.isDebugEnabled()) { 050 log.debug("sending to: " + target + " message: " + msg); 051 } 052 053 connection.doPrivmsg(target, msg); 054 } 055 } 056 catch (Exception e) { 057 throw new RuntimeCamelException(e); 058 } 059 } 060 061 @Override 062 protected void doStart() throws Exception { 063 super.doStart(); 064 065 ircErrorLogger = createIrcErrorLogger(); 066 connection.addIRCEventListener(ircErrorLogger); 067 068 final String target = endpoint.getConfiguration().getTarget(); 069 070 log.debug("joining: " + target); 071 connection.doJoin(target); 072 } 073 074 @Override 075 protected void doStop() throws Exception { 076 super.doStop(); 077 if (connection != null) { 078 connection.removeIRCEventListener(ircErrorLogger); 079 } 080 } 081 082 protected boolean isMessageACommand(String msg) { 083 for (String command : commands) { 084 if (msg.startsWith(command)) { 085 return true; 086 } 087 } 088 return false; 089 } 090 091 protected IRCEventListener createIrcErrorLogger() { 092 return new IrcErrorLogger(log); 093 } 094 095 public final String[] commands = new String[]{ 096 "AWAY", "INVITE", "ISON", "JOIN", 097 "KICK", "LIST", "NAMES", "PRIVMSG", 098 "MODE", "NICK", "NOTICE", "PART", 099 "PONG", "QUIT", "TOPIC", "WHO", 100 "WHOIS", "WHOWAS", "USERHOST" 101 }; 102 }