Initial Commit

internal/sshctx/sshctx.go

@@ -0,0 +1,47 @@
+/*
+ * Seashell - SSH server with virtual hosts and username-based routing
+ *
+ * Copyright (C) 2024 Elara6331 <elara@elara.ws>
+ *
+ * This file is part of Seashell.
+ *
+ * Seashell is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * Seashell is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with Seashell.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package sshctx
+
+import (
+	"context"
+
+	"github.com/gliderlabs/ssh"
+	"go.elara.ws/seashell/internal/config"
+)
+
+type (
+	argCtxKey  struct{}
+	userCtxKey struct{}
+)
+
+func SetArg(ctx ssh.Context, arg string)        { ctx.SetValue(argCtxKey{}, arg) }
+func SetUser(ctx ssh.Context, user config.User) { ctx.SetValue(userCtxKey{}, user) }
+
+func GetArg(ctx context.Context) (string, bool) {
+	arg, ok := ctx.Value(argCtxKey{}).(string)
+	return arg, ok
+}
+
+func GetUser(ctx context.Context) (config.User, bool) {
+	user, ok := ctx.Value(userCtxKey{}).(config.User)
+	return user, ok
+}