initial commit

tlparser/code.go

@@ -0,0 +1,74 @@
+package tlparser
+
+import (
+	"bufio"
+	"fmt"
+	"io"
+	"strings"
+)
+
+func ParseCode(reader io.Reader, schema *Schema) error {
+	var prevLine string
+	var curLine string
+
+	userMethods := map[string]bool{}
+	botMethods := map[string]bool{}
+
+	scanner := bufio.NewScanner(reader)
+	for scanner.Scan() {
+		prevLine = curLine
+		curLine = scanner.Text()
+
+		if strings.Contains(curLine, "CHECK_IS_USER();") {
+			fields := strings.Fields(prevLine)
+			for _, field := range fields {
+				var methodName string
+				n, err := fmt.Sscanf(field, "td_api::%s", &methodName)
+				if err == nil && n > 0 {
+					userMethods[methodName] = true
+				}
+			}
+		}
+
+		if strings.Contains(curLine, "CHECK_IS_BOT();") {
+			fields := strings.Fields(prevLine)
+			for _, field := range fields {
+				var methodName string
+				n, err := fmt.Sscanf(field, "td_api::%s", &methodName)
+				if err == nil && n > 0 {
+					botMethods[methodName] = true
+				}
+			}
+		}
+	}
+
+	err := scanner.Err()
+	if err != nil {
+		return err
+	}
+
+	var ok bool
+
+	for index, _ := range schema.Functions {
+		hasType := false
+		_, ok = userMethods[schema.Functions[index].Name]
+		if ok {
+			schema.Functions[index].Type = FUNCTION_TYPE_USER
+			hasType = true
+		}
+
+		_, ok = botMethods[schema.Functions[index].Name]
+		if ok {
+			schema.Functions[index].Type = FUNCTION_TYPE_BOT
+			hasType = true
+		}
+
+		if !hasType {
+			schema.Functions[index].Type = FUNCTION_TYPE_COMMON
+		}
+
+		ok = false
+	}
+
+	return nil
+}