package gen

import (
	"os"
	"strings"
)

// HandlePbDir 处理pb文件夹
func (g *Generator) HandlePbDir() error {
	err := os.RemoveAll(g.PbPath)
	if err != nil {
		return err
	}
	// 移动新生成的pb文件到pb目录下
	err = os.Rename(g.createPbPath, g.PbPath)
	if err != nil {
		return err
	}
	err = os.RemoveAll(g.GoModule)
	if err != nil {
		return err
	}
	dirs := strings.SplitN(g.GoModule, "/", 2)
	if len(dirs) > 0 {
		err = os.RemoveAll(dirs[0])
		if err != nil {
			return err
		}
	}
	return nil
}