This commit is contained in:
Liang Ding 2015-01-03 12:39:00 +08:00
parent 0d51c6c55d
commit 287fc618a0
1 changed files with 47 additions and 16 deletions

View File

@ -15,19 +15,21 @@
package file package file
import ( import (
"bytes"
"encoding/json" "encoding/json"
"go/ast" "go/ast"
"go/parser" "go/parser"
"go/token" "go/token"
"net/http" "net/http"
"strings"
"github.com/b3log/wide/util" "github.com/b3log/wide/util"
) )
type element struct { type element struct {
Name string Name string
Pos token.Pos Line int
End token.Pos Ch int
} }
// GetOutline gets outfile of a go file. // GetOutline gets outfile of a go file.
@ -56,14 +58,14 @@ func GetOutline(w http.ResponseWriter, r *http.Request) {
//ast.Print(fset, f) //ast.Print(fset, f)
data["package"] = &element{Name: f.Name.Name, Pos: f.Name.Pos(), End: f.Name.End()} line, ch := getCursor(code, int(f.Name.Pos()))
data["package"] = &element{Name: f.Name.Name, Line: line, Ch: ch}
imports := []*element{} imports := []*element{}
for _, astImport := range f.Imports { for _, astImport := range f.Imports {
line, ch := getCursor(code, int(astImport.Path.Pos()))
impt := &element{Name: astImport.Path.Value, Pos: astImport.Path.Pos(), End: astImport.Path.End()} imports = append(imports, &element{Name: astImport.Path.Value, Line: line, Ch: ch})
imports = append(imports, impt)
} }
data["imports"] = imports data["imports"] = imports
@ -77,9 +79,9 @@ func GetOutline(w http.ResponseWriter, r *http.Request) {
case *ast.FuncDecl: case *ast.FuncDecl:
funcDecl := decl.(*ast.FuncDecl) funcDecl := decl.(*ast.FuncDecl)
decl := &element{Name: funcDecl.Name.Name, Pos: funcDecl.Name.Pos(), End: funcDecl.Name.End()} line, ch := getCursor(code, int(funcDecl.Name.Pos()))
funcDecls = append(funcDecls, decl) funcDecls = append(funcDecls, &element{Name: funcDecl.Name.Name, Line: line, Ch: ch})
case *ast.GenDecl: case *ast.GenDecl:
genDecl := decl.(*ast.GenDecl) genDecl := decl.(*ast.GenDecl)
@ -88,27 +90,31 @@ func GetOutline(w http.ResponseWriter, r *http.Request) {
switch genDecl.Tok { switch genDecl.Tok {
case token.VAR: case token.VAR:
variableSpec := spec.(*ast.ValueSpec) variableSpec := spec.(*ast.ValueSpec)
decl := &element{Name: variableSpec.Names[0].Name, Pos: variableSpec.Pos(), End: variableSpec.End()}
varDecls = append(varDecls, decl) for _, varName := range variableSpec.Names {
line, ch := getCursor(code, int(varName.Pos()))
varDecls = append(varDecls, &element{Name: varName.Name, Line: line, Ch: ch})
}
case token.TYPE: case token.TYPE:
typeSpec := spec.(*ast.TypeSpec) typeSpec := spec.(*ast.TypeSpec)
decl := &element{Name: typeSpec.Name.Name, Pos: typeSpec.Name.Pos(), End: typeSpec.Name.End()}
switch typeSpec.Type.(type) { switch typeSpec.Type.(type) {
case *ast.StructType: case *ast.StructType:
structDecls = append(structDecls, decl) structDecls = append(structDecls, &element{Name: typeSpec.Name.Name, Line: line, Ch: ch})
case *ast.InterfaceType: case *ast.InterfaceType:
interfaceDecls = append(interfaceDecls, decl) interfaceDecls = append(interfaceDecls, &element{Name: typeSpec.Name.Name, Line: line, Ch: ch})
} }
case token.CONST: case token.CONST:
constSpec := spec.(*ast.ValueSpec) constSpec := spec.(*ast.ValueSpec)
decl := &element{Name: constSpec.Names[0].Name, Pos: constSpec.Pos(), End: constSpec.End()}
constDecls = append(constDecls, decl) for _, constName := range constSpec.Names {
line, ch := getCursor(code, int(constName.Pos()))
constDecls = append(constDecls, &element{Name: constName.Name, Line: line, Ch: ch})
}
} }
} }
} }
} }
@ -118,3 +124,28 @@ func GetOutline(w http.ResponseWriter, r *http.Request) {
data["structDecls"] = structDecls data["structDecls"] = structDecls
data["interfaceDecls"] = interfaceDecls data["interfaceDecls"] = interfaceDecls
} }
// getCursor calculates the cursor position (line, ch) by the specified offset.
//
// line is the line number, starts with 0 that means the first line
// ch is the column number, starts with 0 that means the first column
func getCursor(code string, offset int) (line, ch int) {
code = code[:offset]
lines := strings.Split(code, "\n")
line = 0
for range lines {
line++
}
var buffer bytes.Buffer
runes := []rune(lines[line-1])
for _, r := range runes {
buffer.WriteString(string(r))
}
ch = len(buffer.String())
return line - 1, ch - 1
}