コード例 #1
0
ファイル: taskmanager.go プロジェクト: lfkeitel/inca-tool
func ValidateTaskFile(filename string) {
	task, err := parser.ParseFile(filename)
	if err != nil {
		fmt.Printf("\nErrors found in \"%s\"\n", filename)
		fmt.Printf("   %s\n", err.Error())
		return
	}

	// Compile the script text
	_, err = parser.CompileCommandText(task.DefaultCommandBlock, task)
	if err != nil {
		if !parser.IsScriptRun(err) {
			fmt.Printf("\nErrors found in \"%s\"\n", filename)
			fmt.Printf("   %s\n", err.Error())
			return
		}
	}

	if verbose {
		fmt.Printf("\nInformation for Task \"%s\"\n", task.GetMetadata("name"))
		fmt.Printf("  Description: %s\n", task.GetMetadata("description"))
		fmt.Printf("  Author: %s\n", task.GetMetadata("author"))
		fmt.Printf("  Last Changed: %s\n", task.GetMetadata("date"))
		fmt.Printf("  Version: %s\n", task.GetMetadata("version"))

		fmt.Printf("  Concurrent Devices: %d\n", task.Concurrent)
		fmt.Printf("  Template: %s\n", task.Template)
		fmt.Printf("  Inventory File: %s\n\n", task.Inventory)

		fmt.Print("  ----Custom Data----\n")
		for k, v := range task.Metadata {
			if k[0] != '_' {
				continue
			}
			fmt.Printf("  %s: %s\n", k[1:], v)
		}

		fmt.Print("\n  ----Task Device Block----\n")
		for _, d := range task.Devices {
			fmt.Printf("  Device(s): %s\n", d)
		}

		fmt.Print("\n  ----Task Command Blocks----\n")
		for _, c := range task.Commands {
			fmt.Printf("  Command block Name: %s\n", c.Name)
			fmt.Printf("  Command block Type: %s\n", c.Type)
			fmt.Printf("  Commands:\n")
			for _, cmd := range c.Commands {
				fmt.Printf("     %s\n", cmd)
			}
			fmt.Println("  ---------------")
		}
	}
	fmt.Printf("The task named \"%s\" has no syntax errors.\n", task.GetMetadata("name"))
}
コード例 #2
0
ファイル: taskmanager.go プロジェクト: lfkeitel/inca-tool
func RunTaskFile(task *parser.TaskFile) {
	// Set scripts package settings
	scripts.SetVerbose(verbose)
	scripts.SetDebug(debug)
	scripts.SetDryRun(dryRun)

	os.RemoveAll("tmp")
	os.Mkdir("tmp", 0755)

	fmt.Printf("Running task %s @ %s\n", task.GetMetadata("name"), time.Now().String())

	// If no devices were given, print err and exit
	if len(task.Devices) == 0 {
		fmt.Println("No devices were given in the task file. Exiting.")
		return
	}

	// Load and filter devices
	if verbose {
		fmt.Printf("Loading inventory from %s\n", task.Inventory)
	}
	deviceList, err := devices.ParseFile(task.Inventory)
	if err != nil {
		fmt.Printf("Error loading devices: %s\n", err.Error())
		return
	}

	deviceList, err = devices.Filter(deviceList, task.Devices)
	if err != nil {
		fmt.Printf("Error: %s\n", err.Error())
		return
	}

	// If no devices will be affected, exit
	if len(deviceList.Devices) == 0 {
		fmt.Println("No devices match running task. Exiting.")
		return
	}

	// Compile the script text
	text, err := parser.CompileCommandText(task.DefaultCommandBlock, task)
	if err != nil {
		if parser.IsScriptRun(err) {
			// Run straight script file if prompted
			if err := scripts.ProcessScriptCommand(text, task, deviceList); err != nil {
				fmt.Printf("Error executing task: %s\n", err.Error())
			}
			return
		}
		fmt.Printf("Error compiling script: %s\n", err.Error())
		return
	}

	// Get the template file
	template := task.Template
	if template == "" {
		template = "expect"
	}
	templateFile := "templates/" + template + "-template.tmpl"
	if _, err := os.Stat(templateFile); os.IsNotExist(err) {
		fmt.Printf("Template not found: %s\n", template)
		return
	}

	// Generate an executable script file
	scriptFilename, err := scripts.GenerateBaseScriptFile(templateFile, text, task.GetAllMetadata())
	if err != nil {
		fmt.Printf("Error generating script: %s\n", err.Error())
		return
	}

	if debug {
		fmt.Printf("Script: %s\n", scriptFilename)
	}

	// Execute the script (the dry run setting will stop before actual execution)
	err = scripts.Execute(deviceList, task, scriptFilename, nil)
	if err != nil {
		fmt.Printf("Error executing task: %s\n", err.Error())
		return
	}

	if !debug {
		// Remove base script file
		os.Remove(scriptFilename)
	}

	if dryRun {
		fmt.Print("\nDry Run\n\n")
		for _, host := range deviceList.Devices {
			fmt.Printf("Hostname: %s\n", host.Name)
			fmt.Printf("Address: %s\n", host.GetSetting("address"))
			proto := host.GetSetting("protocol")
			if proto == "" {
				proto = "ssh"
			}
			fmt.Printf("Protocol: %s\n", proto)
			fmt.Println("---------")
		}
	}

	fmt.Printf("\nHosts touched: %d\n", len(deviceList.Devices))
}