daan/geekbot
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity

Add more run parameters, remove first launch stuff, store token in globals

Browse source
This commit is contained in:
runebaas 2018-05-13 18:47:01 +02:00
parent bb8aee1eda
commit a1b5bd1955
No known key found for this signature in database
GPG key ID: 2677AF508D0300D6
4 changed files with 56 additions and 56 deletions
Download patch file Download diff file Expand all files Collapse all files

18
Geekbot.net/Lib/ReactionListener/ReactionListener.cs
View file

@ -9,22 +9,22 @@ namespace Geekbot.net.Lib.ReactionListener
{
public class ReactionListener : IReactionListener
{
private readonly IDatabase _database;
private readonly IDatabase _redis;
private Dictionary<string, Dictionary<IEmote, ulong>> _listener;
public ReactionListener(IDatabase database)
public ReactionListener(IDatabase redis)
{
_database = database;
_redis = redis;
LoadListeners();
}
private Task LoadListeners()
{
var ids = _database.SetMembers("MessageIds");
var ids = _redis.SetMembers("MessageIds");
_listener = new Dictionary<string, Dictionary<IEmote, ulong>>();
foreach (var id in ids)
{
var reactions = _database.HashGetAll($"Messages:{id}");
var reactions = _redis.HashGetAll($"Messages:{id}");
var messageId = id;
var emojiDict = new Dictionary<IEmote, ulong>();
foreach (var r in reactions)
@ -54,12 +54,12 @@ namespace Geekbot.net.Lib.ReactionListener
public Task AddRoleToListener(string messageId, IEmote emoji, IRole role)
{
if (_database.SetMembers("MessageIds").All(e => e.ToString() != messageId))
if (_redis.SetMembers("MessageIds").All(e => e.ToString() != messageId))
{
_database.SetAdd("MessageIds", messageId);
_redis.SetAdd("MessageIds", messageId);
}
_database.HashSet($"Messages:{messageId}", new[] {new HashEntry(emoji.ToString(), role.Id.ToString())});
_database.SetAdd("MessageIds", messageId);
_redis.HashSet($"Messages:{messageId}", new[] {new HashEntry(emoji.ToString(), role.Id.ToString())});
_redis.SetAdd("MessageIds", messageId);
if (_listener.ContainsKey(messageId))
{
_listener[messageId].Add(emoji, role.Id);

47
Geekbot.net/Lib/RunParameters.cs
View file

@ -1,20 +1,17 @@
using System;
using CommandLine;
using CommandLine;
namespace Geekbot.net.Lib
{
public class RunParameters
{
/**
* General Parameters
*/
[Option('V', "verbose", Default = false, HelpText = "Prints all messages to standard output.")]
/************************************
* General *
************************************/
[Option('V', "verbose", Default = false, HelpText = "Logs everything.")]
public bool Verbose { get; set; }
[Option('r', "reset", Default = false, HelpText = "Resets the bot")]
public bool Reset { get; set; }
[Option('j', "log-json", Default = false, HelpText = "Logs messages as json")]
[Option('j', "log-json", Default = false, HelpText = "Logger outputs json")]
public bool LogJson { get; set; }
[Option('a', "disable-api", Default = false, HelpText = "Disables the web api")]
@ -26,9 +23,10 @@ namespace Geekbot.net.Lib
[Option("token", Default = null, HelpText = "Set a new bot token")]
public string Token { get; set; }
/**
* Database Stuff
*/
/************************************
* Database *
************************************/
[Option("in-memory", Default = false, HelpText = "Uses the in-memory database instead of postgresql")]
public bool InMemory { get; set; }
@ -47,5 +45,28 @@ namespace Geekbot.net.Lib
[Option("db-password", Default = "", HelpText = "Set a posgresql password")]
public string DbPassword { get; set; }
/************************************
* Redis *
************************************/
[Option("redis-host", Default = "127.0.0.1", HelpText = "Set a redis host")]
public string RedisHost { get; set; }
[Option("redis-port", Default = "6379", HelpText = "Set a redis port")]
public string RedisPort { get; set; }
[Option("redis-database", Default = "6", HelpText = "Select a redis database (1-15)")]
public string RedisDatabase { get; set; }
/************************************
* WebApi *
************************************/
[Option("api-host", Default = "127.0.0.1", HelpText = "Host on which the WebApi listens")]
public string ApiHost { get; set; }
[Option("api-port", Default = "12995", HelpText = "Port on which the WebApi listens")]
public string ApiPort { get; set; }
}
}