package main import fmt go ast go format go parser go token os func in

package main

import (
	"fmt"
	"go/ast"
	"go/format"
	"go/parser"
	"go/token"
	"os"
)

func insertHello(file *ast.File) {
	ast.Inspect(file, func(node ast.Node) bool {
		//here code of internal func
		if switchStmt, ok := node.(*ast.SwitchStmt); ok {
			//what to do if we in switch
			var flag bool = false
			for _, element := range switchStmt.Body.List {
				//in for
				if element.(*ast.CaseClause).List == nil {
					flag = true
				}
			}
			if flag == false {
				//add default case and expression for it

				//fmt.Printf("exists switch without default\n")
				switchStmt.Body.List = append(
					[]ast.Stmt{

						&ast.CaseClause{

							List: nil,
							Body: []ast.Stmt{
								&ast.ExprStmt{
									X: &ast.CallExpr{
										Fun: &ast.SelectorExpr{
											X:   ast.NewIdent("fmt"),
											Sel: ast.NewIdent("Printf")},
										Args: []ast.Expr{
											&ast.BasicLit{
												Kind:  token.STRING,
												Value: "\"hello\"",
											},
										},
									},
								},
							},
						},
					}, switchStmt.Body.List...,
				)
			}

		}
		return true
	})
}

func main() {
	if len(os.Args) != 2 {
		fmt.Printf("usage: astprint <filename.go>\n")
		return
	}

	// Создаём хранилище данных об исходных файлах
	fset := token.NewFileSet()

	// Вызываем парсер
	if file, err := parser.ParseFile(
		fset,                 // данные об исходниках
		os.Args[1],           // имя файла с исходником программы
		nil,                  // пусть парсер сам загрузит исходник
		parser.ParseComments, // приказываем сохранять комментарии
	); err == nil {
		// Если парсер отработал без ошибок, печатаем дерево
		//ast.Fprint(os.Stdout, fset, file, nil)
		//insert code here
		insertHello(file)
		format.Node(os.Stdout, fset, file)
	} else {
		// в противном случае, выводим сообщение об ошибке
		fmt.Printf("Error: %v", err)
	}
}