feat: parses flask

Author: Hoblovski

lang/collect/collect.go

@@ -108,6 +108,22 @@ func NewCollector(repo string, cli *LSPClient) *Collector {
 }
 
 func (c *Collector) Collect(ctx context.Context) error {
+	// Example code to configure the LSP client
+	// if c.Language == uniast.Python {
+	// 	conf := map[string]interface{}{
+	// 		"settings": map[string]interface{}{
+	// 			"pylsp": map[string]interface{}{
+	// 				"plugins": map[string]interface{}{
+	// 					"jedi_definition": map[string]interface{}{
+	// 						"follow_builtin_definitions": false,
+	// 					},
+	// 				},
+	// 			},
+	// 		},
+	// 	}
+	// 	c.cli.Notify(ctx, "workspace/didChangeConfiguration", conf)
+	// }
+
 	excludes := make([]string, len(c.Excludes))
 	for i, e := range c.Excludes {
 		if !filepath.IsAbs(e) {

lang/collect/export.go

@@ -320,7 +320,7 @@ func (c *Collector) exportSymbol(repo *uniast.Repository, symbol *DocumentSymbol
 				case lsp.SKStruct, lsp.SKTypeParameter, lsp.SKInterface, lsp.SKEnum, lsp.SKClass:
 					obj.SubStruct = append(obj.SubStruct, uniast.NewDependency(*depid, c.fileLine(dep.Location)))
 				default:
-					log.Error("dep symbol %s not collected for \n", dep.Symbol, id)
+					log.Error("dep symbol %s not collected for %v\n", dep.Symbol, id)
 				}
 			}
 		}

lang/lsp/client.go

@@ -225,7 +225,7 @@ func (rwc rwc) Close() error {
 
 // start a LSP process and return its io
 func startLSPSever(path string) (io.ReadWriteCloser, error) {
-	// Launch rust-analyzer
+	// Launch LSP server
 	cmd := exec.Command(path)
 
 	stdin, err := cmd.StdinPipe()

lang/lsp/handler.go

@@ -93,9 +93,9 @@ loop:
 
 func (h *lspHandler) Handle(ctx context.Context, conn *jsonrpc2.Conn, req *jsonrpc2.Request) {
 	// This method will be called for both requests and notifications
-	log.Info("handle method: %s\n", req.Method)
+	log.Debug("handle method: %s\n", req.Method)
 	if req.Params != nil {
-		log.Info("param: %s\n", string(*req.Params))
+		log.Debug("param: %s\n", string(*req.Params))
 	}
 	if req.Notif {
 		// This is a notification

lang/python/lib.go

@@ -21,7 +21,7 @@ import (
 	"github.com/cloudwego/abcoder/lang/utils"
 )
 
-const MaxWaitDuration = 5 * time.Second
+const MaxWaitDuration = 2 * time.Second
 
 func GetDefaultLSP() (lang uniast.Language, name string) {
 	// needs to use the custom pylsp (see commit message)

lang/python/spec.go

@@ -43,33 +43,20 @@ func (c *PythonSpec) WorkSpace(root string) (map[string]string, error) {
 		return nil, err
 	}
 
-	num_projfiles := 0
-	scanner := func(path string, info os.FileInfo, err error) error {
-		if err != nil {
-			return err
-		}
-		base := filepath.Base(path)
-		if base == "pyproject.toml" {
-			num_projfiles++
-			if num_projfiles > 1 {
-				panic("multiple pyproject.toml files found")
-			}
-			// it's hard to infer the name or package from pyproject.toml
-		}
-		return nil
-	}
-	if err := filepath.Walk(root, scanner); err != nil {
-		return nil, err
-	}
+	// TODO: maybe infer from pyproject.toml?
+	//   should ignore {tests,examples}/**/pyproject.toml
 
 	// XXX ad-hoc way
 	if strings.Contains(c.repo, "astropy") {
 		panic("TODO")
+	} else if strings.Contains(c.repo, "flask") {
+		c.topModulePath = absPath + "/src"
+		c.topModuleName = "flask"
 	} else {
 		c.topModulePath = absPath
 		c.topModuleName = "current"
-		rets[c.topModuleName] = c.topModulePath
 	}
+	rets[c.topModuleName] = c.topModulePath
 	return rets, nil
 }