endpoint-case-alternative.go

package service

import (
	"fmt"
	serverDomain "github.com/aaronchen2k/deeptest/cmd/server/v1/domain"
	agentExec "github.com/aaronchen2k/deeptest/internal/agent/exec"
	valueUtils "github.com/aaronchen2k/deeptest/internal/agent/exec/utils/value"
	"github.com/aaronchen2k/deeptest/internal/pkg/consts"
	"github.com/aaronchen2k/deeptest/internal/pkg/domain"
	"github.com/aaronchen2k/deeptest/internal/pkg/helper/cases"
	"github.com/aaronchen2k/deeptest/internal/server/modules/model"
	"github.com/aaronchen2k/deeptest/internal/server/modules/repo"
	logUtils "github.com/aaronchen2k/deeptest/pkg/lib/log"
	_stringUtils "github.com/aaronchen2k/deeptest/pkg/lib/string"
	"github.com/kataras/iris/v12"
	"github.com/tidwall/sjson"
	"strings"
)

type EndpointCaseAlternativeService struct {
	EndpointCaseAlternativeRepo *repo.EndpointCaseAlternativeRepo `inject:""`
	EndpointCaseRepo            *repo.EndpointCaseRepo            `inject:""`

	EndpointInterfaceRepo *repo.EndpointInterfaceRepo `inject:""`
	ServeServerRepo       *repo.ServeServerRepo       `inject:""`
	DebugInterfaceRepo    *repo.DebugInterfaceRepo    `inject:""`
	EndpointRepo          *repo.EndpointRepo          `inject:""`
	CategoryRepo          *repo.CategoryRepo          `inject:""`
	ConditionRepo         *repo.ConditionRepo         `inject:""`

	EndpointCaseService      *EndpointCaseService      `inject:""`
	EndpointService          *EndpointService          `inject:""`
	DebugInterfaceService    *DebugInterfaceService    `inject:""`
	EndpointMockParamService *EndpointMockParamService `inject:""`
	DebugSceneService        *DebugSceneService        `inject:""`

	EnvironmentRepo *repo.EnvironmentRepo `inject:""`
	SceneService    *SceneService         `inject:""`
}

func (s *EndpointCaseAlternativeService) LoadAlternative(tenantId consts.TenantId, endpointId uint, method consts.HttpMethod) (
	root casesHelper.AlternativeCase, err error) {

	root.Title = "备选用例"
	root.Category = consts.AlternativeCaseRoot
	root.Key = _stringUtils.Uuid()
	root.Slots = iris.Map{"icon": "icon"}
	root.IsDir = true

	//casePo, _ := s.EndpointCaseRepo.Get(baseId)

	_, endpointInterfaceId := s.EndpointInterfaceRepo.GetByMethod(tenantId, endpointId, method)
	if endpointInterfaceId == 0 {
		return
	}

	endpointInterface, _ := s.EndpointInterfaceRepo.Get(tenantId, endpointInterfaceId)
	endpoint, err := s.EndpointRepo.GetWithInterface(tenantId, endpointInterface.EndpointId, "v0.1.0")
	if err != nil {
		return
	}

	// get spec
	doc3 := s.EndpointService.Yaml(tenantId, endpoint)

	// TEST:
	//pth := "/Users/aaron/rd/project/gudi/deeptest/xdoc/openapi/openapi3/test2.yaml"
	//loader := &openapi3.Loader{Context: context.Background(), IsExternalRefsAllowed: true}
	//doc3, err := loader.LoadFromFile(pth)
	//if err != nil || doc3 == nil {
	//	return
	//}

	apiPathItem, _ := casesHelper.GetApiPathItem(doc3)

	apiOperation, err := casesHelper.GetApiOperation(method, apiPathItem)
	if err != nil || apiOperation == nil {
		return
	}

	pathParamsDir := casesHelper.LoadForPathParams(apiPathItem.Parameters)
	if len(pathParamsDir.Children) > 0 {
		root.Children = append(root.Children, pathParamsDir)
	}

	queryParamsDir := casesHelper.LoadForQueryParams(apiOperation.Parameters)
	if len(queryParamsDir.Children) > 0 {
		root.Children = append(root.Children, queryParamsDir)
	}

	headerDir := casesHelper.LoadForHeaders(apiOperation.Parameters)
	if len(headerDir.Children) > 0 {
		root.Children = append(root.Children, headerDir)
	}

	bodyDir := casesHelper.LoadForBody(apiOperation.RequestBody, doc3)
	if len(bodyDir.Children) > 0 {
		root.Children = append(root.Children, bodyDir)
	}

	return
}

