feat: make intro be aware of let_fun (#3115)

Adds support for `let_fun` to the `intro` and `intros` tactics. Also
adds support to `intro` for anonymous binder names, since the default
variable name for a `letFun` with an eta reduced body is anonymous.

src/Lean/Meta/Tactic/Intro.lean

@@ -44,24 +44,35 @@ namespace Lean.Meta
       let fvars  := fvars.push fvar
       loop i lctx fvars j s body
     | i+1, type =>
-      let type := type.instantiateRevRange j fvars.size fvars
-      withReader (fun ctx => { ctx with lctx := lctx }) do
-        withNewLocalInstances fvars j do
-          /- We used to use just `whnf`, but it produces counterintuitive behavior if
-             - `type` is a metavariable `?m` such that `?m := let x := v; b`, or
-             - `type` has `MData` or annotations such as `optParam` around a `let`-expression.
-
-             `whnf` instantiates metavariables, and consumes `MData`, but it also expands the `let`.
-          -/
-          let newType := (← instantiateMVars type).cleanupAnnotations
-          if newType.isForall || newType.isLet then
-            loop (i+1) lctx fvars fvars.size s newType
-          else
-            let newType ← whnf newType
-            if newType.isForall then
+      if let some (n, type, val, body) := type.letFun? then
+        let type   := type.instantiateRevRange j fvars.size fvars
+        let type   := type.headBeta
+        let val    := val.instantiateRevRange j fvars.size fvars
+        let fvarId ← mkFreshFVarId
+        let (n, s) ← mkName lctx n true s
+        let lctx   := lctx.mkLetDecl fvarId n type val
+        let fvar   := mkFVar fvarId
+        let fvars  := fvars.push fvar
+        loop i lctx fvars j s body
+      else
+        let type := type.instantiateRevRange j fvars.size fvars
+        withReader (fun ctx => { ctx with lctx := lctx }) do
+          withNewLocalInstances fvars j do
+            /- We used to use just `whnf`, but it produces counterintuitive behavior if
+              - `type` is a metavariable `?m` such that `?m := let x := v; b`, or
+              - `type` has `MData` or annotations such as `optParam` around a `let`-expression.
+
+              `whnf` instantiates metavariables, and consumes `MData`, but it also expands the `let`.
+            -/
+            let newType := (← instantiateMVars type).cleanupAnnotations
+            if newType.isForall || newType.isLet || newType.isLetFun then
               loop (i+1) lctx fvars fvars.size s newType
             else
-              throwTacticEx `introN mvarId "insufficient number of binders"
+              let newType ← whnf newType
+              if newType.isForall then
+                loop (i+1) lctx fvars fvars.size s newType
+              else
+                throwTacticEx `introN mvarId "insufficient number of binders"
   let (fvars, mvarId) ← loop n lctx #[] 0 s mvarType
   return (fvars.map Expr.fvarId!, mvarId)
 
@@ -97,6 +108,9 @@ private def mkAuxNameImp (preserveBinderNames : Bool) (hygienic : Bool) (useName
       mkAuxNameWithoutGivenName rest
 where
   mkAuxNameWithoutGivenName (rest : List Name) : MetaM (Name × List Name) := do
+    -- Use a nicer binder name than `[anonymous]`, which can appear in for example `letFun x f` when `f` is not a lambda expression.
+    -- In this case, we make sure the name is hygienic.
+    let binderName ← if binderName.isAnonymous then mkFreshUserName `a else pure binderName
     if preserveBinderNames then
       return (binderName, rest)
     else
@@ -169,11 +183,15 @@ abbrev _root_.Lean.MVarId.intro1P (mvarId : MVarId) : MetaM (FVarId × MVarId) :
 abbrev intro1P (mvarId : MVarId) : MetaM (FVarId × MVarId) :=
   mvarId.intro1P
 
-private def getIntrosSize : Expr → Nat
+private partial def getIntrosSize : Expr → Nat
   | .forallE _ _ b _ => getIntrosSize b + 1
   | .letE _ _ _ b _  => getIntrosSize b + 1
   | .mdata _ b       => getIntrosSize b
-  | _                => 0
+  | e                =>
+    if let some (_, _, _, b) := e.letFun? then
+      getIntrosSize b + 1
+    else
+      0
 
 /--
 Introduce as many binders as possible without unfolding definitions.

tests/lean/run/introLetFun.lean

@@ -0,0 +1,26 @@
+import Lean.Elab.Tactic.Basic
+import Lean.Meta.Tactic.Intro
+/-!
+# Testing `intro` with `letFun`
+-/
+
+/-!
+Explicit `intro`.
+-/
+example : have x := 2; ∀ _ : Nat, x = x := by
+  intro x _
+  rfl
+
+/-!
+`intros` is aware of `letFun`.
+-/
+example : have x := 2; ∀ _ : Nat, x = x := by
+  intros
+  rfl
+
+/-!
+Check that it works for `letFun` with an eta reduced argument.
+-/
+example (p : Nat → Prop) (h : ∀ x, p x) : letFun 2 p := by
+  intro
+  apply h