Merge pull request #10 from eNeRGy164/feature/return-statements

Add support for return statements.

src/LivingDocumentation.Analyzer/Analyzers/InvocationsAnalyzer.cs

@@ -96,11 +96,8 @@ public override void VisitInvocationExpression(InvocationExpressionSyntax node)
                     methodName = i.Identifier.ValueText;
                     break;
             }
-            if (methodName == "Fire")
-            {
 
-            }
-                var invocation = new InvocationDescription(containingType, methodName);
+            var invocation = new InvocationDescription(containingType, methodName);
             this.statements.Add(invocation);
 
             foreach (var argument in node.ArgumentList.Arguments)
@@ -111,5 +108,21 @@ public override void VisitInvocationExpression(InvocationExpressionSyntax node)
 
             base.VisitInvocationExpression(node);
         }
+
+        public override void VisitReturnStatement(ReturnStatementSyntax node)
+        {
+            var returnDescription = new ReturnDescription(node.Expression?.ToString() ?? string.Empty);
+            this.statements.Add(returnDescription);
+
+            base.VisitReturnStatement(node);
+        }
+
+        public override void VisitArrowExpressionClause(ArrowExpressionClauseSyntax node)
+        {
+            var returnDescription = new ReturnDescription(node.Expression.ToString());
+            this.statements.Add(returnDescription);
+
+            base.VisitArrowExpressionClause(node);
+        }
     }
 }

src/LivingDocumentation.Descriptions/ReturnDescription.cs

@@ -0,0 +1,18 @@
+using System.Diagnostics;
+
+namespace LivingDocumentation
+{
+    [DebuggerDisplay("Return {Expression}")]
+    public class ReturnDescription : Statement
+    {
+        public string Expression
+        {
+            get;
+        }
+
+        public ReturnDescription(string expression)
+        {
+            this.Expression = expression;
+        }
+    }
+}