func dataSourceAlibabacloudStackOosTemplatesRead()

in alibabacloudstack/data_source_apsarastack_oos_templates.go [169:312]
134 lines of code
35 McCabe index (conditional complexity)

func dataSourceAlibabacloudStackOosTemplatesRead(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*connectivity.AlibabacloudStackClient)

	action := "ListTemplates"
	request := make(map[string]interface{})
	if v, ok := d.GetOk("category"); ok {
		request["Category"] = v
	}
	if v, ok := d.GetOk("created_by"); ok {
		request["CreatedBy"] = v
	}
	if v, ok := d.GetOk("created_date"); ok {
		request["CreatedDateBefore"] = v
	}
	if v, ok := d.GetOk("created_date_after"); ok {
		request["CreatedDateAfter"] = v
	}
	if v, ok := d.GetOkExists("has_trigger"); ok {
		request["HasTrigger"] = v
	}
	if v, ok := d.GetOk("share_type"); ok {
		request["ShareType"] = v
	}
	if v, ok := d.GetOk("sort_field"); ok {
		request["SortField"] = v
	}
	if v, ok := d.GetOk("sort_order"); ok {
		request["SortOrder"] = v
	}
	if v, ok := d.GetOk("tags"); ok {
		respJson, err := convertMaptoJsonString(v.(map[string]interface{}))
		if err != nil {
			return errmsgs.WrapError(err)
		}
		request["Tags"] = respJson
	}
	if v, ok := d.GetOk("template_format"); ok {
		request["TemplateFormat"] = v
	}
	if v, ok := d.GetOk("template_type"); ok {
		request["TemplateType"] = v
	}
	request["MaxResults"] = PageSizeLarge
	request["Product"] = "Oos"
	request["OrganizationId"] = client.Department

	var objects []map[string]interface{}
	var templateNameRegex *regexp.Regexp
	if v, ok := d.GetOk("name_regex"); ok {
		r, err := regexp.Compile(v.(string))
		if err != nil {
			return errmsgs.WrapError(err)
		}
		templateNameRegex = r
	}

	idsMap := make(map[string]string)
	if v, ok := d.GetOk("ids"); ok {
		for _, vv := range v.([]interface{}) {
			if vv == nil {
				continue
			}
			idsMap[vv.(string)] = vv.(string)
		}
	}

	for {
		response, err := client.DoTeaRequest("POST", "Oos", "2019-06-01", action, "", nil, nil, request)
		if err != nil {
			return err
		}

		resp, err := jsonpath.Get("$.Templates", response)
		if err != nil {
			return errmsgs.WrapErrorf(err, errmsgs.FailedGetAttributeMsg, action, "$.Templates", response)
		}
		result, _ := resp.([]interface{})
		for _, v := range result {
			item := v.(map[string]interface{})
			if templateNameRegex != nil {
				if !templateNameRegex.MatchString(fmt.Sprint(item["TemplateName"])) {
					continue
				}
			}
			if len(idsMap) > 0 {
				if _, ok := idsMap[fmt.Sprint(item["TemplateName"])]; !ok {
					continue
				}
			}
			objects = append(objects, item)
		}
		if nextToken, ok := response["NextToken"].(string); ok && nextToken != "" {
			request["NextToken"] = nextToken
		} else {
			break
		}
	}

	ids := make([]string, 0)
	names := make([]interface{}, 0)
	s := make([]map[string]interface{}, 0)
	for _, object := range objects {
		mapping := map[string]interface{}{
			"category":         object["Category"],
			"created_by":       object["CreatedBy"],
			"created_date":     object["CreatedDate"],
			"description":      object["Description"],
			"has_trigger":      object["HasTrigger"],
			"share_type":       object["ShareType"],
			"tags":             object["Tags"],
			"template_format":  object["TemplateFormat"],
			"template_id":      object["TemplateId"],
			"id":               fmt.Sprint(object["TemplateName"]),
			"template_name":    fmt.Sprint(object["TemplateName"]),
			"template_type":    object["TemplateType"],
			"template_version": object["TemplateVersion"],
			"updated_by":       object["UpdatedBy"],
			"updated_date":     object["UpdatedDate"],
		}
		ids = append(ids, fmt.Sprint(object["TemplateName"]))
		names = append(names, object["TemplateName"])
		s = append(s, mapping)
	}

	d.SetId(dataResourceIdHash(ids))
	if err := d.Set("ids", ids); err != nil {
		return errmsgs.WrapError(err)
	}

	if err := d.Set("names", names); err != nil {
		return errmsgs.WrapError(err)
	}

	if err := d.Set("templates", s); err != nil {
		return errmsgs.WrapError(err)
	}
	if output, ok := d.GetOk("output_file"); ok && output.(string) != "" {
		if err := writeToFile(output.(string), s); err != nil {
			return err
		}
	}

	return nil
}