/
main.go
98 lines (83 loc) · 2.31 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
96
97
98
/*
* Copyright (c) 2023 Juice Technologies, Inc. All Rights Reserved.
*/
package main
import (
"crypto/tls"
"flag"
"fmt"
"os"
"github.com/Juice-Labs/Juice-Labs/cmd/agent/app"
"github.com/Juice-Labs/Juice-Labs/cmd/agent/playnite"
"github.com/Juice-Labs/Juice-Labs/cmd/agent/prometheus"
"github.com/Juice-Labs/Juice-Labs/cmd/internal/build"
"github.com/Juice-Labs/Juice-Labs/pkg/appmain"
"github.com/Juice-Labs/Juice-Labs/pkg/crypto"
"github.com/Juice-Labs/Juice-Labs/pkg/logger"
"github.com/Juice-Labs/Juice-Labs/pkg/sentry"
"github.com/Juice-Labs/Juice-Labs/pkg/task"
"github.com/joho/godotenv"
)
var (
certFile = flag.String("cert-file", "", "")
keyFile = flag.String("key-file", "", "")
generateCert = flag.Bool("generate-cert", false, "Generates a certificate for https")
)
func main() {
name := "Juice Agent"
config := appmain.Config{
Name: name,
Version: build.Version,
SentryConfig: sentry.ClientOptions{
Dsn: os.Getenv("JUICE_AGENT_SENTRY_DSN"),
Release: fmt.Sprintf("%s@%s", name, build.Version),
EnableTracing: true,
TracesSampleRate: 1.0,
},
}
err := appmain.Run(config, func(group task.Group) error {
var tlsConfig *tls.Config
if *certFile != "" && *keyFile != "" {
certificate, err := tls.LoadX509KeyPair(*certFile, *keyFile)
if err != nil {
return err
}
tlsConfig = &tls.Config{
InsecureSkipVerify: true,
Certificates: []tls.Certificate{certificate},
}
} else if *generateCert {
certificate, err := crypto.GenerateCertificate()
if err != nil {
return err
}
tlsConfig = &tls.Config{
InsecureSkipVerify: true,
Certificates: []tls.Certificate{certificate},
}
}
if err := godotenv.Load(); err != nil {
logger.Infof("Could not load .env file: %v", err)
}
agent, err := app.NewAgent(group.Ctx(), tlsConfig)
if err == nil {
consumer, err_ := playnite.NewGpuMetricsConsumer(agent)
err = err_
if err != nil {
logger.Warning(err)
}
agent.GpuMetricsProvider.AddConsumer(consumer)
agent.GpuMetricsProvider.AddConsumer(prometheus.NewGpuMetricsConsumer())
err = agent.ConnectToController(group, tlsConfig)
if err == nil {
group.Go("Agent", agent)
} else {
group.Cancel()
}
}
return err
})
if err != nil {
os.Exit(appmain.ExitFailure)
}
}