func (s *EndpointCaseAlternativeService) LoadFactor(tenantId consts.TenantId, caseId uint) (ret map[string]model.EndpointCaseAlternativeFactor, err error) {
	pos, err := s.EndpointCaseAlternativeRepo.LoadFactor(tenantId, caseId)

	ret = make(map[string]model.EndpointCaseAlternativeFactor)
	for _, po := range pos {
		ret[po.Path] = po
	}

	return
}

func (s *EndpointCaseAlternativeService) CreateBenchmarkCase(tenantId consts.TenantId, req serverDomain.EndpointCaseBenchmarkCreateReq) (
	po model.EndpointCase, err error) {

	if req.BaseCaseId > 0 {
		// clone
		//po, _ = s.EndpointCaseService.Copy(req.BaseCaseId, "alter-", req.CreateUserId, req.CreateUserName)
		po, err = s.EndpointCaseService.Get(tenantId, uint(req.BaseCaseId))
	} else if req.EndpointInterfaceId > 0 {
		// convert from endpoint interface define
		endpointInterface, _ := s.EndpointInterfaceRepo.Get(tenantId, req.EndpointInterfaceId)
		debugData, _ := s.DebugInterfaceService.GetDebugInterfaceByEndpointInterface(tenantId, req.EndpointInterfaceId, false)

		saveReq := serverDomain.EndpointCaseSaveReq{
			Name:           req.Name,
			Method:         debugData.Method,
			DebugData:      debugData,
			EndpointId:     endpointInterface.EndpointId,
			CreateUserId:   req.CreateUserId,
			CreateUserName: req.CreateUserName,
		}

		po, err = s.EndpointCaseService.SaveFromDebugInterface(tenantId, saveReq)
	}
	if err != nil {
		return
	}

	po.CaseType = consts.CaseBenchmark
	po.BaseCase = uint(req.BaseCaseId)

	s.EndpointCaseRepo.UpdateInfo(tenantId, po.ID, map[string]interface{}{
		"case_type": po.CaseType,
		"base_case": po.BaseCase,
	})

	if req.BaseCaseId > 0 {
		s.ConditionRepo.CloneAll(tenantId, po.DebugInterfaceId, 0, po.DebugInterfaceId, consts.CaseDebug, consts.CaseDebug, "false")

	}

	return
}

func (s *EndpointCaseAlternativeService) SaveFactor(tenantId consts.TenantId, req serverDomain.EndpointCaseFactorSaveReq) (err error) {
	err = s.EndpointCaseAlternativeRepo.SaveFactor(tenantId, req)

	return
}

func (s *EndpointCaseAlternativeService) SaveCase(tenantId consts.TenantId, req serverDomain.EndpointCaseAlternativeSaveReq) (count int, err error) {
	typ := req.Type
	if typ == "multi" {
		count, err = s.GenMultiCases(tenantId, req)
	} else if typ == "single" {
		count, err = s.GenSingleCase(tenantId, req)
	}

	return
}

