!7

cmodule IWantYouToBePrelude > DynPrintLog {
  // the spaces
  
  sS objectiveSpace = 'objective, computerSpace = 'computery, userSpace = 'usery;
  
  // the names
  
  transient S userName = "usery", computerName = "computery";
  
  // the facts
  
  transient Set<S> facts = litciset(
    "Computery is usery's assistant",
    "It is impossible that {Computery is usery's mother}");

  // the rules

  transient WordTokRule rule_iWantYouToBe = wordTokReplacerRule(
    "I want you to be my assistant", "assistant", "I want that {you are my assistant}");
    
  transient WordTokRule rule_useryWants = new WordTokRule("Usery wants that X", "X") {
    run {
      S x_simplified = simpleSpaces_nTok(getVar("X"));
      if (contains(facts, x_simplified))
        emit("But " + x_simplified + " already");
      else if (contains(facts, "It is impossible that " + curly(x_simplified) + "}"))
        emit("No way brother");
    }
  };
  
  // rules by space
    
  transient MultiMap<S, WordTokRule> rulesForSpace = litmultimap(
    userSpace, ll(rule_iWantYouToBe),
    objectiveSpace, ll(rule_useryWants)
  );
  
  Collection<S> applyRules(L<WordTokRule> rules, S s) {
    ret concatListsToCISet(map(rules, rule ->
      rule.matches(s) ? getStringEmits(rule) : null
    );
  }
  
  Set<S> rewriteInSpace(S space, S input) {
    ret rewriteInSpace(space, ll(input));
  }
  
  Set<S> rewriteInSpace(S space, Collection<S> inputs) {
    Set<S> out = cloneLinkedCISet(inputs);
    L<WordTokRule> rules = rulesForSpace.get(space);
    for unnull (S input : inputs)
      functionReturningList_transitiveHull_ciSet(func(S s) -> Collection<S> { applyRules(rules, s) }, input, seen := out);
    ret out;
  }
  
  S spaceToSpace(S space1, S space2, S input) {
    if (eq(space1, userSpace) && eq(space2, objectiveSpace))
      ret switcheroo_youToName(computerName, switcheroo_iToName(userName, input));
    if (eq(space1, objectiveSpace) && eq(space2, computerSpace))
      ret switcheroo_nameToYou(userName, switcheroo_nameToI(computerName, input));
    ret input;
  }
  
  start-thread {
    S input = "I want you to be my mother";
    
    // first, rewrite in user space
    Set<S> outputs = rewriteInSpace(userSpace, input);
    pnl(outputs);
    
    // translate to objective space
    pnl(map(s -> spaceToSpace(userSpace, objectiveSpace, s), outputs));
 
    // rewrite in objective space
    Set<S> outputs2 = rewriteInSpace(objectiveSpace, outputs);
    pnl(outputs2);
    
    printDone();
  }
}