!752

static S dataProgID = "#1002143"; // use data dir of "These Are Bot" v1 (don't use both bots at once...)

// TODO: make a PersistentMultiMap?

static new MultiMap<S, S> lists;

static new MultiMap<S, S> notLists;

p {

  load(dataProgID, "lists");

  load(dataProgID, "notLists");

  makeBot("These Are Bot.");

}

synchronized answer {

  L<S> tok = javaTok(s);

  if (eqic(get(tok, 1), "these") && eqic(get(tok, 3), "are") && eqic(get(tok, 7), ":")) {

    S kind = unquote(get(tok, 5));

    L<S> rest = subList(tok, 8);

    ret addThem(kind, rest, true);

  }

  if (eqic(get(tok, 1), "these") && eqic(get(tok, 3), "are") && eqic(get(tok, 5), "not") && eqic(get(tok, 9), ":")) {

    S kind = unquote(get(tok, 7));

    L<S> rest = subList(tok, 10);

    ret addThem(kind, rest, false);

  }

  if (match("name a *", s, m)) {

    S kind = singular(simplify(m.unq(0)));

    L<S> items = lists.get(kind);

    if (empty(items))

      ret null; //"I don't know any " + m.unq(0);

    ret randomOne(items);

  }

  if (match("name * *", s, m)) {

    S n = m.get(0);

    if (isInteger(n)) {

      int _n = min(100, parseInt(n));

      S kind = singular(simplify(m.unq(1)));

      L<S> items = cloneList(lists.get(kind));

      if (l(items) < _n)

        ret "I don't know that many " + m.unq(1);

      new L<S> l;

      for (int i = 0; i < _n; i++) {

        S x = randomOne(items);

        l.add(x);

        items.remove(x);

      }

      ret join(", ", l);

    }

  }

  S userName = cast callOpt(mainBot, "getUserName");

  if (match("am i a *", s, m)) {

    if (empty(userName))

      ret "Well I don't know who you are! Please join slack.com";

    S kind = m.unq(0);

    boolean result = containsIgnoreCase(lists.get(kind), userName)

      || containsIgnoreCase(lists.get(kind), dropPrefix("@", userName));

    if (result)

      ret format("Yes, you are a *!", kind);

    else

      ret format("As far as I know, you are not a *.", kind);

  }

  if (match("is * a *", s, m)) {

    S instance = m.unq(0);

    S kind = singular(simplify(m.unq(1)));

    boolean result = containsIgnoreCase(lists.get(kind), instance);

    if (result)

      ret format("Yes, * is a *", instance, kind);

    else

      ret null;

      //ret format("I don't know that * is a *", instance, kind);

  }

  if (match("Are there any * other than *?", s, m)) {

    S kind = singular(simplify(m.unq(0)));

    S instance = m.unq(1);

    L<S> items = cloneList(lists.get(kind));

    removeIgnoreCase(items, instance);

    if (empty(items))

      ret format("No, * is the only *.", instance, kind);

    else

      ret format("Yes, for example *", randomOne(items));

  }

  if (match("list all *", s, m)) {

    S kind = singular(simplify(m.unq(0)));

    L<S> items = lists.get(kind);

    if (empty(items))

      ret null; // format("I don't know any *", kind);

    else

      ret join(", ", items);

  }

  if (match("list all ex *", s, m)) {

    S kind = singular(simplify(m.unq(0)));

    L<S> items = notLists.get(kind);

    if (empty(items))

      ret null; // format("I don't know any ex *", kind);

    else

      ret join(", ", items);

  }

}

static S simplify(S s) {

  ret toLower(s);

}

static S singular(S s) {

  if (s.endsWith("ies"))

    ret dropSuffix("ies", s) + "y";

  ret dropSuffix("s", s);

}

static S addThem(S kind, L<S> rest, boolean plus) {

  kind = singular(simplify(kind));

  boolean comma = rest.contains(",");

  new L<S> items;

  if (comma)

    for (L<S> item : splitListBy(rest, ","))

      items.add(trim(join(simpleSpaces(item))));

  else

    items.addAll(codeTokensOnly(rest));

  printFormat((plus ? "Adding" : "Removing") + " items * to/from kind *", items, kind);

  (plus ? lists : notLists).addAllIfNotThere(kind, items);

  (plus ? notLists : lists).removeAll(kind, items);

  save(dataProgID, "lists");

  save(dataProgID, "notLists");

  ret format("OK, now " + l(lists.get(kind)) + " items in kind *", kind);

}