func (s *EndpointCaseAlternativeService) GenMultiCases(tenantId consts.TenantId, req serverDomain.EndpointCaseAlternativeSaveReq) (count int, err error) {
	for _, val := range req.Values.Children {
		if !val.NeedExec {
			continue
		}

		if val.Category != consts.AlternativeCaseCase {
			req.Values = *val
			s.GenMultiCases(tenantId, req)

			continue
		}

		name := s.getName(val)

		newEndpointCase, err1 := s.EndpointCaseService.Copy(tenantId, req.BaseId, name, 0, 0, req.CreateUserId, req.CreateUserName, "true")
		if err1 != nil {
			err = err1
			return
		}

		s.EndpointCaseRepo.UpdateInfo(tenantId, newEndpointCase.ID, map[string]interface{}{
			"case_type": consts.CaseAlternative,
			"base_case": req.BaseId,
		})

		newDebugData, err1 := s.DebugInterfaceService.GetDebugDataFromDebugInterface(tenantId, newEndpointCase.DebugInterfaceId)
		if err1 != nil {
			err = err1
			return
		}

		fieldIn, fieldNameOrPath := s.getFieldProps(val.Path)
		if fieldIn == "" {
			logUtils.Info("failed to getFieldProps")
			continue
		}
		s.changeFieldProps(&newDebugData, fieldIn, fieldNameOrPath, val.Sample, val.FieldType)

		_, err = s.DebugInterfaceService.Update(tenantId, newDebugData, newDebugData.DebugInterfaceId)

		count += 1
	}

	return
}

func (s *EndpointCaseAlternativeService) GenSingleCase(tenantId consts.TenantId, req serverDomain.EndpointCaseAlternativeSaveReq) (count int, err error) {
	// copy new case
	newEndpointCase, err := s.EndpointCaseService.Copy(tenantId, req.BaseId, "多参数异常", 0, 0,
		req.CreateUserId, req.CreateUserName, "true")

	s.EndpointCaseRepo.UpdateInfo(tenantId, newEndpointCase.ID, map[string]interface{}{
		"case_type": consts.CaseAlternative,
		"base_case": req.BaseId,
	})

	// get new case's debug data
	newDebugData, err := s.DebugInterfaceService.GetDebugDataFromDebugInterface(tenantId, newEndpointCase.DebugInterfaceId)
	if err != nil {
		return
	}

	var validPaths []casesHelper.AlternativeCase
	for _, val := range req.Values.Children {
		s.getValidPaths(*val, &validPaths)
	}
	s.updateDebugData(tenantId, &newDebugData, validPaths)

	// update to db
	_, err = s.DebugInterfaceService.Update(tenantId, newDebugData, newDebugData.DebugInterfaceId)
	if err != nil {
		return
	}

	count = 1

	return
}

func (s *EndpointCaseAlternativeService) changeFieldProps(debugData *domain.DebugData,
	fieldIn, fieldNameOrPath string, sample interface{}, fieldType casesHelper.OasFieldType) {

	if fieldIn == "[query]" {
		s.changeParams(debugData.QueryParams, fieldNameOrPath, sample)
	} else if fieldIn == "[path]" {
		s.changeParams(debugData.PathParams, fieldNameOrPath, sample)
	} else if fieldIn == "[header]" {
		s.changeHeaders(debugData.Headers, fieldNameOrPath, sample)
	} else if fieldIn == "[body]/[application-json]" {
		s.changeBody(debugData, fieldNameOrPath, sample, fieldType)
	} else if fieldIn == "[body]/[multipart-form-data]" {
		s.changeForm(debugData, fieldNameOrPath, sample, fieldType)
	} else if fieldIn == "[body]/[application-x-www-form-urlencoded]" {
		s.changeFormUrlencoded(debugData, fieldNameOrPath, sample, fieldType)
	}

	return
}

