forked from aws/amazon-ecs-agent
/
mockgen.go
94 lines (81 loc) · 2.43 KB
/
mockgen.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
// Copyright 2016-2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may
// not use this file except in compliance with the License. A copy of the
// License is located at
//
// http://aws.amazon.com/apache2.0/
//
// or in the "license" file accompanying this file. This file 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.
package main
import (
"bytes"
"fmt"
"os"
"os/exec"
"path/filepath"
"regexp"
"time"
)
const (
projectVendor = `github.com/aws/amazon-ecs-agent/agent/vendor`
copyrightHeaderFormat = "// Copyright 2015-%v Amazon.com, Inc. or its affiliates. All Rights Reserved."
licenseBlock = `
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may
// not use this file except in compliance with the License. A copy of the
// License is located at
//
// http://aws.amazon.com/apache2.0/
//
// or in the "license" file accompanying this file. This file 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.
`
)
func main() {
if len(os.Args) != 4 {
usage()
os.Exit(1)
}
packageName := os.Args[1]
interfaces := os.Args[2]
outputPath := os.Args[3]
re := regexp.MustCompile("(?m)[\r\n]+^.*" + projectVendor + ".*$")
copyrightHeader := fmt.Sprintf(copyrightHeaderFormat, time.Now().Year())
path, _ := filepath.Split(outputPath)
err := os.MkdirAll(path, os.ModeDir|0755)
if err != nil {
fmt.Println(err)
os.Exit(1)
}
outputFile, err := os.Create(outputPath)
if err != nil {
fmt.Println(err)
os.Exit(1)
}
mockgen := exec.Command("mockgen", packageName, interfaces)
mockgenOut, err := mockgen.Output()
if err != nil {
fmt.Println(err)
os.Exit(1)
}
sanitized := re.ReplaceAllString(string(mockgenOut), "")
withHeader := copyrightHeader + licenseBlock + sanitized
goimports := exec.Command("goimports")
goimports.Stdin = bytes.NewBufferString(withHeader)
goimports.Stdout = outputFile
err = goimports.Run()
outputFile.Close()
if err != nil {
fmt.Println(err)
os.Exit(1)
}
}
func usage() {
fmt.Println(os.Args[0], " PACKAGE INTERFACE_NAMES OUTPUT_FILE")
}