-
Notifications
You must be signed in to change notification settings - Fork 379
/
keepnick.cpp
199 lines (157 loc) · 4.74 KB
/
keepnick.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
/*
* Copyright (C) 2004-2011 See the AUTHORS file for details.
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 as published
* by the Free Software Foundation.
*/
#include "Modules.h"
#include "User.h"
#include "IRCSock.h"
class CKeepNickMod;
class CKeepNickTimer : public CTimer {
public:
CKeepNickTimer(CKeepNickMod *pMod);
~CKeepNickTimer() {}
void RunJob();
private:
CKeepNickMod* m_pMod;
};
class CKeepNickMod : public CModule {
public:
MODCONSTRUCTOR(CKeepNickMod) {}
~CKeepNickMod() {}
bool OnLoad(const CString& sArgs, CString& sMessage) {
m_pTimer = NULL;
// Check if we need to start the timer
if (m_pUser->IsIRCConnected())
OnIRCConnected();
return true;
}
void KeepNick() {
if (!m_pTimer)
// No timer means we are turned off
return;
CIRCSock* pIRCSock = GetUser()->GetIRCSock();
if (!pIRCSock)
return;
// Do we already have the nick we want?
if (pIRCSock->GetNick().Equals(GetNick()))
return;
PutIRC("NICK " + GetNick());
}
CString GetNick() {
CString sConfNick = m_pUser->GetNick();
CIRCSock* pIRCSock = GetUser()->GetIRCSock();
if (pIRCSock)
sConfNick = sConfNick.Left(pIRCSock->GetMaxNickLen());
return sConfNick;
}
void OnNick(const CNick& Nick, const CString& sNewNick, const vector<CChan*>& vChans) {
if (sNewNick == GetUser()->GetIRCSock()->GetNick()) {
// We are changing our own nick
if (Nick.GetNick().Equals(GetNick())) {
// We are changing our nick away from the conf setting.
// Let's assume the user wants this and disable
// this module (to avoid fighting nickserv).
Disable();
} else if (sNewNick.Equals(GetNick())) {
// We are changing our nick to the conf setting,
// so we don't need that timer anymore.
Disable();
}
return;
}
// If the nick we want is free now, be fast and get the nick
if (Nick.GetNick().Equals(GetNick())) {
KeepNick();
}
}
void OnQuit(const CNick& Nick, const CString& sMessage, const vector<CChan*>& vChans) {
// If someone with the nick we want quits, be fast and get the nick
if (Nick.GetNick().Equals(GetNick())) {
KeepNick();
}
}
void OnIRCDisconnected() {
// No way we can do something if we aren't connected to IRC.
Disable();
}
void OnIRCConnected() {
if (!GetUser()->GetIRCSock()->GetNick().Equals(GetNick())) {
// We don't have the nick we want, try to get it
Enable();
}
}
void Enable() {
if (m_pTimer)
return;
m_pTimer = new CKeepNickTimer(this);
AddTimer(m_pTimer);
}
void Disable() {
if (!m_pTimer)
return;
m_pTimer->Stop();
RemTimer(m_pTimer->GetName());
m_pTimer = NULL;
}
virtual EModRet OnUserRaw(CString& sLine) {
// We dont care if we are not connected to IRC
if (!m_pUser->IsIRCConnected())
return CONTINUE;
// We are trying to get the config nick and this is a /nick?
if (!m_pTimer || !sLine.Token(0).Equals("NICK"))
return CONTINUE;
// Is the nick change for the nick we are trying to get?
CString sNick = sLine.Token(1);
// Don't even think of using spaces in your nick!
if (sNick.Left(1) == ":")
sNick.LeftChomp();
if (!sNick.Equals(GetNick()))
return CONTINUE;
// Indeed trying to change to this nick, generate a 433 for it.
// This way we can *always* block incoming 433s from the server.
PutUser(":" + m_pUser->GetIRCServer() + " 433 " + m_pUser->GetIRCNick().GetNick()
+ " " + sNick + " :ZNC is already trying to get this nickname");
return CONTINUE;
}
virtual EModRet OnRaw(CString& sLine) {
// Are we trying to get our primary nick and we caused this error?
// :irc.server.net 433 mynick badnick :Nickname is already in use.
if (m_pTimer && sLine.Token(1) == "433" && sLine.Token(3).Equals(GetNick()))
return HALT;
return CONTINUE;
}
void OnModCommand(const CString& sCommand) {
CString sCmd = sCommand.AsUpper();
if (sCmd == "ENABLE") {
Enable();
PutModule("Trying to get your primary nick");
} else if (sCmd == "DISABLE") {
Disable();
PutModule("No longer trying to get your primary nick");
} else if (sCmd == "STATE") {
if (m_pTimer)
PutModule("Currently trying to get your primary nick");
else
PutModule("Currently disabled, try 'enable'");
} else {
PutModule("Commands: Enable, Disable, State");
}
}
private:
// If this is NULL, we are turned off for some reason
CKeepNickTimer* m_pTimer;
};
CKeepNickTimer::CKeepNickTimer(CKeepNickMod *pMod) : CTimer(pMod, 30, 0,
"KeepNickTimer", "Tries to acquire this user's primary nick") {
m_pMod = pMod;
}
void CKeepNickTimer::RunJob() {
m_pMod->KeepNick();
}
template<> void TModInfo<CKeepNickMod>(CModInfo& Info) {
Info.SetWikiPage("keepnick");
}
MODULEDEFS(CKeepNickMod, "Keep trying for your primary nick")