blob: 7e75eefda731957db5ec92042b86405ba80ba558 [file] [log] [blame]
// Copyright (C) 2016 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//go:generate protoc --proto_path ../.. --js_out ./www/proto ../../tools/robotester/db/db.proto
//go:generate protoc --proto_path ../.. --js_out ./www/proto ../../framework/device/device.proto
package main
import (
"bufio"
"flag"
"fmt"
_ "net/http/pprof"
"os"
"android.googlesource.com/platform/tools/gpu/framework/app"
"android.googlesource.com/platform/tools/gpu/framework/file"
"android.googlesource.com/platform/tools/gpu/framework/log"
"android.googlesource.com/platform/tools/gpu/framework/task"
"android.googlesource.com/platform/tools/gpu/tools/robotester"
"android.googlesource.com/platform/tools/gpu/tools/robotester/db"
"android.googlesource.com/platform/tools/gpu/tools/robotester/server"
"android.googlesource.com/platform/tools/gpu/tools/robotester/service"
)
const defaultDatabase = "mysql://buildbot:buildbot@localhost:3306/buildbot"
var (
build file.Path
www file.Path
data file.Path
port int
dbAddress app.URLFlag
offline bool
ignoreLocalChanges bool
eraseDatabase bool
useLDPreload bool
)
func main() {
dbAddress.Set(defaultDatabase)
flag.Var(&build, "build", "The path to the build output directory.")
flag.Var(&www, "www", "The path to the www directory.")
flag.Var(&data, "data", "The path to data directory.")
flag.IntVar(&port, "port", 9000, "The webserver's port.")
flag.Var(&dbAddress, "db", "The database to connect to")
flag.BoolVar(&offline, "offline", false, "Don't fetch new changes.")
flag.BoolVar(&ignoreLocalChanges, "ignore-local-changes", false, "Permit local changes to the source directory.")
flag.BoolVar(&eraseDatabase, "erase-database", false, "Wipes-clean the entire database and then returns.")
flag.BoolVar(&useLDPreload, "ld-preload", false, "Use LD_PRELOAD to register interceptor instead of JDWP.")
app.DefaultStyle = "Detailed"
app.ShortHelp = "RoboTester is an automated testing server for the GAPI project."
app.Version = app.VersionSpec{Major: 0, Minor: 1}
app.Run(run)
}
func run(ctx log.Context) error {
ctx, cancel := task.WithCancel(ctx)
// Start by logging into the database
db, err := db.Open(ctx, dbAddress.URL)
if err != nil {
return ctx.Cause(err).LogError("Couldn't log into database")
}
defer db.Close(ctx)
if eraseDatabase {
fmt.Println(`
**********************************
*** ERASE THE ENTIRE DATABASE? ***
**********************************
You are about to erase the entire database.
This cannot be undone.
To proceed, type 'YES' and press enter:
`)
l, err := bufio.NewReader(os.Stdin).ReadString('\n')
if err != nil {
return err
}
if l == "YES\n" {
if err := db.EraseAll(ctx); err != nil {
return err
}
println("Database cleared.")
} else {
println("Aborted")
}
return nil
}
go func() {
println("Press enter to stop...")
os.Stdin.Read([]byte{0})
cancel()
}()
if data.IsEmpty() {
data = file.Abs("data")
}
if build.IsEmpty() {
return ctx.AsError("You must supply the path to a valid build directory")
}
config, err := service.Prepare(ctx, build, data)
if err != nil {
return err
}
if www.IsEmpty() {
www = file.Abs(config.Source).Join("cmd/robotester/www")
}
if useLDPreload {
config.UseLDPreload = true
}
if offline {
config.Offline = true
}
if ignoreLocalChanges {
config.IgnoreLocalChanges = true
}
server.Run(ctx, server.Config{
WWW: www.System(),
Data: config.Data,
Src: config.Source,
Artifacts: config.Artifacts,
Address: fmt.Sprintf(":%d", port),
}, db)
return robotester.Run(ctx, config, db)
}