func (s *EndpointCaseAlternativeService) changeParams(params *[]domain.Param,
	fieldNameOrPath string, sample interface{}) {

	found := false

	if params != nil {
		for index, param := range *params {
			if param.Name == fieldNameOrPath {
				(*params)[index].Value = fmt.Sprintf("%v", sample)

				found = true
				break
			}
		}
	}

	if !found {
		if params == nil {
			params = &[]domain.Param{}
		}
		*params = append(*params, domain.Param{
			Name:  fieldNameOrPath,
			Value: fmt.Sprintf("%v", sample),
		})
	}
}
func (s *EndpointCaseAlternativeService) changeHeaders(headers *[]domain.Header,
	fieldNameOrPath string, sample interface{}) {

	found := false
	if headers != nil {
		for index, header := range *headers {
			if header.Name == fieldNameOrPath {
				(*headers)[index].Value = fmt.Sprintf("%v", sample)

				found = true
				break
			}
		}
	}

	if !found {
		if headers == nil {
			headers = &[]domain.Header{}
		}
		*headers = append(*headers, domain.Header{
			Name:  fieldNameOrPath,
			Value: fmt.Sprintf("%v", sample),
		})
	}
}
func (s *EndpointCaseAlternativeService) changeForm(debugData *domain.DebugData,
	fieldNameOrPath string, sample interface{}, fieldType casesHelper.OasFieldType) {
	// form_item2/[format]

	fieldPath := s.getFieldPath(fieldNameOrPath)

	found := false
	if debugData.BodyFormData != nil {
		for index, item := range *debugData.BodyFormData {
			if item.Name == fieldPath {
				(*debugData.BodyFormData)[index].Value = valueUtils.InterfaceToStr(sample)
				found = true
			}
		}
	}

	if !found {
		if debugData.BodyFormData == nil {
			debugData.BodyFormData = &[]domain.BodyFormDataItem{}
		}
		*debugData.BodyFormData = append(*debugData.BodyFormData, domain.BodyFormDataItem{
			Name:  fieldPath,
			Value: valueUtils.InterfaceToStr(sample),
		})
	}
}
func (s *EndpointCaseAlternativeService) changeFormUrlencoded(debugData *domain.DebugData,
	fieldNameOrPath string, sample interface{}, fieldType casesHelper.OasFieldType) {

	fieldPath := s.getFieldPath(fieldNameOrPath)

	found := false

	if debugData.BodyFormUrlencoded != nil {
		for index, item := range *debugData.BodyFormUrlencoded {
			if item.Name == fieldPath {
				(*debugData.BodyFormUrlencoded)[index].Value = valueUtils.InterfaceToStr(sample)
				found = true
			}
		}
	}

	if !found {
		if debugData.BodyFormUrlencoded == nil {
			debugData.BodyFormUrlencoded = &[]domain.BodyFormUrlEncodedItem{}
		}
		*debugData.BodyFormUrlencoded = append(*debugData.BodyFormUrlencoded, domain.BodyFormUrlEncodedItem{
			Name:  fieldPath,
			Value: valueUtils.InterfaceToStr(sample),
		})
	}
}
func (s *EndpointCaseAlternativeService) changeBody(debugData *domain.DebugData,
	fieldNameOrPath string, sample interface{}, fieldType casesHelper.OasFieldType) {
	// id/[required]
	// id/[rule]/[min]
	// pet/age/[rule]/[min]
	// sons/[arr]/email/[format]

	fieldPath := s.getFieldPath(fieldNameOrPath)

	debugData.Body, _ = sjson.Set(debugData.Body, fieldPath, sample)
}

func (s *EndpointCaseAlternativeService) getFieldPath(pth string) (ret string) {
	retArr := []string{}

	arr := strings.Split(pth, "/")

	for _, item := range arr {
		if item == "[arr]" {
			retArr = append(retArr, "0") // change the first item
			continue
		}

		if strings.Index(item, "[") == 0 {
			break
		}

		retArr = append(retArr, item)
	}

	ret = strings.Join(retArr, ".")

	return
}

func (s *EndpointCaseAlternativeService) getFieldProps(pth string) (fieldIn string, fieldNameOrPath string) {
	arr := strings.Split(pth, "/")

	if len(arr) < 3 {
		return
	}

	if arr[0] == "[path]" {
		// [path]/path1/[format]
		fieldIn = arr[0]
		fieldNameOrPath = arr[1]

	} else if arr[0] == "[query]" {
		// [query/count/[required]
		// [query/count/[rule/[min]
		fieldIn = arr[0]
		fieldNameOrPath = arr[1]

	} else if arr[0] == "[header]" {
		// [header/header1/[format
		fieldIn = arr[0]
		fieldNameOrPath = arr[1]

	} else if arr[0] == "[body]" {
		// [body]/[application-json]/id/[required]
		// [body]/[application-json]/id/[rule]/[min]
		// [body]/[application-json]/pet/age/[rule]/[min]
		// [body]/[application-json]/sons/[arr]/email/[format]
		// [body]/[multipart-form-data]
		// [body]/[application-x-www-form-urlencoded]

		if !_stringUtils.StrInArr(arr[1],
			[]string{"[application-json]", "[multipart-form-data]", "[application-x-www-form-urlencoded]"}) {
			return
		} // ignore no-json and no-form request body

		fieldIn = strings.Join(arr[:2], "/")
		fieldNameOrPath = strings.Join(arr[2:], "/")
	}

	return
}

