示例#1
0
func main() {
	flag.Parse()
	args := flag.Args()
	if len(args) < 2 {
		fail("%s", usage)
	}

	sourceDir := getSourceDir(args[0])
	interfaceName := args[1]
	fakeName := getFakeName(interfaceName, *fakeNameFlag)
	outputPath := getOutputPath(sourceDir, fakeName, *outputPathFlag)
	outputDir := filepath.Dir(outputPath)
	fakePackageName := filepath.Base(outputDir)
	shouldPrintToStdout := len(args) >= 3 && args[2] == "-"

	iface, err := locator.GetInterfaceFromFilePath(interfaceName, sourceDir)
	if err != nil {
		fail("%v", err)
	}

	code, err := generator.CodeGenerator{
		Model:       *iface,
		StructName:  fakeName,
		PackageName: fakePackageName,
	}.GenerateFake()

	if err != nil {
		fail("%v", err)
	}

	if shouldPrintToStdout {
		fmt.Println(code)
	} else {
		os.MkdirAll(outputDir, 0777)
		file, err := os.Create(outputPath)
		if err != nil {
			fail("Couldn't create fake file - %v", err)
		}

		_, err = file.WriteString(code)
		if err != nil {
			fail("Couldn't write to fake file - %v", err)
		}

		rel, err := filepath.Rel(cwd(), outputPath)
		if err != nil {
			fail("%v", err)
		}

		fmt.Printf("Wrote `%s` to `%s`\n", fakeName, rel)
	}
}
示例#2
0
func main() {
	flag.Parse()
	args := flag.Args()

	if len(args) < 1 {
		fail("%s", usage)
		return
	}

	argumentParser := arguments.NewArgumentParser(
		fail,
		cwd,
		filepath.EvalSymlinks,
		os.Stat,
		terminal.NewUI(),
		locator.NewInterfaceLocator(),
	)
	parsedArgs := argumentParser.ParseArguments(args...)

	interfaceName := parsedArgs.InterfaceName
	fakeName := parsedArgs.FakeImplName
	sourceDir := parsedArgs.SourcePackageDir
	outputPath := parsedArgs.OutputPath

	outputDir := filepath.Dir(outputPath)
	fakePackageName := filepath.Base(outputDir)

	iface, err := locator.GetInterfaceFromFilePath(interfaceName, sourceDir)
	if err != nil {
		fail("%v", err)
	}

	code, err := generator.CodeGenerator{
		Model:       *iface,
		StructName:  fakeName,
		PackageName: fakePackageName,
	}.GenerateFake()

	if err != nil {
		fail("%v", err)
	}

	newCode, err := format.Source([]byte(code))
	code = string(newCode)

	if parsedArgs.PrintToStdOut {
		fmt.Println(code)
	} else {
		os.MkdirAll(outputDir, 0777)
		file, err := os.Create(outputPath)
		if err != nil {
			fail("Couldn't create fake file - %v", err)
		}

		_, err = file.WriteString(code)
		if err != nil {
			fail("Couldn't write to fake file - %v", err)
		}

		rel, err := filepath.Rel(cwd(), outputPath)
		if err != nil {
			fail("%v", err)
		}

		fmt.Printf("Wrote `%s` to `%s`\n", fakeName, rel)
	}
}