489a742dfdaf54e67a3ab2bf3783d41ed681c44b
[reactos.git] / irc / TechBot / TechBot.Library / TechBotIrcService.cs
1 using System;
2 using System.Text;
3 using System.Collections;
4 using System.Threading;
5
6 using TechBot.IRCLibrary;
7
8 namespace TechBot.Library
9 {
10 public class IrcServiceOutput : IServiceOutput
11 {
12 public void WriteLine(MessageContext context,
13 string message)
14 {
15 if (context is ChannelMessageContext)
16 {
17 ChannelMessageContext channelContext = context as ChannelMessageContext;
18 channelContext.Channel.Talk(message);
19 }
20 else if (context is UserMessageContext)
21 {
22 UserMessageContext userContext = context as UserMessageContext;
23 userContext.User.Talk(message);
24 }
25 else
26 {
27 throw new InvalidOperationException(String.Format("Unhandled message context '{0}'",
28 context.GetType()));
29 }
30 }
31 }
32
33 public class IrcTechBotService : TechBotService
34 {
35 private int port;
36 private string hostname;
37 private string channelnames;
38 private string botname;
39 private string password;
40 private string chmPath;
41 private string mainChm;
42 private IrcClient m_IrcClient;
43 private ArrayList channels = new ArrayList();
44 private bool isStopped = false;
45
46 public IrcTechBotService(string hostname,
47 int port,
48 string channelnames,
49 string botname,
50 string password,
51 string chmPath,
52 string mainChm)
53 : base (new IrcServiceOutput() , chmPath , mainChm)
54 {
55 this.hostname = hostname;
56 this.port = port;
57 this.channelnames = channelnames;
58 this.botname = botname;
59 if (password == null || password.Trim() == "")
60 this.password = null;
61 else
62 this.password = password;
63 this.chmPath = chmPath;
64 this.mainChm = mainChm;
65 }
66
67 public override void Run()
68 {
69 //Call the base class
70 base.Run();
71
72 m_IrcClient = new IrcClient();
73 m_IrcClient.Encoding = Encoding.GetEncoding("iso-8859-1");
74 m_IrcClient.OnConnect += new OnConnectHandler(m_IrcClient_OnConnect);
75 m_IrcClient.OnConnectionLost += new OnConnectionLostHandler(m_IrcClient_OnConnectionLost);
76 m_IrcClient.OnDisconnect += new OnDisconnectHandler(m_IrcClient_OnDisconnect);
77 m_IrcClient.MessageReceived += new MessageReceivedHandler(client_MessageReceived);
78 m_IrcClient.ChannelUserDatabaseChanged += new ChannelUserDatabaseChangedHandler(client_ChannelUserDatabaseChanged);
79
80 Connect();
81 }
82
83 void m_IrcClient_OnConnect()
84 {
85 Console.WriteLine("Connected...");
86 }
87
88 private void Connect()
89 {
90 Console.WriteLine("Connecting to {0} port {1}",
91 hostname,
92 port);
93 m_IrcClient.Connect(hostname, port);
94
95 m_IrcClient.Register(botname, password, null);
96 Console.WriteLine("Registered as {0}...", botname);
97 JoinChannels();
98
99 while (!isStopped)
100 {
101 Thread.Sleep(1000);
102 }
103
104 PartChannels();
105 m_IrcClient.Diconnect();
106 }
107
108 void m_IrcClient_OnDisconnect()
109 {
110 Console.WriteLine("Disconnected...");
111 }
112
113 void m_IrcClient_OnConnectionLost()
114 {
115 //Dispose old connection
116 Disconnect();
117
118 //Sleep for 1 minute
119 Thread.Sleep(1000 * 60);
120
121 //Try to reconnect
122 Connect();
123 }
124
125 private void Disconnect()
126 {
127 try
128 {
129 m_IrcClient.Diconnect();
130 }
131 catch (Exception)
132 {
133 //
134 }
135 }
136
137 public void Stop()
138 {
139 isStopped = true;
140 }
141
142 private void JoinChannels()
143 {
144 foreach (string channelname in channelnames.Split(new char[] { ';' }))
145 {
146 IrcChannel channel = m_IrcClient.JoinChannel(channelname);
147 channels.Add(channel);
148 System.Console.WriteLine(String.Format("Joined channel #{0}...",
149 channel.Name));
150 }
151 }
152
153 private void PartChannels()
154 {
155 foreach (IrcChannel channel in channels)
156 {
157 m_IrcClient.PartChannel(channel, "Caught in the bitstream...");
158 System.Console.WriteLine(String.Format("Parted channel #{0}...",
159 channel.Name));
160 }
161 }
162
163 private string GetMessageSource(MessageContext context)
164 {
165 if (context is ChannelMessageContext)
166 {
167 ChannelMessageContext channelContext = context as ChannelMessageContext;
168 return String.Format("#{0}",
169 channelContext.Channel.Name);
170 }
171 else if (context is UserMessageContext)
172 {
173 UserMessageContext userContext = context as UserMessageContext;
174 return userContext.User.Nickname;
175 }
176 else
177 {
178 throw new InvalidOperationException(String.Format("Unhandled message context '{0}'",
179 context.GetType()));
180 }
181 }
182
183 private void ExtractMessage(string parameters,
184 out string message)
185 {
186 int startIndex = parameters.IndexOf(':');
187 if (startIndex != -1)
188 {
189 message = parameters.Substring(startIndex + 1);
190 }
191 else
192 {
193 message = parameters;
194 }
195 }
196
197 private bool GetChannelName(IrcMessage message,
198 out string channelName)
199 {
200 if (message.Parameters == null || !message.Parameters.StartsWith("#"))
201 {
202 channelName = null;
203 return false;
204 }
205
206 int index = message.Parameters.IndexOf(' ');
207 if (index == -1)
208 index = message.Parameters.Length;
209 else
210 index = index - 1;
211 channelName = message.Parameters.Substring(1, index);
212 return true;
213 }
214
215 private bool GetTargetNickname(IrcMessage message,
216 out string nickname)
217 {
218 if (message.Parameters == null)
219 {
220 nickname = null;
221 return false;
222 }
223
224 int index = message.Parameters.IndexOf(' ');
225 if (index == -1)
226 index = message.Parameters.Length;
227 nickname = message.Parameters.Substring(0, index);
228 Console.WriteLine("nickname: " + nickname);
229 return true;
230 }
231
232 private bool ShouldAcceptMessage(IrcMessage message,
233 out MessageContext context)
234 {
235 if (message.Command.ToUpper().Equals("PRIVMSG"))
236 {
237 string channelName;
238 string nickname;
239 if (GetChannelName(message,
240 out channelName))
241 {
242 foreach (IrcChannel channel in channels)
243 {
244 if (String.Compare(channel.Name, channelName, true) == 0)
245 {
246 context = new ChannelMessageContext(channel);
247 return true;
248 }
249 }
250 }
251 else if (GetTargetNickname(message,
252 out nickname))
253 {
254 IrcUser targetUser = new IrcUser(m_IrcClient,
255 nickname);
256 if (String.Compare(targetUser.Nickname, botname, true) == 0)
257 {
258 IrcUser sourceUser = new IrcUser(m_IrcClient,
259 message.PrefixNickname);
260 context = new UserMessageContext(sourceUser);
261 return true;
262 }
263 }
264 }
265 context = null;
266 return false;
267 }
268
269 private void client_MessageReceived(IrcMessage message)
270 {
271 try
272 {
273 if (message.Command != null &&
274 message.Parameters != null)
275 {
276 string injectMessage;
277 ExtractMessage(message.Parameters,
278 out injectMessage);
279 MessageContext context;
280 if (ShouldAcceptMessage(message,
281 out context))
282 {
283 Console.WriteLine(String.Format("Injecting: {0} from {1}",
284 injectMessage,
285 GetMessageSource(context)));
286 InjectMessage(context,
287 injectMessage);
288 }
289 else
290 {
291 Console.WriteLine("Received: " + message.Line);
292 }
293 }
294 else
295 {
296 Console.WriteLine("Received: " + message.Line);
297 }
298 }
299 catch (Exception ex)
300 {
301 Console.WriteLine(String.Format("Exception: {0}", ex));
302 }
303 }
304
305 private void client_ChannelUserDatabaseChanged(IrcChannel channel)
306 {
307 }
308 }
309 }