feat: Database and replay thread

Author: gcarreno

pasirclogbot.example.config

@@ -1,7 +1,9 @@
 [IRC]
-Host="localhost"
+Host="irc.libera.chat"
 Port=6667
 NickName="[PasLogBot]"
 UserName="paslogbot"
 RealName="IRC channel #pascal log bot"
-Channel="#pascal"
+Channel="#PasLogBot"
+[DB]
+File="paslogbot.db"

paslogbot.db

@@ -9,37 +9,37 @@ interface
 , SysUtils
 , IdContext
 , IdIRC
+, IRCLogBot.Config
+, IRCLogBot.Database
 ;
 
 type
 { TIRCLogBot }
   TIRCLogBot = class(TObject)
   private
     FIRC: TIdIRC;
-    FNickName: String;
-    FUserName: String;
-    FRealName: String;
-    FHost: String;
-    FPort: Word;
-    FChannel: String;
-
     FJoinedChannel: Boolean;
 
+    FConfig: TBotConfig;
+
+    FDB: TDatabase;
+
     procedure OnConnected(Sender: TObject);
     procedure OnDisconnected(Sender: TObject);
     procedure OnNotice(ASender: TIdContext; const ANickname, AHost,
       ATarget, ANotice: String);
+    procedure OnServerQuit(ASender: TIdContext; const ANickname, AHost,
+      AServer, AReason: String);
     procedure OnJoin(ASender: TIdContext; const ANickname, AHost,
       AChannel: String);
     procedure OnPrivateMessage(ASender: TIdContext; const ANickname, AHost,
       ATarget, AMessage: String);
 
     procedure Help(const ATarget: String);
-    procedure Replay(const ATarget: String; Count: Integer);
+    procedure Replay(const ATarget: String; ACount: Integer);
   protected
   public
-    constructor Create(AHost: String; APort: Word;
-      ANickName, AUserName, ARealName, AChannel: String);
+    constructor Create(const AConfig: TBotConfig);
     destructor Destroy; override;
 
     procedure Run;
@@ -53,6 +53,70 @@ implementation
   IRCLogBot.Common
 ;
 
+type
+{ TReplayThread }
+  TReplayThread = class(TTHread)
+  private
+    FIRC: TIdIRC;
+    FTarget: String;
+    FLines: TStringList;
+  protected
+    procedure Execute; override;
+  public
+    constructor Create(const AIRC: TIdIRC; const ATarget: String;
+      const ALines: TStringList);
+  published
+  end;
+
+{ TReplayThread }
+
+procedure TReplayThread.Execute;
+var
+  line: String;
+  index: Integer = 0;
+begin
+  if not FIRC.Connected then;
+  begin
+    debug('Exiting replay thread due not being connected.');
+    exit;
+  end;
+  try
+    FIRC.Say(FTarget, '!! --> To avoid triggering flooding, for each 5 lines, I will pause for 5 seconds <-- !!');
+    FIRC.Say(FTarget, Format('*** Here are the last %d lines ***', [FLines.Count]));
+    for line in FLines do
+    begin
+      if (Terminated) or (not FIRC.Connected) then
+      begin
+        debug('Exiting replay thread due to termination or not being connected.');
+        exit;
+      end;
+      debug('Sending #%d: "%s"', [index, line]);
+      Inc(index);
+      FIRC.Say(FTarget, line);
+      if (index mod 5) = 0 then
+      begin
+        debug('Pausing');
+        Sleep(5000);
+      end;
+    end;
+    FIRC.Say(FTarget, Format('*** End of the last %d lines ***', [FLines.Count]));
+  finally
+    FLines.Free;
+  end;
+end;
+
+constructor TReplayThread.Create(const AIRC: TIdIRC; const ATarget: String;
+      const ALines: TStringList);
+begin
+  inherited Create(True);
+  FIRC:= AIRC;
+  FTarget:= ATarget;
+  FLines:= TStringList.Create;
+  FLines.Text := ALines.Text;
+  FreeOnTerminate:= True;
+  Start;
+end;
+
 { TIRCLogBot }
 
 procedure TIRCLogBot.OnConnected(Sender: TObject);
@@ -68,21 +132,34 @@ procedure TIRCLogBot.OnDisconnected(Sender: TObject);
 procedure TIRCLogBot.OnNotice(ASender: TIdContext; const ANickname, AHost,
   ATarget, ANotice: String);
 begin
