From 53e59ddf8901dbb8a6d903e69301b3ba66e4f5e3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=80=9CxHuPo=E2=80=9D?=
<7513325+vrocwang@users.noreply.github.com>
Date: Wed, 21 Aug 2024 16:35:46 +0800
Subject: [PATCH] first commit
---
cmd/root.go | 83 +++++++++++++++++++++++++
config.yaml | 8 +++
database/database.go | 49 +++++++++++++++
database/init/otp.sql | 6 ++
database/init/users.sql | 5 ++
go.mod | 49 +++++++++++++++
go.sum | 134 ++++++++++++++++++++++++++++++++++++++++
handlers/handler.go | 9 +++
handlers/login.go | 93 ++++++++++++++++++++++++++++
handlers/otp.go | 61 ++++++++++++++++++
main.go | 7 +++
utils/utils.go | 54 ++++++++++++++++
12 files changed, 558 insertions(+)
create mode 100644 cmd/root.go
create mode 100644 config.yaml
create mode 100644 database/database.go
create mode 100644 database/init/otp.sql
create mode 100644 database/init/users.sql
create mode 100644 go.mod
create mode 100644 go.sum
create mode 100644 handlers/handler.go
create mode 100644 handlers/login.go
create mode 100644 handlers/otp.go
create mode 100644 main.go
create mode 100644 utils/utils.go
diff --git a/cmd/root.go b/cmd/root.go
new file mode 100644
index 0000000..3752544
--- /dev/null
+++ b/cmd/root.go
@@ -0,0 +1,83 @@
+package cmd
+
+import (
+ "fmt"
+ "log"
+ "net/http"
+ "os"
+ "otpm/database"
+ "otpm/handlers"
+ "otpm/utils"
+
+ "github.com/jmoiron/sqlx"
+ "github.com/julienschmidt/httprouter"
+ "github.com/spf13/cobra"
+ "github.com/spf13/viper"
+)
+
+var rootCmd = &cobra.Command{
+ Use: "otpm",
+ Short: "otp backend for microapp on wechat",
+ Run: func(cmd *cobra.Command, args []string) {
+ startApp()
+ },
+}
+
+func Execute() {
+ if err := rootCmd.Execute(); err != nil {
+ fmt.Println(err)
+ os.Exit(1)
+ }
+}
+
+func init() {
+ cobra.OnInitialize(initConfig)
+
+ rootCmd.PersistentFlags().StringP("config", "c", "", "config file (default is $HOME/config.yaml)")
+ rootCmd.PersistentFlags().StringP("driver", "d", "sqlite3", "database driver (sqlite3, postgres, mysql)")
+ rootCmd.PersistentFlags().StringP("dsn", "s", "", "database connection string")
+ rootCmd.PersistentFlags().StringP("port", "p", "8080", "port to listen on")
+
+ viper.BindPFlag("database.driver", rootCmd.PersistentFlags().Lookup("driver"))
+ viper.BindPFlag("database.dsn", rootCmd.PersistentFlags().Lookup("dsn"))
+ viper.BindPFlag("port", rootCmd.PersistentFlags().Lookup("port"))
+}
+
+func initConfig() {
+ if cfgFile := viper.GetString("config"); cfgFile != "" {
+ viper.SetConfigFile(cfgFile)
+ } else {
+ viper.AddConfigPath(".")
+ viper.SetConfigName("config")
+ viper.SetConfigType("yaml")
+ }
+
+ if err := viper.ReadInConfig(); err != nil {
+ log.Fatalf("Error reading config file: %v", err)
+ }
+}
+
+func initApp(db *sqlx.DB) {
+ if err := database.MigrateDB(db); err != nil {
+ log.Fatalf("Error migrating the database: %v", err)
+ }
+}
+
+func startApp() {
+ port := viper.GetInt("port")
+ db, err := database.InitDB()
+ if err != nil {
+ log.Fatalf("Error connecting to the database: %v", err)
+ }
+ defer db.Close()
+ initApp(db)
+ handler := &handlers.Handler{DB: db}
+
+ router := httprouter.New()
+ router.POST("/login", utils.AdaptHandler(handler.Login))
+ router.POST("/set", utils.AdaptHandler(handler.UpdateOrCreateOtp))
+ router.GET("/get", utils.AdaptHandler(handler.GetOtp))
+
+ log.Println("Starting server on :8080")
+ log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", port), router))
+}
diff --git a/config.yaml b/config.yaml
new file mode 100644
index 0000000..f23f8fb
--- /dev/null
+++ b/config.yaml
@@ -0,0 +1,8 @@
+database:
+ driver: sqlite
+ dsn: otpm.sqlite
+port: 8080
+
+wechat:
+ appid: "wx57d1033974eb5250"
+ secret: "be494c2a81df685a40b9a74e1736b15d"
diff --git a/database/database.go b/database/database.go
new file mode 100644
index 0000000..bf4ade8
--- /dev/null
+++ b/database/database.go
@@ -0,0 +1,49 @@
+package database
+
+import (
+ _ "embed"
+ "log"
+
+ _ "github.com/go-sql-driver/mysql"
+ "github.com/jmoiron/sqlx"
+ _ "github.com/lib/pq"
+ "github.com/spf13/viper"
+ _ "modernc.org/sqlite"
+)
+
+var (
+ //go:embed init/users.sql
+ userTable string
+ //go:embed init/otp.sql
+ otpTable string
+)
+
+func InitDB() (*sqlx.DB, error) {
+ driver := viper.GetString("database.driver")
+ dsn := viper.GetString("database.dsn")
+
+ db, err := sqlx.Open(driver, dsn)
+ if err != nil {
+ return nil, err
+ }
+
+ if err := db.Ping(); err != nil {
+ return nil, err
+ }
+
+ log.Println("Connected to database!")
+ return db, nil
+}
+
+func MigrateDB(db *sqlx.DB) error {
+ _, err := db.Exec(userTable)
+ if err != nil {
+ return err
+ }
+
+ _, err = db.Exec(otpTable)
+ if err != nil {
+ return err
+ }
+ return nil
+}
diff --git a/database/init/otp.sql b/database/init/otp.sql
new file mode 100644
index 0000000..a9b2442
--- /dev/null
+++ b/database/init/otp.sql
@@ -0,0 +1,6 @@
+CREATE TABLE IF NOT EXISTS otp (
+ id SERIAL PRIMARY KEY,
+ openid VARCHAR(255),
+ num INTEGER,
+ token VARCHAR(255)
+);
\ No newline at end of file
diff --git a/database/init/users.sql b/database/init/users.sql
new file mode 100644
index 0000000..4dbedae
--- /dev/null
+++ b/database/init/users.sql
@@ -0,0 +1,5 @@
+CREATE TABLE IF NOT EXISTS users (
+ id SERIAL PRIMARY KEY,
+ openid VARCHAR(255) UNIQUE NOT NULL,
+ session_key VARCHAR(255) UNIQUE NOT NULL
+);
\ No newline at end of file
diff --git a/go.mod b/go.mod
new file mode 100644
index 0000000..57bb068
--- /dev/null
+++ b/go.mod
@@ -0,0 +1,49 @@
+module otpm
+
+go 1.21.1
+
+require (
+ github.com/go-sql-driver/mysql v1.8.1
+ github.com/jmoiron/sqlx v1.4.0
+ github.com/julienschmidt/httprouter v1.3.0
+ github.com/lib/pq v1.10.9
+ github.com/spf13/cobra v1.8.1
+ github.com/spf13/viper v1.19.0
+ modernc.org/sqlite v1.32.0
+)
+
+require (
+ filippo.io/edwards25519 v1.1.0 // indirect
+ github.com/dustin/go-humanize v1.0.1 // indirect
+ github.com/fsnotify/fsnotify v1.7.0 // indirect
+ github.com/google/uuid v1.6.0 // indirect
+ github.com/hashicorp/golang-lru/v2 v2.0.7 // indirect
+ github.com/hashicorp/hcl v1.0.0 // indirect
+ github.com/inconshreveable/mousetrap v1.1.0 // indirect
+ github.com/magiconair/properties v1.8.7 // indirect
+ github.com/mattn/go-isatty v0.0.20 // indirect
+ github.com/mitchellh/mapstructure v1.5.0 // indirect
+ github.com/ncruces/go-strftime v0.1.9 // indirect
+ github.com/pelletier/go-toml/v2 v2.2.2 // indirect
+ github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
+ github.com/sagikazarmark/locafero v0.4.0 // indirect
+ github.com/sagikazarmark/slog-shim v0.1.0 // indirect
+ github.com/sourcegraph/conc v0.3.0 // indirect
+ github.com/spf13/afero v1.11.0 // indirect
+ github.com/spf13/cast v1.6.0 // indirect
+ github.com/spf13/pflag v1.0.5 // indirect
+ github.com/subosito/gotenv v1.6.0 // indirect
+ go.uber.org/atomic v1.9.0 // indirect
+ go.uber.org/multierr v1.9.0 // indirect
+ golang.org/x/exp v0.0.0-20231108232855-2478ac86f678 // indirect
+ golang.org/x/sys v0.22.0 // indirect
+ golang.org/x/text v0.14.0 // indirect
+ gopkg.in/ini.v1 v1.67.0 // indirect
+ gopkg.in/yaml.v3 v3.0.1 // indirect
+ modernc.org/gc/v3 v3.0.0-20240107210532-573471604cb6 // indirect
+ modernc.org/libc v1.55.3 // indirect
+ modernc.org/mathutil v1.6.0 // indirect
+ modernc.org/memory v1.8.0 // indirect
+ modernc.org/strutil v1.2.0 // indirect
+ modernc.org/token v1.1.0 // indirect
+)
diff --git a/go.sum b/go.sum
new file mode 100644
index 0000000..5dc30bc
--- /dev/null
+++ b/go.sum
@@ -0,0 +1,134 @@
+filippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA=
+filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4=
+github.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
+github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
+github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
+github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
+github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
+github.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nosvA=
+github.com/fsnotify/fsnotify v1.7.0/go.mod h1:40Bi/Hjc2AVfZrqy+aj+yEI+/bRxZnMJyTJwOpGvigM=
+github.com/go-sql-driver/mysql v1.8.1 h1:LedoTUt/eveggdHS9qUFC1EFSa8bU2+1pZjSRpvNJ1Y=
+github.com/go-sql-driver/mysql v1.8.1/go.mod h1:wEBSXgmK//2ZFJyE+qWnIsVGmvmEKlqwuVSjsCm7DZg=
+github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
+github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
+github.com/google/pprof v0.0.0-20240409012703-83162a5b38cd h1:gbpYu9NMq8jhDVbvlGkMFWCjLFlqqEZjEmObmhUy6Vo=
+github.com/google/pprof v0.0.0-20240409012703-83162a5b38cd/go.mod h1:kf6iHlnVGwgKolg33glAes7Yg/8iWP8ukqeldJSO7jw=
+github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
+github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/hashicorp/golang-lru/v2 v2.0.7 h1:a+bsQ5rvGLjzHuww6tVxozPZFVghXaHOwFs4luLUK2k=
+github.com/hashicorp/golang-lru/v2 v2.0.7/go.mod h1:QeFd9opnmA6QUJc5vARoKUSoFhyfM2/ZepoAG6RGpeM=
+github.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4=
+github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=
+github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
+github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
+github.com/jmoiron/sqlx v1.4.0 h1:1PLqN7S1UYp5t4SrVVnt4nUVNemrDAtxlulVe+Qgm3o=
+github.com/jmoiron/sqlx v1.4.0/go.mod h1:ZrZ7UsYB/weZdl2Bxg6jCRO9c3YHl8r3ahlKmRT4JLY=
+github.com/julienschmidt/httprouter v1.3.0 h1:U0609e9tgbseu3rBINet9P48AI/D3oJs4dN7jwJOQ1U=
+github.com/julienschmidt/httprouter v1.3.0/go.mod h1:JR6WtHb+2LUe8TCKY3cZOxFyyO8IZAc4RVcycCCAKdM=
+github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
+github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
+github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
+github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
+github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw=
+github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
+github.com/magiconair/properties v1.8.7 h1:IeQXZAiQcpL9mgcAe1Nu6cX9LLw6ExEHKjN0VQdvPDY=
+github.com/magiconair/properties v1.8.7/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0=
+github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
+github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/mattn/go-sqlite3 v1.14.22 h1:2gZY6PC6kBnID23Tichd1K+Z0oS6nE/XwU+Vz/5o4kU=
+github.com/mattn/go-sqlite3 v1.14.22/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
+github.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyuac5Z2hdY=
+github.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo=
+github.com/ncruces/go-strftime v0.1.9 h1:bY0MQC28UADQmHmaF5dgpLmImcShSi2kHU9XLdhx/f4=
+github.com/ncruces/go-strftime v0.1.9/go.mod h1:Fwc5htZGVVkseilnfgOVb9mKy6w1naJmn9CehxcKcls=
+github.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM=
+github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=
+github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
+github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
+github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
+github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
+github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/sagikazarmark/locafero v0.4.0 h1:HApY1R9zGo4DBgr7dqsTH/JJxLTTsOt7u6keLGt6kNQ=
+github.com/sagikazarmark/locafero v0.4.0/go.mod h1:Pe1W6UlPYUk/+wc/6KFhbORCfqzgYEpgQ3O5fPuL3H4=
+github.com/sagikazarmark/slog-shim v0.1.0 h1:diDBnUNK9N/354PgrxMywXnAwEr1QZcOr6gto+ugjYE=
+github.com/sagikazarmark/slog-shim v0.1.0/go.mod h1:SrcSrq8aKtyuqEI1uvTDTK1arOWRIczQRv+GVI1AkeQ=
+github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=
+github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=
+github.com/spf13/afero v1.11.0 h1:WJQKhtpdm3v2IzqG8VMqrr6Rf3UYpEF239Jy9wNepM8=
+github.com/spf13/afero v1.11.0/go.mod h1:GH9Y3pIexgf1MTIWtNGyogA5MwRIDXGUr+hbWNoBjkY=
+github.com/spf13/cast v1.6.0 h1:GEiTHELF+vaR5dhz3VqZfFSzZjYbgeKDpBxQVS4GYJ0=
+github.com/spf13/cast v1.6.0/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo=
+github.com/spf13/cobra v1.8.1 h1:e5/vxKd/rZsfSJMUX1agtjeTDf+qv1/JdBF8gg5k9ZM=
+github.com/spf13/cobra v1.8.1/go.mod h1:wHxEcudfqmLYa8iTfL+OuZPbBZkmvliBWKIezN3kD9Y=
+github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
+github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+github.com/spf13/viper v1.19.0 h1:RWq5SEjt8o25SROyN3z2OrDB9l7RPd3lwTWU8EcEdcI=
+github.com/spf13/viper v1.19.0/go.mod h1:GQUN9bilAbhU/jgc1bKs99f/suXKeUMct8Adx5+Ntkg=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
+github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
+github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
+github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
+github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
+github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8=
+github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU=
+go.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE=
+go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
+go.uber.org/multierr v1.9.0 h1:7fIwc/ZtS0q++VgcfqFDxSBZVv/Xo49/SYnDFupUwlI=
+go.uber.org/multierr v1.9.0/go.mod h1:X2jQV1h+kxSjClGpnseKVIxpmcjrj7MNnI0bnlfKTVQ=
+golang.org/x/exp v0.0.0-20231108232855-2478ac86f678 h1:mchzmB1XO2pMaKFRqk/+MV3mgGG96aqaPXaMifQU47w=
+golang.org/x/exp v0.0.0-20231108232855-2478ac86f678/go.mod h1:zk2irFbV9DP96SEBUUAy67IdHUaZuSnrz1n472HUCLE=
+golang.org/x/mod v0.16.0 h1:QX4fJ0Rr5cPQCF7O9lh9Se4pmwfwskqZfq5moyldzic=
+golang.org/x/mod v0.16.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.22.0 h1:RI27ohtqKCnwULzJLqkv897zojh5/DwS/ENaMzUOaWI=
+golang.org/x/sys v0.22.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
+golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
+golang.org/x/tools v0.19.0 h1:tfGCXNR1OsFG+sVdLAitlpjAvD/I6dHDKnYrpEZUHkw=
+golang.org/x/tools v0.19.0/go.mod h1:qoJWxmGSIBmAeriMx19ogtrEPrGtDbPK634QFIcLAhc=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo=
+gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/ini.v1 v1.67.0 h1:Dgnx+6+nfE+IfzjUEISNeydPJh9AXNNsWbGP9KzCsOA=
+gopkg.in/ini.v1 v1.67.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+modernc.org/cc/v4 v4.21.4 h1:3Be/Rdo1fpr8GrQ7IVw9OHtplU4gWbb+wNgeoBMmGLQ=
+modernc.org/cc/v4 v4.21.4/go.mod h1:HM7VJTZbUCR3rV8EYBi9wxnJ0ZBRiGE5OeGXNA0IsLQ=
+modernc.org/ccgo/v4 v4.19.2 h1:lwQZgvboKD0jBwdaeVCTouxhxAyN6iawF3STraAal8Y=
+modernc.org/ccgo/v4 v4.19.2/go.mod h1:ysS3mxiMV38XGRTTcgo0DQTeTmAO4oCmJl1nX9VFI3s=
+modernc.org/fileutil v1.3.0 h1:gQ5SIzK3H9kdfai/5x41oQiKValumqNTDXMvKo62HvE=
+modernc.org/fileutil v1.3.0/go.mod h1:XatxS8fZi3pS8/hKG2GH/ArUogfxjpEKs3Ku3aK4JyQ=
+modernc.org/gc/v2 v2.4.1 h1:9cNzOqPyMJBvrUipmynX0ZohMhcxPtMccYgGOJdOiBw=
+modernc.org/gc/v2 v2.4.1/go.mod h1:wzN5dK1AzVGoH6XOzc3YZ+ey/jPgYHLuVckd62P0GYU=
+modernc.org/gc/v3 v3.0.0-20240107210532-573471604cb6 h1:5D53IMaUuA5InSeMu9eJtlQXS2NxAhyWQvkKEgXZhHI=
+modernc.org/gc/v3 v3.0.0-20240107210532-573471604cb6/go.mod h1:Qz0X07sNOR1jWYCrJMEnbW/X55x206Q7Vt4mz6/wHp4=
+modernc.org/libc v1.55.3 h1:AzcW1mhlPNrRtjS5sS+eW2ISCgSOLLNyFzRh/V3Qj/U=
+modernc.org/libc v1.55.3/go.mod h1:qFXepLhz+JjFThQ4kzwzOjA/y/artDeg+pcYnY+Q83w=
+modernc.org/mathutil v1.6.0 h1:fRe9+AmYlaej+64JsEEhoWuAYBkOtQiMEU7n/XgfYi4=
+modernc.org/mathutil v1.6.0/go.mod h1:Ui5Q9q1TR2gFm0AQRqQUaBWFLAhQpCwNcuhBOSedWPo=
+modernc.org/memory v1.8.0 h1:IqGTL6eFMaDZZhEWwcREgeMXYwmW83LYW8cROZYkg+E=
+modernc.org/memory v1.8.0/go.mod h1:XPZ936zp5OMKGWPqbD3JShgd/ZoQ7899TUuQqxY+peU=
+modernc.org/opt v0.1.3 h1:3XOZf2yznlhC+ibLltsDGzABUGVx8J6pnFMS3E4dcq4=
+modernc.org/opt v0.1.3/go.mod h1:WdSiB5evDcignE70guQKxYUl14mgWtbClRi5wmkkTX0=
+modernc.org/sortutil v1.2.0 h1:jQiD3PfS2REGJNzNCMMaLSp/wdMNieTbKX920Cqdgqc=
+modernc.org/sortutil v1.2.0/go.mod h1:TKU2s7kJMf1AE84OoiGppNHJwvB753OYfNl2WRb++Ss=
+modernc.org/sqlite v1.32.0 h1:6BM4uGza7bWypsw4fdLRsLxut6bHe4c58VeqjRgST8s=
+modernc.org/sqlite v1.32.0/go.mod h1:UqoylwmTb9F+IqXERT8bW9zzOWN8qwAIcLdzeBZs4hA=
+modernc.org/strutil v1.2.0 h1:agBi9dp1I+eOnxXeiZawM8F4LawKv4NzGWSaLfyeNZA=
+modernc.org/strutil v1.2.0/go.mod h1:/mdcBmfOibveCTBxUl5B5l6W+TTH1FXPLHZE6bTosX0=
+modernc.org/token v1.1.0 h1:Xl7Ap9dKaEs5kLoOQeQmPWevfnk/DM5qcLcYlA8ys6Y=
+modernc.org/token v1.1.0/go.mod h1:UGzOrNV1mAFSEB63lOFHIpNRUVMvYTc6yu1SMY/XTDM=
diff --git a/handlers/handler.go b/handlers/handler.go
new file mode 100644
index 0000000..db15938
--- /dev/null
+++ b/handlers/handler.go
@@ -0,0 +1,9 @@
+package handlers
+
+import (
+ "github.com/jmoiron/sqlx"
+)
+
+type Handler struct {
+ DB *sqlx.DB
+}
diff --git a/handlers/login.go b/handlers/login.go
new file mode 100644
index 0000000..30812d7
--- /dev/null
+++ b/handlers/login.go
@@ -0,0 +1,93 @@
+package handlers
+
+import (
+ "encoding/json"
+ "fmt"
+ "io"
+ "net/http"
+
+ "github.com/spf13/viper"
+)
+
+var code2Session = "https://api.weixin.qq.com/sns/jscode2session?appid=%s&secret=%s&js_code=%s&grant_type=authorization_code"
+
+type LoginRequest struct {
+ Code string `json:"code"`
+}
+
+// 封装code2session接口返回数据
+type LoginResponse struct {
+ OpenId string `json:"openid"`
+ SessionKey string `json:"session_key"`
+ UnionId string `json:"unionid"`
+ ErrCode int `json:"errcode"`
+ ErrMsg string `json:"errmsg"`
+}
+
+func getLoginResponse(code string) (*LoginResponse, error) {
+ appid := viper.GetString("wechat.appid")
+ secret := viper.GetString("wechat.secret")
+ url := fmt.Sprintf(code2Session, appid, secret, code)
+ resp, err := http.Get(url)
+ if err != nil {
+ return nil, err
+ }
+ defer resp.Body.Close()
+
+ var loginResponse LoginResponse
+ if err := json.NewDecoder(resp.Body).Decode(&loginResponse); err != nil {
+ return nil, err
+ }
+
+ if loginResponse.ErrCode != 0 {
+ return nil, fmt.Errorf("code2session error: %s", loginResponse.ErrMsg)
+ }
+ return &loginResponse, nil
+}
+
+func (h *Handler) Login(w http.ResponseWriter, r *http.Request) {
+
+ var req LoginRequest
+ body, err := io.ReadAll(r.Body)
+ if err != nil {
+ http.Error(w, "Failed to read request body", http.StatusBadRequest)
+ return
+ }
+ if err := json.Unmarshal(body, &req); err != nil {
+ http.Error(w, "Failed to parse request body", http.StatusBadRequest)
+ return
+ }
+
+ loginResponse, err := getLoginResponse(req.Code)
+ if err != nil {
+ http.Error(w, "Failed to get session key", http.StatusInternalServerError)
+ return
+ }
+
+ // // 插入或更新用户的openid和sessionid
+ // query := `
+ // INSERT INTO users (openid, sessionid)
+ // VALUES ($1, $2)
+ // ON CONFLICT (openid) DO UPDATE SET sessionid = $2
+ // RETURNING id;
+ // `
+
+ // var ID int
+ // if err := h.DB.QueryRow(query, loginResponse.OpenId, loginResponse.SessionKey).Scan(&ID); err != nil {
+ // http.Error(w, "Failed to log in user", http.StatusInternalServerError)
+ // return
+ // }
+
+ data := map[string]interface{}{
+ "openid": loginResponse.OpenId,
+ "session_key": loginResponse.SessionKey,
+ }
+ respData, err := json.Marshal(data)
+ if err != nil {
+ http.Error(w, "Failed to marshal response data", http.StatusInternalServerError)
+ return
+ }
+ w.Header().Set("Content-Type", "application/json")
+ w.WriteHeader(http.StatusOK)
+ w.Write([]byte(respData))
+}
diff --git a/handlers/otp.go b/handlers/otp.go
new file mode 100644
index 0000000..61230ec
--- /dev/null
+++ b/handlers/otp.go
@@ -0,0 +1,61 @@
+package handlers
+
+import (
+ "encoding/json"
+ "net/http"
+)
+
+type OtpRequest struct {
+ OpenID string `json:"openid"`
+ Num int `json:"num"`
+ Token *[]OTP `json:"token"`
+}
+type OTP struct {
+ Issuer string `json:"issuer"`
+ Remark string `json:"remark"`
+ Secret string `json:"secret"`
+}
+
+func (h *Handler) UpdateOrCreateOtp(w http.ResponseWriter, r *http.Request) {
+ var req OtpRequest
+ if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+ http.Error(w, "Invalid request payload", http.StatusBadRequest)
+ return
+ }
+
+ num := len(*req.Token)
+
+ // 插入或更新 OTP 记录
+ query := `
+ INSERT INTO otp (openid, num, token)
+ VALUES ($1, $2, $3)
+ `
+
+ _, err := h.DB.Exec(query, req.OpenID, req.Token, num)
+ if err != nil {
+ http.Error(w, "Failed to update or create OTP", http.StatusInternalServerError)
+ return
+ }
+
+ w.WriteHeader(http.StatusOK)
+ w.Write([]byte("OTP updated or created successfully"))
+}
+
+func (h *Handler) GetOtp(w http.ResponseWriter, r *http.Request) {
+ openid := r.URL.Query().Get("openid")
+ if openid == "" {
+ http.Error(w, "未登录", http.StatusBadRequest)
+ return
+ }
+
+ var otp OtpRequest
+
+ err := h.DB.Get(&otp, "SELECT token, num, openid FROM otp WHERE openid=$1", openid)
+ if err != nil {
+ http.Error(w, "OTP not found", http.StatusNotFound)
+ return
+ }
+
+ w.Header().Set("Content-Type", "application/json")
+ json.NewEncoder(w).Encode(otp)
+}
diff --git a/main.go b/main.go
new file mode 100644
index 0000000..1b34398
--- /dev/null
+++ b/main.go
@@ -0,0 +1,7 @@
+package main
+
+import "otpm/cmd"
+
+func main() {
+ cmd.Execute()
+}
diff --git a/utils/utils.go b/utils/utils.go
new file mode 100644
index 0000000..113ac27
--- /dev/null
+++ b/utils/utils.go
@@ -0,0 +1,54 @@
+package utils
+
+import (
+ "crypto/aes"
+ "crypto/cipher"
+ "encoding/base64"
+ "net/http"
+
+ "github.com/julienschmidt/httprouter"
+)
+
+func AdaptHandler(h func(http.ResponseWriter, *http.Request)) httprouter.Handle {
+ return func(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
+ h(w, r)
+ }
+}
+
+func AesDecrypt(encryptedData, sessionKey, iv string) ([]byte, error) {
+ //Base64解码
+ keyBytes, err := base64.StdEncoding.DecodeString(sessionKey)
+ if err != nil {
+ return nil, err
+ }
+ ivBytes, err := base64.StdEncoding.DecodeString(iv)
+ if err != nil {
+ return nil, err
+ }
+ cryptData, err := base64.StdEncoding.DecodeString(encryptedData)
+ if err != nil {
+ return nil, err
+ }
+ origData := make([]byte, len(cryptData))
+ //AES
+ block, err := aes.NewCipher(keyBytes)
+ if err != nil {
+ return nil, err
+ }
+ //CBC
+ mode := cipher.NewCBCDecrypter(block, ivBytes)
+ //解密
+ mode.CryptBlocks(origData, cryptData)
+ //去除填充位
+ origData = PKCS7UnPadding(origData)
+ return origData, nil
+}
+
+func PKCS7UnPadding(plantText []byte) []byte {
+ length := len(plantText)
+ if length > 0 {
+ unPadding := int(plantText[length-1])
+ return plantText[:(length - unPadding)]
+ }
+ return plantText
+}