func (s *EndpointCaseAlternativeService) LoadCasesForExec(tenantId consts.TenantId, req agentExec.CasesExecReq) (
	ret agentExec.CaseExecProcessor, err error) {

	if req.ExecType == "multi" {
		s.loadMultiCasesData(tenantId, req.ExecObj, &ret, req.BaseCaseId, req.EnvironmentId)

	} else if req.ExecType == "single" {
		ret, _ = s.loadSingleCasesData(tenantId, req, req.BaseCaseId, req.EnvironmentId)
	}

	ret.Key = "root"

	return
}

func (s *EndpointCaseAlternativeService) loadMultiCasesData(tenantId consts.TenantId, cs casesHelper.AlternativeCase, parent *agentExec.CaseExecProcessor,
	baseCaseId, envId uint) (err error) {

	if !cs.NeedExec {
		return
	}

	if cs.Category != "case" {
		processor := agentExec.CaseExecProcessor{
			Title:    cs.Title,
			Category: fmt.Sprintf("%v", cs.Category),
			Key:      cs.Key,
		}

		for _, son := range cs.Children {
			s.loadMultiCasesData(tenantId, *son, &processor, baseCaseId, envId)
		}

		parent.Children = append(parent.Children, &processor)

		return
	}

	execObj := agentExec.InterfaceExecObj{}

	cs.BaseCaseId = baseCaseId
	if cs.Title == "required" {
		cs.Title += ": 空"
	}
	execObj.DebugData, _ = s.LoadDebugDataForExec(tenantId, cs, envId)

	s.loadConditionsAndScene(tenantId, &execObj, envId)

	title := cs.Title
	if cs.Sample != "" {
		title += fmt.Sprintf(": %v", cs.Sample)
	}

	child := agentExec.CaseExecProcessor{
		Title:    title,
		Category: fmt.Sprintf("%v", cs.Category),
		Key:      cs.Key,

		Data: &execObj,
	}
	parent.Children = append(parent.Children, &child)

	return
}

func (s *EndpointCaseAlternativeService) loadSingleCasesData(tenantId consts.TenantId, req agentExec.CasesExecReq, baseCaseId, envId uint) (
	ret agentExec.CaseExecProcessor, err error) {

	execObj := agentExec.InterfaceExecObj{}

	endpointCase, err := s.EndpointCaseService.Get(tenantId, baseCaseId)
	if err != nil {
		return
	}

	execObj.DebugData, err = s.DebugInterfaceService.GetDebugDataFromDebugInterface(tenantId, endpointCase.DebugInterfaceId)
	if err != nil {
		return
	}

	var validPaths []casesHelper.AlternativeCase
	s.getValidPaths(req.ExecObj, &validPaths)
	s.updateDebugData(tenantId, &execObj.DebugData, validPaths)

	s.loadConditionsAndScene(tenantId, &execObj, envId)

	root := agentExec.CaseExecProcessor{
		Title:    req.ExecObj.Title,
		Category: "root",
		Key:      "root",

		Children: []*agentExec.CaseExecProcessor{
			&agentExec.CaseExecProcessor{
				Title:    "多参数异常",
				Category: "case",
				Key:      req.ExecObj.Key,

				Data: &execObj,
			},
		},
	}

	ret.Children = append(ret.Children, &root)

	return
}

func (s *EndpointCaseAlternativeService) LoadDebugDataForExec(tenantId consts.TenantId, req casesHelper.AlternativeCase, envId uint) (
	ret domain.DebugData, err error) {

	endpointCase, err := s.EndpointCaseService.Get(tenantId, req.BaseCaseId)
	if err != nil {
		return
	}

	ret, err1 := s.DebugInterfaceService.GetDebugDataFromDebugInterface(tenantId, endpointCase.DebugInterfaceId)
	if err1 != nil {
		err = err1
		return
	}

	// update base url by selected environment
	server, err := s.ServeServerRepo.FindByServeAndExecEnv(tenantId, ret.ServeId, envId)
	if err == nil {
		ret.BaseUrl = server.Url
	}

	// update debugData
	fieldIn, fieldNameOrPath := s.getFieldProps(req.Path)
	if fieldIn == "" {
		logUtils.Info("failed to getFieldProps")
		return
	}

	s.changeFieldProps(&ret, fieldIn, fieldNameOrPath, req.Sample, req.FieldType)

	return
}

