diff --git a/spec/functional_test/fixtures/crystal_lucky/src/actions/api/sign_ups/create.cr b/spec/functional_test/fixtures/crystal_lucky/src/actions/api/sign_ups/create.cr index 15bbd04b..0d822f9c 100644 --- a/spec/functional_test/fixtures/crystal_lucky/src/actions/api/sign_ups/create.cr +++ b/spec/functional_test/fixtures/crystal_lucky/src/actions/api/sign_ups/create.cr @@ -3,6 +3,9 @@ class Api::SignUps::Create < ApiAction post "/api/sign_ups" do user = SignUpUser.create!(params) + cookies.get("name1") + cookies["name2"] + json({token: UserToken.generate(user)}) end end diff --git a/spec/functional_test/testers/crystal_lucky_spec.cr b/spec/functional_test/testers/crystal_lucky_spec.cr index 00e8f462..c84d2b2f 100644 --- a/spec/functional_test/testers/crystal_lucky_spec.cr +++ b/spec/functional_test/testers/crystal_lucky_spec.cr @@ -10,7 +10,10 @@ extected_endpoints = [ Param.new("X-Forwarded-For", "", "header"), ]), Endpoint.new("/api/sign_ins", "POST", [Param.new("users", "", "json")]), - Endpoint.new("/api/sign_ups", "POST"), + Endpoint.new("/api/sign_ups", "POST", [ + Param.new("name1", "", "cookie"), + Param.new("name2", "", "cookie"), + ]), ] FunctionalTester.new("fixtures/crystal_lucky/", { diff --git a/src/analyzer/analyzers/analyzer_crystal_lucky.cr b/src/analyzer/analyzers/analyzer_crystal_lucky.cr index 25d0e69d..d391e18e 100644 --- a/src/analyzer/analyzers/analyzer_crystal_lucky.cr +++ b/src/analyzer/analyzers/analyzer_crystal_lucky.cr @@ -71,6 +71,16 @@ class AnalyzerCrystalLucky < Analyzer return Param.new(param, "", "header") end + if content.includes? "cookies.get(" + param = content.split("cookies.get(")[1].split(")")[0].gsub("\"", "").gsub("'", "") + return Param.new(param, "", "cookie") + end + + if content.includes? "cookies[" + param = content.split("cookies[")[1].split("]")[0].gsub("\"", "").gsub("'", "") + return Param.new(param, "", "cookie") + end + Param.new("", "", "") end