summaryrefslogtreecommitdiff
path: root/cmd
diff options
context:
space:
mode:
authorMike Crute <mike@crute.us>2023-08-03 13:48:28 -0700
committerMike Crute <mike@crute.us>2023-08-03 13:48:28 -0700
commita09b5eb2a787fc6671ea1d2452452b1e58df6a96 (patch)
tree5dfd67d648363895b365003262cb92025561b59d /cmd
parent08c9439879329c653c2fabd7c8c9bf7f4c145d66 (diff)
downloadwebsocket_proxy-a09b5eb2a787fc6671ea1d2452452b1e58df6a96.tar.bz2
websocket_proxy-a09b5eb2a787fc6671ea1d2452452b1e58df6a96.tar.xz
websocket_proxy-a09b5eb2a787fc6671ea1d2452452b1e58df6a96.zip
Migrate MongoDbBasicStore to golib version
Diffstat (limited to 'cmd')
-rw-r--r--cmd/register/register.go5
-rw-r--r--cmd/web/server.go7
2 files changed, 5 insertions, 7 deletions
diff --git a/cmd/register/register.go b/cmd/register/register.go
index fdd083c..c00b54b 100644
--- a/cmd/register/register.go
+++ b/cmd/register/register.go
@@ -10,7 +10,6 @@ import (
10 glecho "code.crute.us/mcrute/golib/echo" 10 glecho "code.crute.us/mcrute/golib/echo"
11 "code.crute.us/mcrute/ssh-proxy/app" 11 "code.crute.us/mcrute/ssh-proxy/app"
12 "code.crute.us/mcrute/ssh-proxy/app/models" 12 "code.crute.us/mcrute/ssh-proxy/app/models"
13 "code.crute.us/mcrute/ssh-proxy/db"
14 "github.com/spf13/cobra" 13 "github.com/spf13/cobra"
15) 14)
16 15
@@ -43,13 +42,13 @@ func registerMain(cfg app.Config, username string) {
43 log.Fatalf("Error connecting to mongodb: %s", err) 42 log.Fatalf("Error connecting to mongodb: %s", err)
44 } 43 }
45 44
46 userStore := &db.MongoDbBasicStore[*models.User]{ 45 userStore := &mongodb.MongoDbBasicStore[*models.User]{
47 Db: mongo, 46 Db: mongo,
48 CollectionName: "users", 47 CollectionName: "users",
49 } 48 }
50 49
51 authSessionStore := &models.AuthSessionStoreMongodb{ 50 authSessionStore := &models.AuthSessionStoreMongodb{
52 MongoDbBasicStore: &db.MongoDbBasicStore[*models.AuthSession]{ 51 MongoDbBasicStore: &mongodb.MongoDbBasicStore[*models.AuthSession]{
53 Db: mongo, 52 Db: mongo,
54 CollectionName: "oauth_sessions", 53 CollectionName: "oauth_sessions",
55 }, 54 },
diff --git a/cmd/web/server.go b/cmd/web/server.go
index 83f4fc8..19c1bb7 100644
--- a/cmd/web/server.go
+++ b/cmd/web/server.go
@@ -14,7 +14,6 @@ import (
14 "code.crute.us/mcrute/ssh-proxy/app/controllers" 14 "code.crute.us/mcrute/ssh-proxy/app/controllers"
15 "code.crute.us/mcrute/ssh-proxy/app/middleware" 15 "code.crute.us/mcrute/ssh-proxy/app/middleware"
16 "code.crute.us/mcrute/ssh-proxy/app/models" 16 "code.crute.us/mcrute/ssh-proxy/app/models"
17 "code.crute.us/mcrute/ssh-proxy/db"
18 17
19 "code.crute.us/mcrute/golib/cli" 18 "code.crute.us/mcrute/golib/cli"
20 "code.crute.us/mcrute/golib/clients/autocert/v2" 19 "code.crute.us/mcrute/golib/clients/autocert/v2"
@@ -127,18 +126,18 @@ func webMain(cfg app.Config, embeddedTemplates, embeddedClients fs.FS, appVersio
127 log.Fatalf("Error creating session store: %s", err) 126 log.Fatalf("Error creating session store: %s", err)
128 } 127 }
129 128
130 userStore := &db.MongoDbBasicStore[*models.User]{ 129 userStore := &mongodb.MongoDbBasicStore[*models.User]{
131 Db: mongo, 130 Db: mongo,
132 CollectionName: "users", 131 CollectionName: "users",
133 } 132 }
134 133
135 oauthClientStore := &db.MongoDbBasicStore[*models.OauthClient]{ 134 oauthClientStore := &mongodb.MongoDbBasicStore[*models.OauthClient]{
136 Db: mongo, 135 Db: mongo,
137 CollectionName: "oauth_clients", 136 CollectionName: "oauth_clients",
138 } 137 }
139 138
140 authSessionStore := &models.AuthSessionStoreMongodb{ 139 authSessionStore := &models.AuthSessionStoreMongodb{
141 MongoDbBasicStore: &db.MongoDbBasicStore[*models.AuthSession]{ 140 MongoDbBasicStore: &mongodb.MongoDbBasicStore[*models.AuthSession]{
142 Db: mongo, 141 Db: mongo,
143 CollectionName: "oauth_sessions", 142 CollectionName: "oauth_sessions",
144 }, 143 },