]>
Commit | Line | Data |
---|---|---|
1 | #include <string.h> | |
2 | #include <assert.h> | |
3 | #include "../lib/version.h" | |
4 | #include "../nick/nick.h" | |
5 | #include "../chanindex/chanindex.h" | |
6 | #include "../channel/channel.h" | |
7 | #include "../core/hooks.h" | |
8 | #include "whowas.h" | |
9 | ||
10 | MODULE_VERSION(""); | |
11 | ||
12 | static int wwcnext, wwccext; | |
13 | ||
14 | static void wwc_refchannel(chanindex *cip) { | |
15 | uintptr_t *refcount = (uintptr_t *)&cip->exts[wwccext]; | |
16 | (*refcount)++; | |
17 | } | |
18 | ||
19 | static void wwc_derefchannel(chanindex *cip) { | |
20 | uintptr_t *refcount = (uintptr_t *)&cip->exts[wwccext]; | |
21 | (*refcount)--; | |
22 | ||
23 | if (*refcount == 0) | |
24 | releasechanindex(cip); | |
25 | } | |
26 | ||
27 | static void wwc_hook_joincreate(int hooknum, void *arg) { | |
28 | void **args = arg; | |
29 | channel *cp = args[0]; | |
30 | nick *np = args[1]; | |
31 | chanindex **wchans = np->exts[wwcnext]; | |
32 | ||
33 | if (!wchans) { | |
34 | wchans = calloc(sizeof(chanindex *), WW_MAXCHANNELS); | |
35 | np->exts[wwcnext] = wchans; | |
36 | } | |
37 | ||
38 | wwc_refchannel(cp->index); | |
39 | ||
40 | memmove(&wchans[1], &wchans[0], sizeof(chanindex *) * (WW_MAXCHANNELS - 1)); | |
41 | wchans[0] = cp->index; | |
42 | } | |
43 | ||
44 | static void wwc_hook_lostnick(int hooknum, void *arg) { | |
45 | nick *np = arg; | |
46 | chanindex **wchans = np->exts[wwcnext]; | |
47 | int i; | |
48 | ||
49 | if (!wchans) | |
50 | return; | |
51 | ||
52 | for (i = 0; i < WW_MAXCHANNELS; i++) { | |
53 | if (!wchans[i]) | |
54 | break; | |
55 | ||
56 | wwc_derefchannel(wchans[i]); | |
57 | } | |
58 | ||
59 | free(wchans); | |
60 | np->exts[wwcnext] = NULL; | |
61 | } | |
62 | ||
63 | static void wwc_hook_newrecord(int hooknum, void *arg) { | |
64 | void **args = arg; | |
65 | whowas *ww = args[0]; | |
66 | nick *np = args[1]; | |
67 | chanindex **wchans = np->exts[wwcnext]; | |
68 | int i; | |
69 | ||
70 | memset(ww->channels, 0, sizeof(ww->channels)); | |
71 | ||
72 | if (!wchans) | |
73 | return; | |
74 | ||
75 | for (i = 0; i < WW_MAXCHANNELS; i++) { | |
76 | if (!wchans[i]) | |
77 | break; | |
78 | ||
79 | wwc_refchannel(wchans[i]); | |
80 | ww->channels[i] = wchans[i]; | |
81 | } | |
82 | } | |
83 | ||
84 | static void wwc_hook_lostrecord(int hooknum, void *arg) { | |
85 | whowas *ww = arg; | |
86 | int i; | |
87 | ||
88 | for (i = 0; i < WW_MAXCHANNELS; i++) { | |
89 | if (!ww->channels[i]) | |
90 | break; | |
91 | ||
92 | wwc_derefchannel(ww->channels[i]); | |
93 | } | |
94 | } | |
95 | ||
96 | void _init(void) { | |
97 | wwcnext = registernickext("whowas_channels"); | |
98 | wwccext = registerchanext("whowas_channels"); | |
99 | ||
100 | registerhook(HOOK_CHANNEL_JOIN, &wwc_hook_joincreate); | |
101 | registerhook(HOOK_CHANNEL_CREATE, &wwc_hook_joincreate); | |
102 | registerhook(HOOK_NICK_LOSTNICK, &wwc_hook_lostnick); | |
103 | registerhook(HOOK_WHOWAS_NEWRECORD, &wwc_hook_newrecord); | |
104 | registerhook(HOOK_WHOWAS_LOSTRECORD, &wwc_hook_lostrecord); | |
105 | } | |
106 | ||
107 | void _fini(void) { | |
108 | releasenickext(wwcnext); | |
109 | releasechanext(wwccext); | |
110 | ||
111 | deregisterhook(HOOK_CHANNEL_JOIN, &wwc_hook_joincreate); | |
112 | deregisterhook(HOOK_CHANNEL_CREATE, &wwc_hook_joincreate); | |
113 | deregisterhook(HOOK_NICK_LOSTNICK, &wwc_hook_lostnick); | |
114 | deregisterhook(HOOK_WHOWAS_NEWRECORD, &wwc_hook_newrecord); | |
115 | deregisterhook(HOOK_WHOWAS_LOSTRECORD, &wwc_hook_lostrecord); | |
116 | } |