2021-11-30 02:35:21 +00:00
using System.Text ;
using Humanizer ;
using NodaTime ;
using NodaTime.Text ;
using NodaTime.TimeZones ;
using PluralKit.Core ;
namespace PluralKit.Bot ;
public class Config
{
private record PaginatedConfigItem ( string Key , string Description , string? CurrentValue , string DefaultValue ) ;
public async Task ShowConfig ( Context ctx )
{
var items = new List < PaginatedConfigItem > ( ) ;
2022-06-15 23:28:34 +00:00
var allowAutoproxy = await ctx . Repository . GetAutoproxyEnabled ( ctx . Author . Id ) ;
2021-11-30 02:35:21 +00:00
items . Add ( new (
"autoproxy account" ,
"Whether autoproxy is enabled for the current account" ,
2022-06-15 23:28:34 +00:00
EnabledDisabled ( allowAutoproxy ) ,
2021-11-30 02:35:21 +00:00
"enabled"
) ) ;
items . Add ( new (
"autoproxy timeout" ,
"If this is set, latch-mode autoproxy will not keep autoproxying after this amount of time has elapsed since the last message sent in the server" ,
ctx . Config . LatchTimeout . HasValue
? (
ctx . Config . LatchTimeout . Value ! = 0
? Duration . FromSeconds ( ctx . Config . LatchTimeout . Value ) . ToTimeSpan ( ) . Humanize ( 4 )
: "disabled"
)
: null ,
ProxyMatcher . DefaultLatchExpiryTime . ToTimeSpan ( ) . Humanize ( 4 )
) ) ;
items . Add ( new (
"timezone" ,
"The system's time zone - shows timestamps in your local time" ,
ctx . Config . UiTz ,
"UTC"
) ) ;
items . Add ( new (
"ping" ,
$"Whether other users are able to mention you via a {Emojis.Bell} reaction" ,
EnabledDisabled ( ctx . Config . PingsEnabled ) ,
"enabled"
) ) ;
2021-12-01 16:48:49 +00:00
items . Add ( new (
"private member" ,
"Whether member privacy is automatically set to private for new members" ,
EnabledDisabled ( ctx . Config . MemberDefaultPrivate ) ,
"disabled"
) ) ;
items . Add ( new (
"private group" ,
"Whether group privacy is automatically set to private for new groups" ,
EnabledDisabled ( ctx . Config . GroupDefaultPrivate ) ,
"disabled"
) ) ;
2021-12-06 09:01:42 +00:00
items . Add ( new (
"show private" ,
"Whether private information is shown to linked accounts by default" ,
ctx . Config . ShowPrivateInfo . ToString ( ) . ToLower ( ) ,
"true"
) ) ;
2021-11-30 02:35:21 +00:00
items . Add ( new (
"Member limit" ,
"The maximum number of registered members for your system" ,
ctx . Config . MemberLimitOverride ? . ToString ( ) ,
Limits . MaxMemberCount . ToString ( )
) ) ;
items . Add ( new (
"Group limit" ,
"The maximum number of registered groups for your system" ,
ctx . Config . GroupLimitOverride ? . ToString ( ) ,
Limits . MaxGroupCount . ToString ( )
) ) ;
2022-11-23 08:48:24 +00:00
items . Add ( new (
2022-11-25 04:35:14 +00:00
"Proxy case" ,
"Whether proxy tags are case sensitive" ,
2022-11-23 08:48:24 +00:00
EnabledDisabled ( ctx . Config . CaseSensitiveProxyTags ) ,
"enabled"
) ) ;
2021-11-30 02:35:21 +00:00
await ctx . Paginate < PaginatedConfigItem > (
items . ToAsyncEnumerable ( ) ,
items . Count ,
10 ,
"Current settings for your system" ,
ctx . System . Color ,
( eb , l ) = >
{
var description = new StringBuilder ( ) ;
foreach ( var item in l )
{
description . Append ( item . Key . AsCode ( ) ) ;
description . Append ( $" **({item.CurrentValue ?? item.DefaultValue})**" ) ;
if ( item . CurrentValue ! = null & & item . CurrentValue ! = item . DefaultValue )
description . Append ( "\ud83d\udd39" ) ;
description . AppendLine ( ) ;
description . Append ( item . Description ) ;
description . AppendLine ( ) ;
description . AppendLine ( ) ;
}
eb . Description ( description . ToString ( ) ) ;
2022-02-11 17:25:56 +00:00
// using *large* blue diamond here since it's easier to see in the small footer
eb . Footer ( new ( "\U0001f537 means this setting was changed. Type `pk;config <setting name> clear` to reset it to the default." ) ) ;
2021-11-30 02:35:21 +00:00
return Task . CompletedTask ;
}
) ;
}
2022-01-19 23:59:44 +00:00
private string EnabledDisabled ( bool value ) = > value ? "enabled" : "disabled" ;
2021-11-30 02:35:21 +00:00
public async Task AutoproxyAccount ( Context ctx )
{
2022-06-15 23:28:34 +00:00
var allowAutoproxy = await ctx . Repository . GetAutoproxyEnabled ( ctx . Author . Id ) ;
2022-01-19 23:59:44 +00:00
if ( ! ctx . HasNext ( ) )
2021-11-30 02:35:21 +00:00
{
2022-06-15 23:28:34 +00:00
await ctx . Reply ( $"Autoproxy is currently **{EnabledDisabled(allowAutoproxy)}** for account <@{ctx.Author.Id}>." ) ;
2022-01-19 23:59:44 +00:00
return ;
2021-11-30 02:35:21 +00:00
}
2022-02-11 17:25:56 +00:00
var allow = ctx . MatchToggle ( true ) ;
2021-11-30 02:35:21 +00:00
var statusString = EnabledDisabled ( allow ) ;
2022-06-15 23:28:34 +00:00
if ( allowAutoproxy = = allow )
2021-11-30 02:35:21 +00:00
{
await ctx . Reply ( $"{Emojis.Note} Autoproxy is already {statusString} for account <@{ctx.Author.Id}>." ) ;
return ;
}
var patch = new AccountPatch { AllowAutoproxy = allow } ;
2022-01-22 08:05:01 +00:00
await ctx . Repository . UpdateAccount ( ctx . Author . Id , patch ) ;
2021-11-30 02:35:21 +00:00
await ctx . Reply ( $"{Emojis.Success} Autoproxy {statusString} for account <@{ctx.Author.Id}>." ) ;
}
public async Task AutoproxyTimeout ( Context ctx )
{
if ( ! ctx . HasNext ( ) )
{
var timeout = ctx . Config . LatchTimeout . HasValue
? Duration . FromSeconds ( ctx . Config . LatchTimeout . Value )
: ( Duration ? ) null ;
if ( timeout = = null )
await ctx . Reply ( $"You do not have a custom autoproxy timeout duration set. The default latch timeout duration is {ProxyMatcher.DefaultLatchExpiryTime.ToTimeSpan().Humanize(4)}." ) ;
else if ( timeout = = Duration . Zero )
await ctx . Reply ( "Latch timeout is currently **disabled** for your system. Latch mode autoproxy will never time out." ) ;
else
await ctx . Reply ( $"The current latch timeout duration for your system is {timeout.Value.ToTimeSpan().Humanize(4)}." ) ;
return ;
}
Duration ? newTimeout ;
Duration overflow = Duration . Zero ;
if ( ctx . Match ( "off" , "stop" , "cancel" , "no" , "disable" , "remove" ) ) newTimeout = Duration . Zero ;
2022-02-11 17:25:56 +00:00
else if ( await ctx . MatchClear ( ) ) newTimeout = null ;
2021-11-30 02:35:21 +00:00
else
{
var timeoutStr = ctx . RemainderOrNull ( ) ;
var timeoutPeriod = DateUtils . ParsePeriod ( timeoutStr ) ;
if ( timeoutPeriod = = null ) throw new PKError ( $"Could not parse '{timeoutStr}' as a valid duration. Try using a syntax such as \" 3 h5m \ " (i.e. 3 hours and 5 minutes)." ) ;
if ( timeoutPeriod . Value . TotalHours > 100000 )
{
// sanity check to prevent seconds overflow if someone types in 999999999
overflow = timeoutPeriod . Value ;
newTimeout = Duration . Zero ;
}
else newTimeout = timeoutPeriod ;
}
2022-01-22 08:05:01 +00:00
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { LatchTimeout = ( int? ) newTimeout ? . TotalSeconds } ) ;
2021-11-30 02:35:21 +00:00
if ( newTimeout = = null )
await ctx . Reply ( $"{Emojis.Success} Latch timeout reset to default ({ProxyMatcher.DefaultLatchExpiryTime.ToTimeSpan().Humanize(4)})." ) ;
else if ( newTimeout = = Duration . Zero & & overflow ! = Duration . Zero )
await ctx . Reply ( $"{Emojis.Success} Latch timeout disabled. Latch mode autoproxy will never time out. ({overflow.ToTimeSpan().Humanize(4)} is too long)" ) ;
else if ( newTimeout = = Duration . Zero )
await ctx . Reply ( $"{Emojis.Success} Latch timeout disabled. Latch mode autoproxy will never time out." ) ;
else
await ctx . Reply ( $"{Emojis.Success} Latch timeout set to {newTimeout.Value!.ToTimeSpan().Humanize(4)}." ) ;
}
public async Task SystemTimezone ( Context ctx )
{
if ( ctx . System = = null ) throw Errors . NoSystemError ;
if ( await ctx . MatchClear ( ) )
{
2022-01-22 08:05:01 +00:00
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { UiTz = "UTC" } ) ;
2021-11-30 02:35:21 +00:00
await ctx . Reply ( $"{Emojis.Success} System time zone cleared (set to UTC)." ) ;
return ;
}
var zoneStr = ctx . RemainderOrNull ( ) ;
if ( zoneStr = = null )
{
await ctx . Reply (
2021-11-30 20:40:00 +00:00
$"Your current system time zone is set to **{ctx.Config.UiTz}**. It is currently **{SystemClock.Instance.GetCurrentInstant().FormatZoned(ctx.Config.Zone)}** in that time zone. To change your system time zone, type `pk;config tz <zone>`." ) ;
2021-11-30 02:35:21 +00:00
return ;
}
var zone = await FindTimeZone ( ctx , zoneStr ) ;
if ( zone = = null ) throw Errors . InvalidTimeZone ( zoneStr ) ;
var currentTime = SystemClock . Instance . GetCurrentInstant ( ) . InZone ( zone ) ;
var msg = $"This will change the system time zone to **{zone.Id}**. The current time is **{currentTime.FormatZoned()}**. Is this correct?" ;
if ( ! await ctx . PromptYesNo ( msg , "Change Timezone" ) ) throw Errors . TimezoneChangeCancelled ;
2022-01-22 08:05:01 +00:00
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { UiTz = zone . Id } ) ;
2021-11-30 02:35:21 +00:00
await ctx . Reply ( $"System time zone changed to **{zone.Id}**." ) ;
}
private async Task < DateTimeZone > FindTimeZone ( Context ctx , string zoneStr )
{
// First, if we're given a flag emoji, we extract the flag emoji code from it.
zoneStr = Core . StringUtils . ExtractCountryFlag ( zoneStr ) ? ? zoneStr ;
// Then, we find all *locations* matching either the given country code or the country name.
var locations = TzdbDateTimeZoneSource . Default . Zone1970Locations ;
var matchingLocations = locations . Where ( l = > l . Countries . Any ( c = >
string . Equals ( c . Code , zoneStr , StringComparison . InvariantCultureIgnoreCase ) | |
string . Equals ( c . Name , zoneStr , StringComparison . InvariantCultureIgnoreCase ) ) ) ;
// Then, we find all (unique) time zone IDs that match.
var matchingZones = matchingLocations . Select ( l = > DateTimeZoneProviders . Tzdb . GetZoneOrNull ( l . ZoneId ) )
. Distinct ( ) . ToList ( ) ;
// If the set of matching zones is empty (ie. we didn't find anything), we try a few other things.
if ( matchingZones . Count = = 0 )
{
// First, we try to just find the time zone given directly and return that.
var givenZone = DateTimeZoneProviders . Tzdb . GetZoneOrNull ( zoneStr ) ;
if ( givenZone ! = null ) return givenZone ;
// If we didn't find anything there either, we try parsing the string as an offset, then
// find all possible zones that match that offset. For an offset like UTC+2, this doesn't *quite*
// work, since there are 57(!) matching zones (as of 2019-06-13) - but for less populated time zones
// this could work nicely.
var inputWithoutUtc = zoneStr . Replace ( "UTC" , "" ) . Replace ( "GMT" , "" ) ;
var res = OffsetPattern . CreateWithInvariantCulture ( "+H" ) . Parse ( inputWithoutUtc ) ;
if ( ! res . Success ) res = OffsetPattern . CreateWithInvariantCulture ( "+H:mm" ) . Parse ( inputWithoutUtc ) ;
// If *this* didn't parse correctly, fuck it, bail.
if ( ! res . Success ) return null ;
var offset = res . Value ;
// To try to reduce the count, we go by locations from the 1970+ database instead of just the full database
// This elides regions that have been identical since 1970, omitting small distinctions due to Ancient History(tm).
var allZones = TzdbDateTimeZoneSource . Default . Zone1970Locations . Select ( l = > l . ZoneId ) . Distinct ( ) ;
matchingZones = allZones . Select ( z = > DateTimeZoneProviders . Tzdb . GetZoneOrNull ( z ) )
. Where ( z = > z . GetUtcOffset ( SystemClock . Instance . GetCurrentInstant ( ) ) = = offset ) . ToList ( ) ;
}
// If we have a list of viable time zones, we ask the user which is correct.
// If we only have one, return that one.
if ( matchingZones . Count = = 1 )
return matchingZones . First ( ) ;
// Otherwise, prompt and return!
return await ctx . Choose ( "There were multiple matches for your time zone query. Please select the region that matches you the closest:" , matchingZones ,
z = >
{
if ( TzdbDateTimeZoneSource . Default . Aliases . Contains ( z . Id ) )
return $"**{z.Id}**, {string.Join(" , ", TzdbDateTimeZoneSource.Default.Aliases[z.Id])}" ;
return $"**{z.Id}**" ;
} ) ;
}
2021-12-01 16:48:49 +00:00
2022-01-19 23:59:44 +00:00
public async Task SystemPing ( Context ctx )
{
// note: this is here because this is also used in `pk;system ping`, which does not CheckSystem
ctx . CheckSystem ( ) ;
// todo: move all the other config settings to this format
String Response ( bool isError , bool val )
= > $"Reaction pings are {(isError ? " already " : " currently ")} **{EnabledDisabled(val)}** for your system. "
+ $"To {EnabledDisabled(!val)[..^1]} reaction pings, type `pk;config ping {EnabledDisabled(!val)[..^1]}`." ;
if ( ! ctx . HasNext ( ) )
{
await ctx . Reply ( Response ( false , ctx . Config . PingsEnabled ) ) ;
return ;
}
2022-02-11 17:25:56 +00:00
var value = ctx . MatchToggle ( true ) ;
2022-01-19 23:59:44 +00:00
if ( ctx . Config . PingsEnabled = = value )
await ctx . Reply ( Response ( true , ctx . Config . PingsEnabled ) ) ;
else
{
2022-01-22 08:05:01 +00:00
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { PingsEnabled = value } ) ;
2022-01-19 23:59:44 +00:00
await ctx . Reply ( $"Reaction pings have now been {EnabledDisabled(value)}." ) ;
}
}
2021-12-07 06:58:05 +00:00
2021-12-01 16:48:49 +00:00
public async Task MemberDefaultPrivacy ( Context ctx )
{
if ( ! ctx . HasNext ( ) )
{
if ( ctx . Config . MemberDefaultPrivate ) { await ctx . Reply ( "Newly created members will currently have their privacy settings set to private. To change this, type `pk;config private member off`" ) ; }
else { await ctx . Reply ( "Newly created members will currently have their privacy settings set to public. To automatically set new members' privacy settings to private, type `pk;config private member on`" ) ; }
}
else
{
2022-02-11 17:25:56 +00:00
if ( ctx . MatchToggle ( false ) )
2021-12-01 16:48:49 +00:00
{
2022-01-22 08:05:01 +00:00
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { MemberDefaultPrivate = true } ) ;
2021-12-01 16:48:49 +00:00
await ctx . Reply ( "Newly created members will now have their privacy settings set to private." ) ;
}
2022-01-19 23:59:44 +00:00
else
2021-12-01 16:48:49 +00:00
{
2022-01-22 08:05:01 +00:00
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { MemberDefaultPrivate = false } ) ;
2021-12-01 16:48:49 +00:00
await ctx . Reply ( "Newly created members will now have their privacy settings set to public." ) ;
}
}
}
public async Task GroupDefaultPrivacy ( Context ctx )
{
if ( ! ctx . HasNext ( ) )
{
2021-12-06 09:01:42 +00:00
if ( ctx . Config . GroupDefaultPrivate ) { await ctx . Reply ( "Newly created groups will currently have their privacy settings set to private. To change this, type `pk;config private group off`" ) ; }
2021-12-01 16:48:49 +00:00
else { await ctx . Reply ( "Newly created groups will currently have their privacy settings set to public. To automatically set new groups' privacy settings to private, type `pk;config private group on`" ) ; }
}
else
{
2022-02-11 17:25:56 +00:00
if ( ctx . MatchToggle ( false ) )
2021-12-01 16:48:49 +00:00
{
2022-01-22 08:05:01 +00:00
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { GroupDefaultPrivate = true } ) ;
2021-12-01 16:48:49 +00:00
await ctx . Reply ( "Newly created groups will now have their privacy settings set to private." ) ;
}
2022-01-19 23:59:44 +00:00
else
2021-12-01 16:48:49 +00:00
{
2022-01-22 08:05:01 +00:00
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { GroupDefaultPrivate = false } ) ;
2021-12-01 16:48:49 +00:00
await ctx . Reply ( "Newly created groups will now have their privacy settings set to public." ) ;
}
}
}
2021-12-06 09:01:42 +00:00
public async Task ShowPrivateInfo ( Context ctx )
{
if ( ! ctx . HasNext ( ) )
{
if ( ctx . Config . ShowPrivateInfo ) await ctx . Reply ( "Private information is currently **shown** when looking up your own info. Use the `-public` flag to hide it." ) ;
else await ctx . Reply ( "Private information is currently **hidden** when looking up your own info. Use the `-private` flag to show it." ) ;
return ;
}
2022-02-11 17:25:56 +00:00
if ( ctx . MatchToggle ( true ) )
2021-12-06 09:01:42 +00:00
{
2022-01-22 08:05:01 +00:00
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { ShowPrivateInfo = true } ) ;
2021-12-06 09:01:42 +00:00
await ctx . Reply ( "Private information will now be **shown** when looking up your own info. Use the `-public` flag to hide it." ) ;
}
2022-01-19 23:59:44 +00:00
else
2021-12-06 09:01:42 +00:00
{
2022-01-22 08:05:01 +00:00
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { ShowPrivateInfo = false } ) ;
2021-12-06 09:01:42 +00:00
await ctx . Reply ( "Private information will now be **hidden** when looking up your own info. Use the `-private` flag to show it." ) ;
}
}
2022-11-23 08:48:24 +00:00
public async Task CaseSensitiveProxyTags ( Context ctx )
{
if ( ! ctx . HasNext ( ) )
{
2022-11-25 04:35:14 +00:00
if ( ctx . Config . CaseSensitiveProxyTags ) { await ctx . Reply ( "Proxy tags are currently case **sensitive**." ) ; }
else { await ctx . Reply ( "Proxy tags are currently case **insensitive**." ) ; }
2022-11-23 08:48:24 +00:00
return ;
}
if ( ctx . MatchToggle ( true ) )
{
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { CaseSensitiveProxyTags = true } ) ;
2022-11-25 04:35:14 +00:00
await ctx . Reply ( "Proxy tags are now case sensitive." ) ;
2022-11-23 08:48:24 +00:00
}
else
{
await ctx . Repository . UpdateSystemConfig ( ctx . System . Id , new ( ) { CaseSensitiveProxyTags = false } ) ;
2022-11-25 04:35:14 +00:00
await ctx . Reply ( "Proxy tags are now case insensitive." ) ;
2022-11-23 08:48:24 +00:00
}
}
2021-11-30 02:35:21 +00:00
}