open-im-server/magefile.go at main · openimsdk/open-im-server · 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
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
//go:build mage
// +build mage

package main

import (
	"flag"
	"fmt"
	"os"

	"github.com/openimsdk/gomake/mageutil"
	"github.com/openimsdk/open-im-server/v3/version"
	"github.com/openimsdk/tools/utils/datautil"
)

var Default = Build

var Aliases = map[string]any{
	"buildcc": BuildWithCustomConfig,
	"startcc": StartWithCustomConfig,
}

var (
	customRootDir   = "."       // workDir in mage, default is "./"(project root directory)
	customSrcDir    = "cmd"     // source code directory, default is "cmd"
	customOutputDir = "_output" // output directory, default is "_output"
	customConfigDir = "config"  // configuration directory, default is "config"
	customToolsDir  = "tools"   // tools source code directory, default is "tools"
)

// Build support specifical binary build.
//
// Example: `mage build openim-api openim-rpc-user seq`
func Build() {
	flag.Parse()
	bin := flag.Args()
	if len(bin) != 0 {
		bin = bin[1:]
	}
	mageutil.WithSpinner("Building binaries...", func() { mageutil.Build(bin, nil, nil) })
}

func BuildWithCustomConfig() {
	flag.Parse()
	bin := flag.Args()
	if len(bin) != 0 {
		bin = bin[1:]
	}

	config := &mageutil.PathOptions{
		RootDir:   &customRootDir,
		OutputDir: &customOutputDir,
		SrcDir:    &customSrcDir,
		ToolsDir:  &customToolsDir,
	}

	mageutil.WithSpinner("Building binaries with custom config...", func() {
		mageutil.Build(bin, config, nil)
	})
}

func Start() {
	mageutil.InitForSSC()
	err := setMaxOpenFiles()
	if err != nil {
		mageutil.PrintRed("setMaxOpenFiles failed " + err.Error())
		os.Exit(1)
	}

	flag.Parse()
	bin := flag.Args()
	if len(bin) != 0 {
		bin = bin[1:]
	}

	mageutil.WithSpinner("Starting...", func() {
		mageutil.StartToolsAndServices(bin, nil)
	})
}

func StartWithCustomConfig() {
	mageutil.InitForSSC()
	err := setMaxOpenFiles()
	if err != nil {
		mageutil.PrintRed("setMaxOpenFiles failed " + err.Error())
		os.Exit(1)
	}

	flag.Parse()
	bin := flag.Args()
	if len(bin) != 0 {
		bin = bin[1:]
	}

	config := &mageutil.PathOptions{
		RootDir:   &customRootDir,
		OutputDir: &customOutputDir,
		ConfigDir: &customConfigDir,
	}

	mageutil.WithSpinner("Starting with custom config...", func() {
		mageutil.StartToolsAndServices(bin, config)
	})
}

func Stop() {
	mageutil.WithSpinner("Stopping...", mageutil.StopAndCheckBinaries)
}

func Check() {
	mageutil.WithSpinner("Checking binaries...", mageutil.CheckAndReportBinariesStatus)
}

func Export() {
	mappingPaths, err := mageutil.GetDefaultExportMappingPaths([]string{
		"cmd",
		"internal",
		"pkg",
		"test",
		"tools",
		"**/*.go",
		"go.mod",
		"go.work",
	})
	if err != nil {
		mageutil.PrintRed("GetDefaultExportMappingPaths failed " + err.Error())
		os.Exit(1)
	}

	mageutil.WithSpinner("Exporting...", func() {
		mageutil.ExportMageLauncherArchived(mappingPaths, &mageutil.ExportOptions{
			ProjectName: datautil.ToPtr(fmt.Sprintf("open-im-server_%s", version.Version)),
			BuildOpt: &mageutil.BuildOptions{
				Release:  datautil.ToPtr(true),
				Compress: datautil.ToPtr(true),
			},
		})
	})
}