!7

!include once #1017511 // Telegram Bots

import org.telegram.telegrambots.generics.BotSession;
import org.telegram.telegrambots.api.objects.inlinequery.InlineQuery;
import org.telegram.telegrambots.api.methods.*;
import org.telegram.telegrambots.api.objects.inlinequery.result.*;
import org.telegram.telegrambots.api.objects.inlinequery.inputmessagecontent.*;

static TelegramLongPollingBot bot;
static BotSession botSession;

sbool useWitAI = true; // only on Linux (but that can be changed) & when you have a token

p {
  fS apiToken = telegramBotToken_mandatory();

  ApiContextInitializer.init();
  bot = new TelegramLongPollingBot {
    public S getBotUsername() { ret "HelloComputer_bot"; }
    public S getBotToken() { ret apiToken; }
    
    public void onUpdateReceived(Update update) {
      pcall {
        print("Got update! " + update);
        
        InlineQuery iq = update.getInlineQuery();
        if (iq != null) {
          answerInlineQuery(new AnswerInlineQuery()
            .setInlineQueryId(iq.getId())
            .setResults(new InlineQueryResultArticle()
              .setId(aGlobalID())
              .setTitle("Title.")
              .setDescription("Description.")
              .setInputMessageContent(
                new InputTextMessageContent()
                  .setMessageText("Hello there. You said: " + iq.getQuery()))));
        }
        
        Message msg = update.getMessage(); //update.getChannelPost();
        logQuotedWithDate(telegramFullUpdatesLogFile(), str(update));
        if (msg == null) ret;
        final Chat chat = msg.getChat();
        final User from = msg.getFrom();
        
        final Map map = litorderedmap(
          where := chat == null ? null : chat.getId(),
          type := 'heard,
          date := localDateWithMilliseconds(),
          globalID := aGlobalID());

        Voice voice = msg.getVoice();
        if (voice != null) {
          Int duration = voice.getDuration();
          fS mimeType = voice.getMimeType();
          S fileID = voice.getFileId();
          print("Incoming voice message, downloading");
          printStructs(+duration, +mimeType, +fileID);
          
          replace TgFile with org.telegram.telegrambots.api.objects.File.
          
          // Download voice message
          new GetFile getFileMethod;
          getFileMethod.setFileId(fileID);
          TgFile tgFile = execute(getFileMethod);
          
          downloadFileAsync(tgFile, new DownloadFileCallback<TgFile> {
            public void onResult(TgFile file, java.io.File f) {
              print("Got voice message: " + fileInfo(f));
              // Assuming ogg
              final File f2 = javaxDataDir("Stored Voice Messages/" + (from != null ? from.getId() : 0) + "-" + now() + ".ogg");
              copyFileVerbose(f, f2);
              if (useWitAI && nempty(witAITokenOpt())) thread "Wit.AI" {
                File f3 = f2;
                S _mimeType = mimeType;
                if (eq(mimeType, "audio/ogg")) {
                  pcall {
                    oggToMP3(f2, f3 = replaceFileExtension(f2, ".mp3"));
                  }
                  if (fileEmpty(f3)) fail("Can't convert to mp3");
                  _mimeType = "audio/mpeg3";
                  printFileInfo(f3);
                }

                fS text = witAI_recognizeAudio(f3, _mimeType);
                if (nempty(text)) {
                  putAll(map, litorderedmap(
                    +text,
                    fromVoice := true));
                  logStructure(telegramLogFile(), map);
                  botAppendToMechList_combining("Telegram Bot Log", struct(map));
                  // Allow think bot to answer, then post what was said
                  doAfter(5000, r { telegram_postToSendLog(
                    "(I think you said: " + quote(text) + ")",
                    chat == null ? null : chat.getId());
                  });
                }
              }
            }
            
            public void onException(TgFile file, Exception e) {
              print("Couldn't download voice message: " + e);
            }
          });
        }
        
        if (msg.hasText()) {
          S text = msg.getText();
          print("Incoming text: " + text);
          putAll(map, litorderedmap(
            originalDate := msg.getDate(),
            +text));
            
          if (from != null)
            putAll(map, litorderedmap(
              userID := from.getId(),
              userFirstName := from.getFirstName(),
              userLastName := from.getLastName()));

          logStructure(telegramLogFile(), map);
          botAppendToMechList_combining("Telegram Bot Log", struct(map));
        }
      }
    }
  };

  botSession = new TelegramBotsApi().registerBot(bot);
  
  watchStructureLog_future(telegramBotToSendLog(), 100, voidfunc(Map map) {
    if (eq(map.get('action), 'send)) {
      long chatID = or((Long) _get(map, 'where), telegram_chatID());
      S globalID = or2(getString(map, 'globalID), aGlobalID());
      S voice = getString(map, 'voice);
      S text = getString(map, 'text);
      
      if (voice != null)
        telegram_sendVoiceMessageWithLogging(bot, chatID, text, voice, globalID);
      else
        telegram_sendWithLogging(bot, chatID, text, globalID);
        
      putAll(map, litorderedmap(
        date := localDateWithMilliseconds(),
        type := 'sent,
        +globalID,
        +chatID));
      botAppendToMechList_combining("Telegram Bot Log", struct(map));
    }
  });
  
  print("Telegram Bot running.");
}

svoid cleanMeUp {
  if (botSession != null) {
    final BotSession _botSession = botSession;
    botSession = null;
    if (vmExiting()) {
      print("Closing bot session during VM close");
      // As stop() takes unusually long, don't bother waiting the whole time
      joinThreadForSomeTime(startThread(r { _botSession.stop(); }), 2000);
    } else {
      print("Closing bot session with profiling");
      temp relicense();
      temp profileThisThreadToConsole();
      _botSession.stop();
      print("Closed bot session");
    }
  }
}