github.com/JarrahG/buffalocli@v0.0.0-20230801092127-b85bfd5d395a/internal/genny/fix/deprecations.go

package fix

import (
	"bytes"
	"fmt"
	"go/ast"
	"go/format"
	"go/parser"
	"go/printer"
	"go/token"
	"io"
	"os"
	"path/filepath"
	"regexp"
	"strings"

	"github.com/gobuffalo/genny/v2"
	"golang.org/x/tools/go/ast/astutil"
)

// DeprecationsCheck will either log, or fix, deprecated items in the application
func DeprecationsCheck(opts *Options) genny.RunFn {
	return func(r *genny.Runner) error {
		fmt.Println("~~~ Checking for deprecations ~~~")
		f, err := r.FindFile("cmd/app/main.go")
		if err != nil {
			return err
		}

		if strings.Contains(f.String(), "app.Start") {
			opts.warnings = append(opts.warnings, "app.Start has been removed in v0.11.0. Use app.Serve Instead. [main.go]")
		}

		err = walkDisk(r.Disk, "actions", packrMigrateFun(r, opts))
		if err != nil {
			return err
		}

		return walkDisk(r.Disk, ".", updateSuiteFun(r, opts))
	}
}

func packrMigrateFun(r *genny.Runner, opts *Options) func(path string, info os.FileInfo, err error) error {
	return func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}

		if info.IsDir() {
			return nil
		}

		if filepath.Ext(path) != ".go" {
			return nil
		}

		f, err := r.FindFile(path)
		if err != nil {
			return err
		}
		b, err := io.ReadAll(f)
		if err != nil {
			return err
		}

		if bytes.Contains(b, []byte("AssetsBox")) {
			b, err = addImport(path, b, fmt.Sprintf("%s/public", opts.App.PackagePkg))
			if err != nil {
				return err
			}

			rx := regexp.MustCompile("AssetsBox:.*,")
			b = rx.ReplaceAll(b, []byte("AssetsFS: public.FS(),"))

			rx = regexp.MustCompile(`(?m)^.*assetsBox.*=.*packr\.New.*$`)
			b = rx.ReplaceAll(b, []byte(""))
		}

		if bytes.Contains(b, []byte("TemplatesBox")) {
			b, err = addImport(path, b, fmt.Sprintf("%s/templates", opts.App.PackagePkg))
			if err != nil {
				return err
			}

			rx := regexp.MustCompile("TemplatesBox:.*,")
			b = rx.ReplaceAll(b, []byte("TemplatesFS: templates.FS(),"))
		}

		rx := regexp.MustCompile(`i18n\.New\(packr\.New\(.*\),(?P<Lang>.*)\)`)
		if rx.Match(b) {
			b, err = addImport(path, b, fmt.Sprintf("%s/locales", opts.App.PackagePkg))
			if err != nil {
				return err
			}

			match := rx.FindSubmatch(b)
			new := fmt.Sprintf("i18n.New(locales.FS(),%s)", match[1])
			b = rx.ReplaceAll(b, []byte(new))
		}

		if bytes.Contains(b, []byte("app.ServeFiles(\"/\", assetsBox)")) {
			b, err = addImport(path, b, fmt.Sprintf("%s/public", opts.App.PackagePkg))
			if err != nil {
				return err
			}

			b, err = addImport(path, b, "net/http")
			if err != nil {
				return err
			}

			rx := regexp.MustCompile(`app\.ServeFiles\(.*assetsBox\)`)
			b = rx.ReplaceAll(b, []byte("app.ServeFiles(\"/\", http.FS(public.FS()))"))
		}

		b, err = format.Source(b)
		if err != nil {
			return err
		}

		if _, err := f.Write(b); err != nil {
			return err
		}
		return r.File(f)
	}
}

func updateSuiteFun(r *genny.Runner, opts *Options) func(path string, info os.FileInfo, err error) error {
	return func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}

		if info.IsDir() {
			return nil
		}

		if filepath.Ext(path) != ".go" {
			return nil
		}

		f, err := r.FindFile(path)
		if err != nil {
			return err
		}
		b, err := io.ReadAll(f)
		if err != nil {
			return err
		}

		rx := regexp.MustCompile(`suite\.NewModelWithFixtures\(packr\.New\(".*", (?P<path>.*)\)\)`)
		if rx.Match(b) {
			b, err = addImport(path, b, "os")
			if err != nil {
				return err
			}

			match := rx.FindSubmatch(b)
			new := fmt.Sprintf("suite.NewModelWithFixtures(os.DirFS(%s))", match[1])
			b = rx.ReplaceAll(b, []byte(new))
		}

		rx = regexp.MustCompile(`suite\.NewActionWithFixtures\((?P<app>.*), packr\.New\(".*", (?P<path>.*)\)\)`)
		if rx.Match(b) {
			b, err = addImport(path, b, "os")
			if err != nil {
				return err
			}

			match := rx.FindSubmatch(b)
			new := fmt.Sprintf("suite.NewActionWithFixtures(%s, os.DirFS(%s))", match[1], match[2])
			b = rx.ReplaceAll(b, []byte(new))
		}

		b, err = format.Source(b)
		if err != nil {
			return err
		}

		if _, err := f.Write(b); err != nil {
			return err
		}
		return r.File(f)
	}
}

func addImport(path string, src []byte, value string) ([]byte, error) {
	fset := token.NewFileSet()
	f, err := parser.ParseFile(fset, path, src, parser.ParseComments)
	if err != nil {
		return nil, err
	}

	astutil.AddImport(fset, f, value)
	ast.SortImports(fset, f)

	bb := &bytes.Buffer{}
	err = printer.Fprint(bb, fset, f)
	return bb.Bytes(), err
}