From b2eda9d30806191e3722b68b180b2f0ee2c62232 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E4=BC=9F=E4=B9=90?= <mogfee@163.com>
Date: Thu, 30 Mar 2023 08:57:55 +0800
Subject: [PATCH] x

---
 cmd/kit/main.go      | 1 +
 protogen/protogen.go | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/cmd/kit/main.go b/cmd/kit/main.go
index 6460df3..e0f5e7e 100644
--- a/cmd/kit/main.go
+++ b/cmd/kit/main.go
@@ -36,6 +36,7 @@ func (u *Kit) Generate(plugin *protogen.Plugin) error {
 	}
 	u.addImports("context")
 	u.addImports("git.diulo.com/mogfee/kit/transport/http")
+	u.addImports("git.diulo.com/mogfee/kit/middleware/jwt")
 	for _, f := range plugin.Files {
 		if len(f.Services) == 0 {
 			continue
diff --git a/protogen/protogen.go b/protogen/protogen.go
index 182a84f..a164611 100644
--- a/protogen/protogen.go
+++ b/protogen/protogen.go
@@ -40,6 +40,9 @@ func GetAuthKey(m *protogen.Method) (authKey string, needAuth bool) {
 			//log.Println(err)
 		} else {
 			if vv, ok := opts.(*string); ok {
+				if *vv == "none" {
+					return "", false
+				}
 				return *vv, true
 			}
 		}