Goの標準パッケージではじめる静的解析入門③実践編

$ mkdir example
$ touch example/example.go 

$ touch main.go

package example

import (
  "fmt"
  "time"
)

func example() {
  fmt.Println("Now :", time.Now())
}

package main

import (
  "fmt"
  "go/parser"
  "go/token"
)

func main() {
  fset := token.NewFileSet()
  // ファイルをパースしてを抽象構文木(AST)に変換する
  f, _ := parser.ParseFile(fset, "./example/example.go", nil, parser.Mode(0)))
}

const (
    PackageClauseOnly Mode             = 1 << iota // stop parsing after package clause
    ImportsOnly                                    // stop parsing after import declarations
    ParseComments                                  // parse comments and add them to AST
    Trace                                          // print a trace of parsed productions
    DeclarationErrors                              // report declaration errors
    SpuriousErrors                                 // same as AllErrors, for backward-compatibility
    AllErrors         = SpuriousErrors             // report all errors (not just the first 10 on different lines)
)

func main() {
  fset := token.NewFileSet()
  // package名とimportされているパッケージが取得できれば良いのでImportsOnlyモードにする
  f, _ := parser.ParseFile(fset, "./example/example.go", nil, parser.ImportsOnly)

  // package名を出力する
  fmt.Println(f.Name)

  // importされているパッケージを出力する
  for _, s := range f.Imports {
    fmt.Println(s.Path.Value)
  }
}

"fmt"
"time"

type File struct {
    Doc        *CommentGroup   // associated documentation; or nil
    Package    token.Pos       // position of "package" keyword
    Name       *Ident          // package name
    Decls      []Decl          // top-level declarations; or nil
    Scope      *Scope          // package scope (this file only)
    Imports    []*ImportSpec   // imports in this file
    Unresolved []*Ident        // unresolved identifiers in this file
    Comments   []*CommentGroup // list of all comments in the source file
}

type Position struct {
    Filename string // filename, if any
    Offset   int    // offset, starting at 0
    Line     int    // line number, starting at 1
    Column   int    // column number, starting at 1 (byte count)
}

func main() {
  fset := token.NewFileSet()
  f, _ := parser.ParseFile(fset, "./example/example.go", nil, parser.ImportsOnly)

  // importされているパッケージのファイル名/行数/カラム数と、パッケージ名を出力する
  // token.Pos()はノードがソースコード上に占める位置を指す
  for _, s := range f.Imports {
    fmt.Println(fset.Position(s.Pos()), s.Path.Value)
  }
}

./example/example.go:4:2 "fmt"
./example/example.go:5:2 "time"

func main() {
  fset := token.NewFileSet()
  f, _ := parser.ParseFile(fset, "./example/example.go", nil, 0)

  ast.Inspect(f, func(n ast.Node) bool {
    var s string
    // 型によって処理を分岐する
    switch x := n.(type) {
    case *ast.BasicLit: // リテラル
      s = x.Value
    case *ast.Ident: // 識別子名
      s = x.Name 
    }
    if s != "" {
      fmt.Println(fset.Position(n.Pos()), s)
    }
    return true
  })
}

./example/example.go:1:9 example
./example/example.go:4:2 "fmt"
./example/example.go:5:2 "time"
./example/example.go:8:6 example
./example/example.go:9:2 fmt
./example/example.go:9:6 Println
./example/example.go:9:14 "Now :"
./example/example.go:9:23 time
./example/example.go:9:28 Now