server/discovery: add discovery service

tools/cmd/cproto/main.go

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package main
 
 import (
         "flag"
         "fmt"
         "go/build"
         "io/ioutil"
         "os"
         "os/exec"
         "path"
         "path/filepath"
         "strings"
         "syscall"
 
         "github.com/luci/luci-go/common/logging"
         "github.com/luci/luci-go/common/logging/gologger"
         gol "github.com/op/go-logging"
         "golang.org/x/net/context"
 )
 
 var (
-        verbose    = flag.Bool("verbose", false, "print debug messages to stderr")
+        verbose       = flag.Bool("verbose", false, "print debug messages to stderr")
+        withDiscovery = flag.Bool(
+                "discovery", true,
+                "generate pb.discovery.go file")
         withGithub = flag.Bool(
                 "with-github", true,
                 "include $GOPATH/src/github.com in proto search path")
         withGoogleProtobuf = flag.Bool(
                 "with-google-protobuf", true,
                 "map .proto files in gitub.com/luci/luci-go/common/proto/google to google/protobuf/*.proto")
         renameToTestGo = flag.Bool(
                 "test", false,
                 "rename generated files from *.go to *_test.go")
 )
@@ skipping 112 matching lines @@
                 }
 
                 if *renameToTestGo {
                         newName := strings.TrimSuffix(goFile, ".go") + "_test.go"
                         if err := os.Rename(goFile, newName); err != nil {
                                 return err
                         }
                 }
         }
 
+        if !*withDiscovery {
+                return nil
+        }
         // Generate pb.prpc.go
         discoveryFile := "pb.discovery.go"
         if *renameToTestGo {
                 discoveryFile = "pb.discovery_test.go"
         }
         return genDiscoveryFile(c, filepath.Join(dir, discoveryFile), descPath, packageName)
 }
 
 func setupLogging(c context.Context) context.Context {
         logCfg := gologger.LoggerConfig{
@@ skipping 62 matching lines @@
         src := strings.TrimSuffix(dir, pkg)
         src = path.Clean(src)
         goPaths := strings.Split(os.Getenv("GOPATH"), string(filepath.ListSeparator))
         for _, goPath := range goPaths {
                 if filepath.Join(goPath, "src") == src {
                         return true, nil
                 }
         }
         return false, nil
 }