bilirec/main.go at main · eric2788/bilirec · GitHub

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
package main

import (
	"os"
	"time"

	"github.com/eric2788/bilirec/internal/controllers/convert"
	"github.com/eric2788/bilirec/internal/controllers/file"
	"github.com/eric2788/bilirec/internal/controllers/record"
	"github.com/eric2788/bilirec/internal/controllers/room"
	"github.com/eric2788/bilirec/internal/modules/bilibili"
	"github.com/eric2788/bilirec/internal/modules/config"
	"github.com/eric2788/bilirec/internal/modules/rest"
	co "github.com/eric2788/bilirec/internal/services/convert"
	fi "github.com/eric2788/bilirec/internal/services/file"
	pa "github.com/eric2788/bilirec/internal/services/path"
	re "github.com/eric2788/bilirec/internal/services/recorder"
	ro "github.com/eric2788/bilirec/internal/services/room"
	st "github.com/eric2788/bilirec/internal/services/stream"
	"github.com/eric2788/bilirec/utils"
	"go.uber.org/fx"
)

func MainModule() fx.Option {
	return fx.Module("main",
		config.Module,
		bilibili.Module,
		rest.Module,

		fx.Provide(pa.NewService),
		fx.Provide(co.NewService),
		fx.Provide(st.NewService),
		fx.Provide(re.NewService),
		fx.Provide(ro.NewService),
		fx.Provide(fi.NewService),

		fx.Invoke(room.NewController),
		fx.Invoke(record.NewController),
		fx.Invoke(file.NewController),
		fx.Invoke(convert.NewController),
	)
}

func main() {

	app := fx.New(
		MainModule(),
		fx.StartTimeout(utils.Ternary(os.Getenv("ANONYMOUS_LOGIN") == "true", 15*time.Second, 1*time.Minute)),
		fx.StopTimeout(1*time.Minute),
	)

	app.Run()
}