4 files changed, 20 insertions, 3 deletions
diff --git a/script-beta/core/hover/arg.lua b/script-beta/core/hover/arg.lua
index 5dcf94b6..ef5eef57 100644
--- a/script-beta/core/hover/arg.lua
+++ b/script-beta/core/hover/arg.lua
@@ -90,7 +90,11 @@ local function asDocFunction(source)
     for i = 1, #source.args do
         local arg = source.args[i]
         local name = arg.name[1]
-        args[i] = ('%s: %s'):format(name, vm.getInferType(arg.extends))
+        args[i] = ('%s: %s%s'):format(
+            name,
+            vm.getInferType(arg.extends),
+            arg.optional and '?' or ''
+        )
     end
     return table.concat(args, ', ')
 end
diff --git a/script-beta/core/hover/name.lua b/script-beta/core/hover/name.lua
index 4d5b015c..9f19d8f3 100644
--- a/script-beta/core/hover/name.lua
+++ b/script-beta/core/hover/name.lua
@@ -71,6 +71,7 @@ end
 
 local function asDocFunction(source)
     local doc = guide.getParentType(source, 'doc.type')
+            or  guide.getParentType(source, 'doc.overload')
     if not doc or not doc.bindSources then
         return ''
     end
diff --git a/script-beta/core/hover/return.lua b/script-beta/core/hover/return.lua
index 4159e039..a0b4363d 100644
--- a/script-beta/core/hover/return.lua
+++ b/script-beta/core/hover/return.lua
@@ -126,7 +126,10 @@ local function asDocFunction(source)
     end
     local returns = {}
     for i, rtn in ipairs(source.returns) do
-        local rtnText = vm.getInferType(rtn)
+        local rtnText = ('%s%s'):format(
+            vm.getInferType(rtn),
+            rtn.optional and '?' or ''
+        )
         if i == 1 then
             returns[#returns+1] = ('  -> %s'):format(rtnText)
         else
diff --git a/script-beta/parser/luadoc.lua b/script-beta/parser/luadoc.lua
index ad2dea14..ca4ee25d 100644
--- a/script-beta/parser/luadoc.lua
+++ b/script-beta/parser/luadoc.lua
@@ -52,6 +52,7 @@ Symbol              <-  ({} {
                         /   '>'
                         /   '('
                         /   ')'
+                        /   '?'
                         } {})
                     ->  Symbol
 ]], {
@@ -291,6 +292,10 @@ local function  parseTypeUnitFunction()
         if not arg.extends then
             break
         end
+        if checkToken('symbol', '?', 1) then
+            nextToken()
+            arg.optional = true
+        end
         arg.finish = getFinish()
         typeUnit.args[#typeUnit.args+1] = arg
         if checkToken('symbol', ',', 1) then
@@ -303,10 +308,14 @@ local function  parseTypeUnitFunction()
     if checkToken('symbol', ':', 1) then
         nextToken()
         while true do
-            local rtn = parseType(arg)
+            local rtn = parseType(typeUnit)
             if not rtn then
                 break
             end
+            if checkToken('symbol', '?', 1) then
+                nextToken()
+                rtn.optional = true
+            end
             typeUnit.returns[#typeUnit.returns+1] = rtn
             if checkToken('symbol', ',', 1) then
                 nextToken()