!7

p-experiment {
  S snippetID = #1015725;
  S orig = loadSnippet(snippetID);
  L<S> tok = javaTok(orig);
  Map<Int> bracketMap = getBracketMap(tok);
  L<Int> l = jfindAll(tok, "? <id> =");
  for (int i : l) {
    S type = or2(tok_inferTypeAt(tok, bracketMap, i+6), "?");
    //print(tok.get(i+2) + " => " + type);
    if (nempty(type)) tok.set(i, type);
  }
  S text = join(tok);
  print();
  printIndent(text);
  if (neq(text, orig))
    editSnippet(snippetID, text);
}

sS tok_inferTypeAt(L<S> tok, Map<Int> bracketMap, int i) {
  int j = scanToEndOfInitializer(tok, bracketMap, i);
  L<S> tok2 = subList(tok, i-1, j+1);
  //printStruct(tok2);
  fS fname = pairA(tok_parseFunctionCall(tok2));
  if (fname == null) null;
  //print("fname: " + fname);
  S snippetID = stdFunctions_cached().get(fname);
  if (snippetID == null) null;
  
  LL<S> defs = findFullFunctionDefs(javaTokSnippet(snippetID), true);
  //print("Have " + l(defs) + " defs");
  defs = filter(func(L<S> def) -> bool { eq(tok_functionName(def), fname) }, defs);
  //print("Have " + l(defs) + " defs");
  new HashSet<S> types;
  for (L<S> def : defs) {
    int k = tok_skipModifiers(def, 1);
    // TODO: skip type args
    S type = joinSubList(def, k, indexOf(def, k, fname)-1);
    types.add(type);
  }
  ret singletonOpt(types);
}