-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
95 lines (81 loc) · 1.93 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
package main
import (
"context"
"flag"
"log"
"os"
"os/signal"
"path/filepath"
"strings"
"syscall"
"github.com/ophum/k8s-secret-fs/pkg/fs"
"github.com/ophum/k8s-secret-fs/pkg/k8s"
"gopkg.in/yaml.v3"
"k8s.io/client-go/util/homedir"
)
var (
client *k8s.Client
config *Config
)
type Config struct {
Kubeconfig string `yaml:"kubeconfig"`
Namespace string `yaml:"namespace"`
SecretName string `yaml:"secretName"`
MountPoint string `yaml:"mountPoint"`
}
func init() {
var configPath string
flag.StringVar(&configPath, "config", "", "-config config.yaml")
var kubeconfig *string
if home := homedir.HomeDir(); home != "" {
kubeconfig = flag.String("kubeconfig", filepath.Join(home, ".kube", "config"), "(optional) absolute path to the kubeconfig file")
} else {
kubeconfig = flag.String("kubeconfig", "", "absolute path to the kubeconfig file")
}
flag.Parse()
config = &Config{}
if configPath == "" {
config.Kubeconfig = *kubeconfig
namespacedSecretName := flag.Arg(0)
splited := strings.Split(namespacedSecretName, "/")
switch len(splited) {
case 1:
config.Namespace = "default"
config.SecretName = splited[0]
case 2:
config.Namespace = splited[0]
config.SecretName = splited[1]
default:
log.Panic("invalid namespace/secretName")
}
config.MountPoint = flag.Arg(1)
} else {
f, err := os.Open(configPath)
if err != nil {
log.Panic(err)
}
defer f.Close()
if err := yaml.NewDecoder(f).Decode(config); err != nil {
log.Panic(err)
}
}
var err error
client, err = k8s.NewClient(config.Kubeconfig, config.Namespace, config.SecretName)
if err != nil {
log.Panic(err)
}
}
func main() {
secretFS := fs.NewSecretFS(client)
server, err := secretFS.Mount(config.MountPoint)
if err != nil {
log.Panic(err)
}
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, os.Kill, syscall.SIGTERM)
defer stop()
go func() {
<-ctx.Done()
server.Unmount()
}()
server.Wait()
}