From a0e906b5e15893c210cee241559d5f90e214a4f1 Mon Sep 17 00:00:00 2001
From: JaguarJack <njphper@gmail.com>
Date: Thu, 10 Jun 2021 19:52:57 +0800
Subject: [PATCH] update route generate

---
 extend/catcher/generate/factory/Route.php | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/extend/catcher/generate/factory/Route.php b/extend/catcher/generate/factory/Route.php
index 0bf2fb2..447a5ab 100644
--- a/extend/catcher/generate/factory/Route.php
+++ b/extend/catcher/generate/factory/Route.php
@@ -38,7 +38,7 @@ class Route extends Factory
         $content = $this->generateRoute($router);
 
         $content = '<?php' . PHP_EOL .
-            trim(str_replace(['$scapegoat', '='], ['', ''], $content), ';') . ';';
+            trim(str_replace(['$scapegoat ='], [''], $content), ';') . ';';
 
         if (! file_exists($router)) {
             return FileSystem::put($router, $content);
@@ -112,11 +112,13 @@ class Route extends Factory
 
             $expression = $ast[0];
 
-            $stmts = $expression->expr->var->args[1]->value->stmts;
+            $argKey = count($expression->expr->var->args) == 1 ? 0 : 1;
+
+            $stmts = $expression->expr->var->args[$argKey]->value->stmts;
 
             $stmts = array_merge($stmts, $this->parseRouteMethods());
 
-            $expression->expr->var->args[1]->value->stmts = $stmts;
+            $expression->expr->var->args[$argKey]->value->stmts = $stmts;
 
             $ast[0] = $expression;