!7

!include #1007827 // Slack Bot

sbool doSlack = true;

!include #1008066 // SelectBot with guess_match

static Set<S> botNames = asSet(splitAtSpace("time greet greetbot random thanks thanksbot bye byebot byeall"));

static O extensionModule;

static Throwable extensionModuleError;

extend SelectBot {

  S name;

}

concept ExtensionModule {

  S snippetID;

}

p {

  // DB start & migration

  bootstrapConceptsFrom(#1007829);

  db();

  if (countConcepts(SelectBot) == 1)

    cset(uniq(SelectBot), name := "greet");

  reloadExtensionModule();

  slackSpeed = 1000;

  if (doSlack) initSlackBot();

  dediSay("Back online!");

  if (doSlack) slackBotLoop();

}

answer {

  s = trim(s);

  if (startsWith(s, "!data:", m)) ret struct(getBot($1));

  if (startsWith(s, "!plus:", m)) ret struct(keysWithValue(getBot($1).examples, true));

  if (startsWith(s, "!minus:", m)) ret struct(keysWithValue(getBot($1).examples, false));

  if (eq(s, "!dedicated")) ret yesno(dedicated());

  if (eq(s, "!channel")) {

    Map map = objectToMap(channel!);

    map.remove("token");

    ret structForUser(map);

  }

  // Train bots

  for (S bot : botNames) {

    if (startsWithWord(s, "!" + bot, m)) { getBot(bot).put($1, true); ret "ok"; }

    if (startsWithWord(s, "!no" + bot, m)) { getBot(bot).put($1, false); ret "ok"; }

  }

  // general

  if (eq(s, "!reload")) {

    actionsAfterPost.add(f restart);

    ret "krasser reload";

  }

  if (eq(s, "!source")) ret progLink();

  if (eq(s, "!help")) ret botAutoHelp();

  // for debugging

  if (eq(s, "!username")) ret slackUserName!;

  // extension module

  if (eq(s, "!ext")) ret "Yo: " + joinWithSpace(collect(list(ExtensionModule), "snippetID"));

  if (eq(s, "!ext reload")) {

    reloadExtensionModule();

    ret "OK";

  }

  if (eq(s, "!authed")) ret yn(authed());

  if (authed()) {

    if (startsWithWord(s, "!ext", m)) {

      cset(uniq(ExtensionModule), snippetID := fsI($1));

      reloadExtensionModule();

      ret "OK";

    }

  }

  // match on bots

  if (isMatch("greet", s))

    ret s + " => hi!";

  if (isMatch("time", s))

    ret s + " => It is " + hmsWithColons();

  if (isMatch("random", s))

    ret s + " => " + aGlobalID();

  if (isMatch("thanks", s))

    ret s + " => " + "You're welcome.";

  if (isMatch("bye", s) || isMatch("byeall", s))

    ret s + " => " + "Bye bye!";

  try answer callStaticAnswerMethod(extensionModule, s);

}

static bool isMatch(S bot, S s) {

  ret dedicated() && isMatch2(bot, s) || isMatch2(bot + "bot", s);

}

static bool isMatch2(S bot, S s) {

  ret isTrue(getBot(bot).get(s));

}

static SelectBot getBot(S name) {

  ret uniq(SelectBot, +name);

}

static L<S> botAutoHelp_commands() {

  L<S> cmds = _botAutoHelp_commands();

  for (S bot : botNames)

    addAll(cmds, "!" + bot, "!no" + bot);

  ret cmds;

}

static bool dedicated() {

  Channel c = channel!;

  ret c == null ? false : c.dedicated;

}

sbool authed() { ret eq(slackUserName!, "drjava"); }

svoid reloadExtensionModule {

  cleanUp(extensionModule);

  extensionModule = null;

  extensionModuleError = null;

  for (ExtensionModule m) {

    try {

      extensionModule = run(m.snippetID);

    } catch e {

      extensionModuleError = e;

    }

    break;

  }

}