89 lines
2.2 KiB
Go
89 lines
2.2 KiB
Go
package ast
|
|
|
|
import (
|
|
"go/ast"
|
|
"go/format"
|
|
"go/parser"
|
|
"go/token"
|
|
"io"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// Ast AST操作接口
|
|
type Ast interface {
|
|
Parse(filename string, writer io.Writer) (file *ast.File, err error)
|
|
Rollback(file *ast.File) error
|
|
Injection(file *ast.File) error
|
|
Format(filename string, writer io.Writer, file *ast.File) error
|
|
}
|
|
|
|
// Base 基础AST操作
|
|
type Base struct {
|
|
AutoCodeRoot string
|
|
AutoCodeServer string
|
|
}
|
|
|
|
func (a *Base) Parse(filename string, writer io.Writer) (file *ast.File, err error) {
|
|
fileSet := token.NewFileSet()
|
|
if writer != nil {
|
|
file, err = parser.ParseFile(fileSet, filename, nil, parser.ParseComments)
|
|
} else {
|
|
file, err = parser.ParseFile(fileSet, filename, writer, parser.ParseComments)
|
|
}
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "[filepath:%s]打开/解析文件失败!", filename)
|
|
}
|
|
return file, nil
|
|
}
|
|
|
|
func (a *Base) Rollback(file *ast.File) error {
|
|
return nil
|
|
}
|
|
|
|
func (a *Base) Injection(file *ast.File) error {
|
|
return nil
|
|
}
|
|
|
|
func (a *Base) Format(filename string, writer io.Writer, file *ast.File) error {
|
|
fileSet := token.NewFileSet()
|
|
if writer == nil {
|
|
open, err := os.OpenFile(filename, os.O_WRONLY|os.O_TRUNC, 0666)
|
|
defer open.Close()
|
|
if err != nil {
|
|
return errors.Wrapf(err, "[filepath:%s]打开文件失败!", filename)
|
|
}
|
|
writer = open
|
|
}
|
|
err := format.Node(writer, fileSet, file)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "[filepath:%s]注入失败!", filename)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// RelativePath 绝对路径转相对路径
|
|
func (a *Base) RelativePath(filePath string) string {
|
|
server := filepath.Join(a.AutoCodeRoot, a.AutoCodeServer)
|
|
hasServer := strings.Index(filePath, server)
|
|
if hasServer != -1 {
|
|
filePath = strings.TrimPrefix(filePath, server)
|
|
keys := strings.Split(filePath, string(filepath.Separator))
|
|
filePath = path.Join(keys...)
|
|
}
|
|
return filePath
|
|
}
|
|
|
|
// AbsolutePath 相对路径转绝对路径
|
|
func (a *Base) AbsolutePath(filePath string) string {
|
|
server := filepath.Join(a.AutoCodeRoot, a.AutoCodeServer)
|
|
keys := strings.Split(filePath, "/")
|
|
filePath = filepath.Join(keys...)
|
|
filePath = filepath.Join(server, filePath)
|
|
return filePath
|
|
}
|