/[sudobot]/trunk/src/services/ChannelLockManager.ts
ViewVC logotype

Diff of /trunk/src/services/ChannelLockManager.ts

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 241 by rakin, Mon Jul 29 17:29:10 2024 UTC revision 244 by rakin, Mon Jul 29 17:29:11 2024 UTC
# Line 1  Line 1 
1  import { GuildChannel, PermissionString, User } from "discord.js";  import { GuildChannel, PermissionString, Role, RoleResolvable, User } from "discord.js";
2  import MessageEmbed from "../client/MessageEmbed";  import MessageEmbed from "../client/MessageEmbed";
3  import ChannelLock from "../models/ChannelLock";  import ChannelLock from "../models/ChannelLock";
4  import Service from "../utils/structures/Service";  import Service from "../utils/structures/Service";
# Line 7  export interface ChannelLockOptions { Line 7  export interface ChannelLockOptions {
7      reason?: string;      reason?: string;
8      sendConfirmation?: boolean;      sendConfirmation?: boolean;
9      force?: boolean;      force?: boolean;
10        role?: RoleResolvable;
11  }  }
12    
13  export default class ChannelLockManager extends Service {  export default class ChannelLockManager extends Service {
14      async lock(channel: GuildChannel, user: User, { reason, sendConfirmation }: ChannelLockOptions = {}) {      async lock(channel: GuildChannel, user: User, { reason, sendConfirmation, role }: ChannelLockOptions = {}) {
15            const lockRole = role ? (role instanceof Role ? role : (await channel.guild.roles.fetch(role))!) : channel.guild.roles.everyone;
16    
17          const channelLock = await ChannelLock.findOne({          const channelLock = await ChannelLock.findOne({
18              where: {              where: {
19                  channel_id: channel.id,                  channel_id: channel.id,
20                  guild_id: channel.guild.id                  guild_id: channel.guild.id,
21                    role_id: lockRole.id
22              }              }
23          });          });
24    
25          if (channelLock) {          if (channelLock) {
26                console.log('exists');
27              return false;              return false;
28          }          }
29    
30          const permissions = channel.permissionOverwrites.cache.get(channel.guild.id);          console.log(lockRole?.name);
           
         if (permissions) {  
             const permJson = {  
                 allow: permissions.allow.toArray(),  
                 deny: permissions.deny.toArray(),  
             };  
31    
32              console.log(permJson);          let permissions = channel.permissionOverwrites.cache.get(lockRole.id);
33    
34              await ChannelLock.create({          const permJson = {
35                  user_id: user.id,              allow: permissions?.allow?.toArray() ?? null,
36                  guild_id: channel.guild.id,              deny: permissions?.deny?.toArray() ?? null,
37                  channel_id: channel.id,          };
                 reason,  
                 previous_perms: permJson  
             });  
         }  
38    
39          await channel.permissionOverwrites.edit(channel.guild.id, {          console.log(permJson);
40    
41            await ChannelLock.create({
42                user_id: user.id,
43                guild_id: channel.guild.id,
44                channel_id: channel.id,
45                reason,
46                previous_perms: permJson,
47                role_id: lockRole.id
48            });
49    
50            await channel.permissionOverwrites.edit(lockRole, {
51              SEND_MESSAGES: false,              SEND_MESSAGES: false,
52              SEND_MESSAGES_IN_THREADS: false,              SEND_MESSAGES_IN_THREADS: false,
             ADD_REACTIONS: false,  
53              REQUEST_TO_SPEAK: false,              REQUEST_TO_SPEAK: false,
54              SPEAK: false,              SPEAK: false,
55          }, { reason });          }, { reason });
56    
57            console.log('success');
58    
59          if (sendConfirmation && channel.isText()) {          if (sendConfirmation && channel.isText()) {
60              await channel.send({              await channel.send({
61                  embeds: [                  embeds: [
# Line 71  export default class ChannelLockManager Line 78  export default class ChannelLockManager
78          return true;          return true;
79      }      }
80    
81      async unlock(channel: GuildChannel, { reason, sendConfirmation, force }: ChannelLockOptions = {}) {      async unlock(channel: GuildChannel, { reason, sendConfirmation, force, role }: ChannelLockOptions = {}) {
82            const lockRole = role ? (role instanceof Role ? role : (await channel.guild.roles.fetch(role))!) : channel.guild.roles.everyone;
83    
84          const channelLock = await ChannelLock.findOne({          const channelLock = await ChannelLock.findOne({
85              where: {              where: {
86                  channel_id: channel.id,                  channel_id: channel.id,
87                  guild_id: channel.guild.id                  guild_id: channel.guild.id,
88                    role_id: lockRole.id
89              }              }
90          });          });
91    
92          const permissions = channelLock?.get('previous_perms') as { allow: PermissionString[], deny: PermissionString[] };          if (!channelLock) {
93                console.log('Channel not locked');            
94                return false;
95            }
96    
97            const permissions = channelLock?.get('previous_perms') as { allow: PermissionString[] | null, deny: PermissionString[] | null };
98    
99          if (!permissions && !force) {          if (!permissions && !force) {
100                console.log('Permission error');
101              return false;              return false;
102          }          }
103    
104          const transform = (key: PermissionString) => {          const transform = (key: PermissionString) => {
105                if (!permissions.allow || !permissions.deny) {
106                    return undefined;
107                }
108    
109              if (!permissions) {              if (!permissions) {
110                  return force ? true : undefined;                  return force ? true : undefined;
111              }              }
# Line 101  export default class ChannelLockManager Line 121  export default class ChannelLockManager
121              }              }
122          };          };
123    
124          await channel.permissionOverwrites.edit(channel.guild.id, {          if (!permissions.allow && !permissions.deny) {
125              SEND_MESSAGES: transform('SEND_MESSAGES'),              await channel.permissionOverwrites.delete(lockRole);
126              SEND_MESSAGES_IN_THREADS: transform('SEND_MESSAGES_IN_THREADS'),          }
127              ADD_REACTIONS: transform('ADD_REACTIONS'),          else {
128              REQUEST_TO_SPEAK: transform('REQUEST_TO_SPEAK'),              await channel.permissionOverwrites.edit(lockRole, {
129              SPEAK: transform('SPEAK'),                  SEND_MESSAGES: transform('SEND_MESSAGES'),
130          }, { reason });                  SEND_MESSAGES_IN_THREADS: transform('SEND_MESSAGES_IN_THREADS'),
131                    REQUEST_TO_SPEAK: transform('REQUEST_TO_SPEAK'),
132                    SPEAK: transform('SPEAK'),
133                }, { reason });
134            }        
135    
136          await channelLock?.destroy();          await channelLock?.destroy();
137    
# Line 133  export default class ChannelLockManager Line 157  export default class ChannelLockManager
157          return true;          return true;
158      }      }
159    
160      lockAll(channels: GuildChannel[], user: User, options: ChannelLockOptions = {}) {      async lockAll(channels: GuildChannel[], user: User, options: ChannelLockOptions = {}) {
161          return Promise.all(channels.map(c => this.lock(c, user, options)));          let success = 0, failure = 0;
162    
163            for await (const channel of channels) {
164                console.log('Locking', channel.name);
165                
166                if (await this.lock(channel, user, options)) {
167                    success++;
168                }
169                else {
170                    failure++;
171                }
172            }
173    
174            return [success, failure];
175      }      }
176    
177      unlockAll(channels: GuildChannel[], options: ChannelLockOptions = {}) {      async unlockAll(channels: GuildChannel[], options: ChannelLockOptions = {}) {
178          return Promise.all(channels.map(c => this.unlock(c, options)));          let success = 0, failure = 0;
179    
180            for await (const channel of channels) {
181                console.log('Unlocking', channel.name);
182    
183                if (await this.unlock(channel, options)) {
184                    success++;
185                }
186                else {
187                    failure++;
188                }
189            }
190    
191            return [success, failure];
192      }      }
193  }  }
Legend:
Removed from v.241  
changed lines
  Added in v.244
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26