mirror of
https://github.com/taigrr/wails.git
synced 2026-04-13 18:38:11 -07:00
132 lines
2.9 KiB
Go
132 lines
2.9 KiB
Go
package parser
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"text/template"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/wailsapp/wails/v2/internal/fs"
|
|
)
|
|
|
|
// GenerateWailsFrontendPackage will generate a Javascript/Typescript
|
|
// package in `<project>/frontend/wails` that defines which methods
|
|
// and structs are bound to your frontend
|
|
func GenerateWailsFrontendPackage() error {
|
|
|
|
dir, err := os.Getwd()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
p := NewParser()
|
|
|
|
err = p.ParseProject(dir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = p.generateModule()
|
|
|
|
return err
|
|
}
|
|
|
|
func (p *Parser) generateModule() error {
|
|
|
|
moduleDir, err := createBackendJSDirectory()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, pkg := range p.packages {
|
|
|
|
// Calculate directory
|
|
dir := filepath.Join(moduleDir, pkg.gopackage.Name)
|
|
|
|
// Create the directory if it doesn't exist
|
|
fs.Mkdir(dir)
|
|
|
|
err := generatePackage(pkg, dir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func createBackendJSDirectory() (string, error) {
|
|
|
|
// Calculate the package directory
|
|
// Note this is *always* called from the project directory
|
|
// so using paths relative to CWD is fine
|
|
dir, err := fs.RelativeToCwd("./frontend/wails")
|
|
if err != nil {
|
|
return "", errors.Wrap(err, "Error creating wails js directory")
|
|
}
|
|
|
|
// Remove directory if it exists - REGENERATION!
|
|
err = os.RemoveAll(dir)
|
|
if err != nil {
|
|
return "", errors.Wrap(err, "Error removing module directory")
|
|
}
|
|
|
|
// Make the directory
|
|
err = fs.Mkdir(dir)
|
|
|
|
return dir, err
|
|
}
|
|
|
|
func generatePackage(pkg *Package, moduledir string) error {
|
|
|
|
// Get path to local file
|
|
typescriptTemplateFile := fs.RelativePath("./package.d.template")
|
|
|
|
// Load typescript template
|
|
typescriptTemplateData := fs.MustLoadString(typescriptTemplateFile)
|
|
typescriptTemplate, err := template.New("typescript").Parse(typescriptTemplateData)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Error creating template")
|
|
}
|
|
|
|
// Execute javascript template
|
|
var buffer bytes.Buffer
|
|
err = typescriptTemplate.Execute(&buffer, pkg)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Error generating code")
|
|
}
|
|
|
|
// Save typescript file
|
|
err = ioutil.WriteFile(filepath.Join(moduledir, "index.d.ts"), buffer.Bytes(), 0755)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Error writing backend package file")
|
|
}
|
|
|
|
// Get path to local file
|
|
javascriptTemplateFile := fs.RelativePath("./package.template")
|
|
|
|
// Load javascript template
|
|
javascriptTemplateData := fs.MustLoadString(javascriptTemplateFile)
|
|
javascriptTemplate, err := template.New("javascript").Parse(javascriptTemplateData)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Error creating template")
|
|
}
|
|
|
|
// Reset the buffer
|
|
buffer.Reset()
|
|
|
|
err = javascriptTemplate.Execute(&buffer, pkg)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Error generating code")
|
|
}
|
|
|
|
// Save javascript file
|
|
err = ioutil.WriteFile(filepath.Join(moduledir, "index.js"), buffer.Bytes(), 0755)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Error writing backend package file")
|
|
}
|
|
|
|
return nil
|
|
}
|