フラグメントを解析してトップレベルの定義をすべて見つけよう

Question

私が書いているサービスはコードスニペットを受け取って処理しますが、スニペットは補完プログラムまたはフラグメントのいずれかになります。フラグメントの場合は、それを囲むmain関数。例えば、スニペット：

var v int 
v = 3 
fmt.Println(v) 

フラグメントとして分類され、それにmainを追加しなければならない。

func main() { 
    var v int 
    v = 3 
    fmt.Println(v) 
} 

スニペットの場合：

package main 
import "fmt" 
func main() { 
    fmt.Println("hello") 
} 

次いで全く変更があってはなりません完了しました。

は私が今やっている方法は、スニペットに反するパーサーを実行している：

var fset *token.FileSet 
file, err := parser.ParseFile(fset, "stdin", code, 0) 
if err != nil { 
     // add function 
     code = fmt.Sprintf("func main() {\n%s\n}", code) 
     // ... 
} 

これは、上記第一のスニペットのために働くのフラグメントは、いくつかの他の宣言の後にmain機能を持っている場合しかし、それは失敗し、例えば

type S struct { 
    a int 
} 
func main() { 
    fmt.Println("foo") 
} 

私はまた、ParseFileによって返さfileに見てみてくださいDeclsをチェックし、それはそれは第一にエラーの後で解析を停止します見えますので、Declsはこのケースでnilです。だから私の質問は、これを処理する堅牢な方法ですか？

PS。とにかく処理されたコードをgolang.org/x/tools/importsに送っているので、package句と必要なインポートのインクルードは関係ありません。

Answer 1

ファイル内の読み込み後（最も可能性が高い）は、func main(){の文字列検索を実行することです。

まだフォーマットされていない場合は、空白を含む正規表現に変更する必要があるかもしれませんが、かなり単純です。