go-tdlib/cmd/generate-json.go

package main

import (
	"bufio"
	"encoding/json"
	"flag"
	"log"
	"net/http"
	"os"
	"path/filepath"
	"strings"

	"git.zio.sh/astra/go-tdlib/tlparser"
)

func main() {
	var version string
	var outputFilePath string

	flag.StringVar(&version, "version", "", "TDLib version")
	flag.StringVar(&outputFilePath, "output", "./td_api.json", "json schema file")

	flag.Parse()

	resp, err := http.Get("https://raw.githubusercontent.com/tdlib/td/" + version + "/td/generate/scheme/td_api.tl")
	if err != nil {
		log.Fatalf("http.Get error: %s", err)
		return
	}
	defer resp.Body.Close()

	schema, err := tlparser.Parse(resp.Body)
	if err != nil {
		log.Fatalf("schema parse error: %s", err)
		return
	}

	resp, err = http.Get("https://raw.githubusercontent.com/tdlib/td/" + version + "/td/telegram/Td.cpp")
	if err != nil {
		log.Fatalf("http.Get error: %s", err)
		return
	}
	defer resp.Body.Close()

	err = tlparser.ParseCode(resp.Body, schema)
	if err != nil {
		log.Fatalf("parse code error: %s", err)
		return
	}

	err = os.MkdirAll(filepath.Dir(outputFilePath), os.ModePerm)
	if err != nil {
		log.Fatalf("make dir error: %s", filepath.Dir(outputFilePath))
	}

	file, err := os.OpenFile(outputFilePath, os.O_CREATE|os.O_RDWR|os.O_TRUNC, os.ModePerm)
	if err != nil {
		log.Fatalf("open file error: %s", err)
		return
	}

	data, err := json.MarshalIndent(schema, "", strings.Repeat(" ", 4))
	if err != nil {
		log.Fatalf("json marshal error: %s", err)
		return
	}
	bufio.NewWriter(file).Write(data)
}