-  debug('>> NOTICE: <%s> "%s"', [
+  debug('>> NOTICE: <%s:%s> (%s) "%s"', [
     ANickname,
+    AHost,
+    ATarget,
     ANotice
   ]);
 end;
 
+procedure TIRCLogBot.OnServerQuit(ASender: TIdContext; const ANickname, AHost,
+  AServer, AReason: String);
+begin
+  debug('>> QUIT: <%s:%s> %s "%s"',[
+    ANickname,
+    AHost,
+    AServer,
+    AReason
+  ]);
+end;
+
 procedure TIRCLogBot.OnJoin(ASender: TIdContext; const ANickname, AHost,
   AChannel: String);
 begin
-  debug('>> JOIN: <%s@%s> %s', [
+  debug('>> JOIN: <%s:%s> %s', [
     ANickname,
     AHost,
     AChannel
   ]);
-  if (ANickname = FNickName) and (AChannel = FChannel) then
+  if (ANickname = FConfig.NickName) and (AChannel = FConfig.Channel) then
   begin
     debug('Successfully joined my channel');
     FJoinedChannel:= True;
@@ -96,13 +173,24 @@ procedure TIRCLogBot.OnPrivateMessage(ASender: TIdContext; const ANickname,
   strings: TStringArray;
   count: Integer;
 begin
-  debug('>> PRIVMSG: <%s@%s>(%s) "%s"', [
+  debug('>> PRIVMSG: <%s:%s>(%s) "%s"', [
     ANickname,
     AHost,
     ATarget,
     AMessage
   ]);
-  if ATarget = FNickName then
+  if ATarget = FConfig.Channel then
+  begin
+    debug('Inserting: %s, %s, %s, %s', [
+      ANickname,
+      AHost,
+      ATarget,
+      AMessage
+    ]);
+    FDB.Insert(ANickname, ATarget, AMessage);
+    exit;
+  end;
+  if ATarget = FConfig.NickName then
   begin
     if Pos('.', AMessage) = 1 then
     begin
@@ -115,18 +203,23 @@ procedure TIRCLogBot.OnPrivateMessage(ASender: TIdContext; const ANickname,
       if Pos('.replay', Trim(AMessage)) = 1 then
       begin
         strings:= AMessage.Split([' ']);
+        //debug('Strings: %d', [Length(strings)]);
         try
           if Length(strings) > 1 then
           begin
             count:= StrToInt(strings[1]);
           end
           else
           begin
-            count:= 0;
+            count:= 10;
           end;
+          //debug('Count: %d', [count]);
           Replay(ANickname, count);
         except
-          FIRC.Say(ANickname, 'That <count> is not a number.');
+          on e:Exception do
+          begin
+            FIRC.Say(ANickname, 'Something went wrong: "' + e.Message + '". It''s been logged. Please contact the admin if I stop working.');
+          end;
         end;
         exit;
       end;
@@ -136,6 +229,7 @@ procedure TIRCLogBot.OnPrivateMessage(ASender: TIdContext; const ANickname,
       debug('No command.');
       FIRC.Say(ANickname, 'Not a command. Please use ".help" to see a list of commands.');
     end;
+    exit;
   end;
 end;
 
@@ -147,10 +241,16 @@ procedure TIRCLogBot.Help(const ATarget: String);
   FIRC.Say(ATarget, '.replay [count] - Raplays last <count> lines. Default is last 10 lines.');
 end;
 
-procedure TIRCLogBot.Replay(const ATarget: String; Count: Integer);
+procedure TIRCLogBot.Replay(const ATarget: String; ACount: Integer);
+var
+  replayThread: TReplayThread;
+  lines: TStringList;
 begin
-  debug('Replay command.');
-  FIRC.Say(ATarget, Format('Not fully implemented yet: %d',[Count]));
+  debug('Replay command(%d).', [ACount]);
+  lines:= FDB.Get(ACount);
+  debug('Lines: %d', [lines.Count]);
+  replayThread:= TReplayThread.Create(FIRC, ATarget, lines);
+  lines.Free;
 end;
 
 procedure TIRCLogBot.Run;
@@ -164,9 +264,9 @@ procedure TIRCLogBot.Run;
       debug('Error connecting: %s', [e.Message]);
     end;
   end;
-  debug('Joining channel: "%s"...', [FChannel]);
+  debug('Joining channel: "%s"...', [FConfig.Channel]);
   try
-    FIRC.Join(FChannel);
+    FIRC.Join(FConfig.Channel);
   except
     on e:Exception do
     begin
@@ -181,7 +281,8 @@ procedure TIRCLogBot.Shutdown;
   begin
     debug('Disconnecting...');
     try
-      FIRC.Disconnect('Need to go and have a wee nap.');
+      if FJoinedChannel then FIRC.Say(FConfig.Channel, 'Boss sais I need to have a wee nap. See Y''All later...');
+      FIRC.Disconnect('ZzZzZzZzZzZzZzZz...');
     except
       on e:Exception do
       begin
@@ -191,33 +292,41 @@ procedure TIRCLogBot.Shutdown;
   end;
 end;
 
-constructor TIRCLogBot.Create(AHost: String; APort: Word; ANickName, AUserName,
-  ARealName, AChannel: String);
+constructor TIRCLogBot.Create(const AConfig: TBotConfig);
 begin
-  FNickname:= ANickname;
-  FUsername:= AUserName;
-  FRealName:= ARealName;
-  FHost:= AHost;
-  FPort:= APort;
-  FChannel:= AChannel;
+  FJoinedChannel:= False;
+
+  // Config
+  FConfig:= AConfig;
 
+  // Setup IRC Client
   FIRC:= TIdIRC.Create;
-  FIRC.Nickname:= FNickName;
-  FIRC.Username:= FUserName;
-  FIRC.RealName:= FRealName;
-  FIRC.Host:= FHost;
-  FIRC.Port:= FPort;
+  FIRC.Nickname:= FConfig.NickName;
+  FIRC.Username:= FConfig.UserName;
+  FIRC.RealName:= FConfig.RealName;
+  FIRC.Host:= FConfig.Host;
+  FIRC.Port:= FConfig.Port;
   FIRC.OnConnected:= @OnConnected;
   FIRC.OnDisconnected:= @OnDisconnected;
+  FIRC.OnServerQuit:= @OnServerQuit;
   FIRC.OnJoin:= @OnJoin;
   FIRC.OnNotice:= @OnNotice;
   FIRC.OnPrivateMessage:= @OnPrivateMessage;
 
-  FJoinedChannel:= False;
+  // Setup Database
+  try
+    FDB:= TDatabase.Create(FConfig.Database);
+  except
+    on e:Exception do
+    begin
+      debug('Error creating db: ', [e.Message]);
+    end;
+  end;
 end;
 
 destructor TIRCLogBot.Destroy;
 begin
+  FDB.Free;
   FIRC.Free;
   inherited Destroy;
 end;