-
Notifications
You must be signed in to change notification settings - Fork 11
/
main.go
190 lines (152 loc) Β· 4.28 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
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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
package main
import (
"fmt"
"log"
"os"
"path/filepath"
"runtime"
"strconv"
"strings"
"time"
"github.com/cloudfoundry-attic/jibber_jabber"
"github.com/itchio/itch-setup/bindata"
"github.com/itchio/itch-setup/cl"
"github.com/itchio/itch-setup/localize"
"github.com/itchio/itch-setup/native"
"github.com/pkg/errors"
kingpin "gopkg.in/alecthomas/kingpin.v2"
)
var (
version = "head" // set by command-line on CI release builds
builtAt = "" // set by command-line on CI release builds
commit = "" // set by command-line on CI release builds
versionString = "" // formatted on boot from 'version' and 'builtAt'
app = kingpin.New("itch-setup", "The itch installer and self-updater")
)
var cli cl.CLI
func init() {
app.Flag("prefer-launch", "Launch if a valid version of itch is installed").BoolVar(&cli.PreferLaunch)
app.Flag("upgrade", "Upgrade the itch app if necessary").BoolVar(&cli.Upgrade)
app.Flag("uninstall", "Uninstall the itch app").BoolVar(&cli.Uninstall)
app.Flag("relaunch", "Relaunch a new version of the itch app").BoolVar(&cli.Relaunch)
app.Flag("relaunch-pid", "PID to wait for before relaunching").IntVar(&cli.RelaunchPID)
app.Flag("appname", "Application name (itch or kitch)").StringVar(&cli.AppName)
app.Flag("silent", "Run installation silently").BoolVar(&cli.Silent)
app.Arg("args", "Arguments to pass down to itch (only supported on Linux & Windows)").StringsVar(&cli.Args)
}
func must(err error) {
if err != nil {
log.Fatalf("%+v", err)
}
}
func detectAppName() {
if cli.AppName != "" {
log.Printf("App name specified on command-line: %s", cli.AppName)
} else {
execPath, err := os.Executable()
must(err)
ext := ""
if runtime.GOOS == "windows" {
ext = ".exe"
}
kitchBinary := fmt.Sprintf("kitch-setup%s", ext)
if filepath.Base(execPath) == kitchBinary {
cli.AppName = "kitch"
} else {
cli.AppName = "itch"
}
log.Printf("App name detected: %s", cli.AppName)
}
app.Name = fmt.Sprintf("%s-setup", cli.AppName)
}
const DefaultLocale = "en-US"
var localizer *localize.Localizer
func main() {
app.UsageTemplate(kingpin.CompactUsageTemplate)
app.HelpFlag.Short('h')
if builtAt != "" {
epoch, err := strconv.ParseInt(builtAt, 10, 64)
must(err)
versionString = fmt.Sprintf("%s, built on %s", version, time.Unix(epoch, 0).Format("Jan _2 2006 @ 15:04:05"))
} else {
versionString = fmt.Sprintf("%s, no build date", version)
}
if commit != "" {
versionString = fmt.Sprintf("%s, ref %s", versionString, commit)
}
app.Version(versionString)
app.VersionFlag.Short('V')
app.Author("Amos Wenger <amos@itch.io>")
cli.VersionString = versionString
_, err := app.Parse(os.Args[1:])
must(err)
detectAppName()
userLocale, err := jibber_jabber.DetectIETF()
if err != nil {
log.Println("Couldn't detect locale, falling back to default", DefaultLocale)
userLocale = "en-US"
}
log.Println("Locale: ", userLocale)
localizer, err = localize.NewLocalizer(bindata.Asset)
if err != nil {
log.Fatal(err)
}
err = localizer.LoadLocale(userLocale)
if err != nil {
userLocale = userLocale[:2]
err = localizer.LoadLocale(userLocale)
}
if err == nil {
localizer.SetLang(userLocale)
}
cli.Localizer = localizer
nc, err := native.NewNativeCore(cli)
if err != nil {
panic(err)
}
var verbs []string
if cli.Upgrade {
verbs = append(verbs, "upgrade")
}
if cli.Relaunch {
verbs = append(verbs, "relaunch")
}
if cli.Uninstall {
verbs = append(verbs, "uninstall")
}
if len(verbs) > 1 {
nc.ErrorDialog(errors.Errorf("Cannot specify more than one verb: got %s", strings.Join(verbs, ", ")))
}
if len(verbs) == 0 {
verbs = append(verbs, "install")
}
switch verbs[0] {
case "install":
err = nc.Install()
if err != nil {
nc.ErrorDialog(err)
}
case "upgrade":
err = nc.Upgrade()
if err != nil {
jsonlBail(errors.WithMessage(err, "Fatal upgrade error"))
}
case "relaunch":
if cli.RelaunchPID <= 0 {
jsonlBail(errors.Errorf("--relaunch needs a valid --relaunch-pid (got %d)", cli.RelaunchPID))
}
err = nc.Relaunch()
if err != nil {
jsonlBail(errors.WithMessage(err, "Fatal relaunch error"))
}
case "uninstall":
err = nc.Uninstall()
if err != nil {
nc.ErrorDialog(err)
}
}
}
func jsonlBail(err error) {
// TODO: use json-lines
log.Fatalf("%+v", err)
}