!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") {
    // { debug = true; }
    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");
      else if (debug) print("Can't decide on: " + x_simplified);
    }
  };
  
  // 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;
  }
  
  LS spaceToSpace(S space1, S space2, Collection<S> inputs) {
    ret map(s -> spaceToSpace(space1, space2, s), inputs);
  }
  
  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 {
    for (S input : ll("I want you to be my mother", "I want you to be my assistant")) {
      print_nlBefore(input);
      // first, rewrite in user space
      Set<S> outputs = rewriteInSpace(userSpace, input);
      pnlIndent("User space > ", outputs);
      
      // translate to objective space
      LS outputs2 = spaceToSpace(userSpace, objectiveSpace, outputs);
      pnlIndent("Objective space > ", outputs2);
   
      // rewrite in objective space
      Set<S> outputs3 = rewriteInSpace(objectiveSpace, outputs2);
      pnlIndent("Objective space > ", outputs3);
      
      // translate to computer space
      LS outputs4 = spaceToSpace(objectiveSpace, computerSpace, outputs3);
      pnlIndent("Bot space > ", outputs4);
    }
    
    printDone();
  }
}