dashboard/cmd/gomote/put.go

// Copyright 2015 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

// +build extdep

package main

import (
	"flag"
	"fmt"
	"io"
	"net/http"
	"os"
)

// put a .tar.gz
func putTar(args []string) error {
	fs := flag.NewFlagSet("put", flag.ContinueOnError)
	fs.Usage = func() {
		fmt.Fprintln(os.Stderr, "create usage: gomote puttar [put-opts] <buildlet-name> [tar.gz file or '-' for stdin]\n")
		fs.PrintDefaults()
		os.Exit(1)
	}
	var rev string
	fs.StringVar(&rev, "gorev", "", "If non-empty, git hash to download from gerrit and put to the buildlet. e.g. 886b02d705ff for Go 1.4.1")

	fs.Parse(args)
	if fs.NArg() < 1 || fs.NArg() > 2 {
		fs.Usage()
	}

	name := fs.Arg(0)
	bc, err := namedClient(name)
	if err != nil {
		return err
	}

	var tgz io.Reader = os.Stdin
	if rev != "" {
		if fs.NArg() != 1 {
			fs.Usage()
		}
		// TODO(bradfitz): tell the buildlet to do this
		// itself, to avoid network to & from home networks.
		// Staying Google<->Google will be much faster.
		res, err := http.Get("https://go.googlesource.com/go/+archive/" + rev + ".tar.gz")
		if err != nil {
			return fmt.Errorf("Error fetching rev %s from Gerrit: %v", rev, err)
		}
		defer res.Body.Close()
		if res.StatusCode != 200 {
			return fmt.Errorf("Error fetching rev %s from Gerrit: %v", rev, res.Status)
		}
		tgz = res.Body
	} else if fs.NArg() == 2 && fs.Arg(1) != "-" {
		f, err := os.Open(fs.Arg(1))
		if err != nil {
			return err
		}
		defer f.Close()
		tgz = f
	}
	return bc.PutTarball(tgz)
}

// put single files
func put(args []string) error {
	fs := flag.NewFlagSet("put", flag.ContinueOnError)
	fs.Usage = func() {
		fmt.Fprintln(os.Stderr, "create usage: gomote put [put-opts] <type>\n\n")
		fs.PrintDefaults()
		os.Exit(1)
	}
	fs.Parse(args)
	if fs.NArg() != 1 {
		fs.Usage()
	}
	return fmt.Errorf("TODO")
	builderType := fs.Arg(0)
	_ = builderType
	return nil
}