func (s *EndpointCaseAlternativeService) loadConditionsAndScene(tenantId consts.TenantId, execObj *agentExec.InterfaceExecObj, envId uint) {
	// load default environment for user
	env, _ := s.EnvironmentRepo.Get(tenantId, envId)
	if env.ID > 0 {
		server, _ := s.ServeServerRepo.FindByServeAndExecEnv(tenantId, execObj.DebugData.ServeId, env.ID)
		if server.ID > 0 {
			execObj.DebugData.ServerId = server.ID
		}
	}

	execObj.PreConditions, _ = s.ConditionRepo.ListTo(tenantId,
		execObj.DebugData.DebugInterfaceId, execObj.DebugData.EndpointInterfaceId,
		execObj.DebugData.UsedBy, "true", consts.ConditionSrcPre)
	execObj.PostConditions, _ = s.ConditionRepo.ListTo(tenantId,
		execObj.DebugData.DebugInterfaceId, execObj.DebugData.EndpointInterfaceId,
		execObj.DebugData.UsedBy, "true", consts.ConditionSrcPost)

	execObj.DebugData.EnvDataToView = &domain.EnvDataToView{}

	//
	_, execObj.ExecScene.ShareVars, _, _, *execObj.DebugData.GlobalParams =
		s.DebugSceneService.LoadScene(tenantId, &execObj.DebugData, 0, envId)
	execObj.DebugData.EnvDataToView = nil

	// get environment and settings on project level
	s.SceneService.LoadEnvVars(tenantId, &execObj.ExecScene, execObj.DebugData.ServerId, execObj.DebugData.DebugInterfaceId)
	s.SceneService.LoadProjectSettings(tenantId, &execObj.ExecScene, execObj.DebugData.ProjectId)
}

func (s *EndpointCaseAlternativeService) getValidPaths(alternativeCase casesHelper.AlternativeCase,
	validPaths *[]casesHelper.AlternativeCase) {

	if !alternativeCase.NeedExec {
		return
	}

	if alternativeCase.Category != "case" {
		for _, child := range alternativeCase.Children {
			s.getValidPaths(*child, validPaths)
		}

		return
	}

	validPath := casesHelper.AlternativeCase{
		NeedExec:  alternativeCase.NeedExec,
		Category:  alternativeCase.Category,
		Key:       alternativeCase.Key,
		Path:      alternativeCase.Path,
		Sample:    alternativeCase.Sample,
		FieldType: alternativeCase.FieldType,
	}

	*validPaths = append(*validPaths, validPath)
}

func (s *EndpointCaseAlternativeService) updateDebugData(tenantId consts.TenantId, debugData *domain.DebugData, validPaths []casesHelper.AlternativeCase) {
	for _, val := range validPaths {
		if val.Category != consts.AlternativeCaseCase || !val.NeedExec {
			continue
		}

		fieldIn, fieldNameOrPath := s.getFieldProps(val.Path)
		if fieldIn == "" {
			logUtils.Error("failed to getFieldProps")
			continue
		}
		s.changeFieldProps(debugData, fieldIn, fieldNameOrPath, val.Sample, val.FieldType)
	}
}

func (s *EndpointCaseAlternativeService) getName(val *casesHelper.AlternativeCase) (ret string) {
	arr := strings.Split(val.Path, "/")
	category := casesHelper.Category[strings.Trim(arr[0], "[]")]

	index := 0
	for i := len(arr) - 1; i >= 0; i -= 1 {
		if strings.Index(arr[i], "[") < 0 {
			index = i
			break
		}
	}

	ret = fmt.Sprintf("%s-%s-%s", category, arr[index], val.Title)

	return
}