iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

add API JWT auth.

Browse Source
This commit is contained in:
Miroslav Šedivý
2020-10-31 10:48:24 +01:00
parent 31bd61e2d3
commit 71d39a5c74
5 changed files with 73 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/types/config/server.go
View File

@ -10,6 +10,8 @@ type Server struct {
Key string
Bind string
Static string
UserToken string
AdminToken string
}
func (Server) Init(cmd *cobra.Command) error {
@ -33,6 +35,16 @@ func (Server) Init(cmd *cobra.Command) error {
return err
}
cmd.PersistentFlags().String("user_token", "user_secret", "JWT token for users")
if err := viper.BindPFlag("user_token", cmd.PersistentFlags().Lookup("user_token")); err != nil {
return err
}
cmd.PersistentFlags().String("admin_token", "admin_secret", "JWT token for admins")
if err := viper.BindPFlag("admin_token", cmd.PersistentFlags().Lookup("admin_token")); err != nil {
return err
}
return nil
}
@ -41,4 +53,6 @@ func (s *Server) Set() {
s.Key = viper.GetString("key")
s.Bind = viper.GetString("bind")
s.Static = viper.GetString("static")
s.UserToken = viper.GetString("user_token")
s.AdminToken = viper.GetString("admin_token")
}