package mysql

import (
	"bytes"
	"database/sql"
	"fmt"

	"gitee.com/ideadawn/dbvm/manager"
	"github.com/VividCortex/mysqlerr"
	driver "github.com/go-sql-driver/mysql"
)

// Deploy 部署指定版本
func (m *MySQL) Deploy(plan *manager.Plan) error {
	if m == nil || m.db == nil {
		return errConnection
	}
	if m.table == `` {
		return errTableNotInit
	}

	parser := &sqlParser{
		file: plan.Deploy,
	}

	parser.parseSqlBlocks()
	if parser.err != nil {
		return parser
	}
	if len(parser.blocks) == 0 {
		return errDeployNothing
	}

	parser.blocks = append(parser.blocks, &sqlBlock{
		ignores: []uint16{
			mysqlerr.ER_DUP_ENTRY,
		},
		line: 999999,
		sqlArr: [][]byte{
			[]byte(fmt.Sprintf(
				"INSERT INTO `%s` (`name`,`time`,`status`) VALUES ('%s', '%d', '%d');",
				m.table,
				plan.Name,
				plan.Time.Unix(),
				StatusDeployed,
			)),
		},
	})

	var tx *sql.Tx
	var err error
	for _, block := range parser.blocks {
		for tries := 0; tries < retry; tries++ {
			if !block.noTrans {
				tx, err = m.db.Begin()
				if err != nil {
					fmt.Println(`Deploy BEGIN:`, err)
					continue
				}
			} else {
				tx = nil
			}

			exec := string(bytes.Join(block.sqlArr, myCnf.newLine))
			if tx == nil {
				_, err = m.db.Exec(exec)
			} else {
				_, err = tx.Exec(exec)
			}
			if err == nil {
				if block.noTrans {
					break
				}
				err = tx.Commit()
				if err == nil {
					break
				}
				fmt.Println(`Deploy COMMIT:`, err)
				continue
			} else {
				fmt.Println(`Deploy`, plan.Deploy, `on line`, block.line, `:`, err)
				fmt.Println(exec)
				fmt.Println("")
			}

			if tx != nil {
				_ = tx.Rollback()
			}

			//是否语法错误，是否可以忽略
			myerr, ok := err.(*driver.MySQLError)
			if ok {
				if inUint16Array(block.ignores, myerr.Number) {
					err = nil
					break
				}
				return err
			}
		}

		if err != nil {
			break
		}
	}

	return err
}