get_form_data.go

package uadmin

import (
	"fmt"
	"html/template"

	//"github.com/jinzhu/gorm"
	"net/http"
	"reflect"
	"strconv"
	"strings"
	"time"

	"github.com/arbrix/uadmin/helper"
)

func getFormData(a interface{}, r *http.Request, session *Session, s *ModelSchema, user *User) {
	// This holds the formatted value of the field
	var value interface{}
	var f *F
	var err error

	// Get the type of model
	t := reflect.TypeOf(a)

	// Get the value of the model
	modelValue := reflect.ValueOf(a)

	// Get the primary key
	newForm := r.FormValue("ModelID") == "0"
	ModelID64, _ := strconv.ParseUint(r.FormValue("ModelID"), 10, 64)

	// Loop over the fields of the model and get schema information
	for index := 0; index < t.NumField(); index++ {
		// Read field value
		fieldValue := modelValue.Field(index)

		// Get the field from schema
		f = &F{}
		fName := t.Field(index).Name
		if t.Field(index).Anonymous {
			fName = "ID"
		}
		for i := range s.Fields {
			if s.Fields[i].Name == fName {
				f = &s.Fields[i]
				break
			}
		}
		if f.Hidden || !f.FormDisplay {
			continue
		}
		if f.Type == cFK {
			fieldValue = modelValue.FieldByName(f.Name + "ID")
		}

		// Check if the field was not found in schema
		if f.Name == "" {
			continue
		}

		// For new records:
		// Overide field value with any values passed in request
		// If not available check if there is a default value for the field
		//if newForm {
		if r.FormValue(t.Field(index).Name) != "" {
			if f.Type == cNUMBER || f.Type == cLIST {
				fValue, _ := strconv.ParseInt(r.FormValue(t.Field(index).Name), 10, 64)
				fieldValue = reflect.ValueOf(fValue)
			} else if f.Type == cDATE {
				var tm time.Time
				if t.Field(index).Type.Kind() == reflect.Ptr {
					var ptm *time.Time
					if r.FormValue(t.Field(index).Name) == "" {
						fieldValue = reflect.ValueOf(ptm)
					} else {
						tm, err = time.Parse("2006-01-02 15:04", r.FormValue(t.Field(index).Name))
						if err != nil {
							Trail(ERROR, "getFormData unable to parse time format(%s). %s", r.FormValue(t.Field(index).Name), err)
						}
						fieldValue = reflect.ValueOf(&tm)
					}
				} else {
					tm, err = time.Parse("2006-01-02 15:04", r.FormValue(t.Field(index).Name))
					if err != nil {
						Trail(ERROR, "getFormData unable to parse time format(%s). %s", r.FormValue(t.Field(index).Name), err)
					}
					fieldValue = reflect.ValueOf(tm)
				}
			} else if f.Type == cBOOL {
				fieldValue = reflect.ValueOf(r.FormValue(t.Field(index).Name) == "on")
			} else {
				fieldValue = reflect.ValueOf(r.FormValue(t.Field(index).Name))
			}
		} else if f.DefaultValue != "" && newForm {
			DefaultValue := f.DefaultValue
			DefaultValue = strings.Replace(DefaultValue, "{NOW}", time.Now().Format("2006-01-02 15:04:05"), -1)
			fieldValue = reflect.ValueOf(DefaultValue)
		}

		// Check for approval
		if f.Approval {
			// Check if there is an approval record
			approvals := []Approval{}
			Filter(&approvals, "model_name = ? AND column_name = ? AND model_pk = ?", strings.ToLower(t.Name()), f.Name, ModelID64)
			if len(approvals) != 0 {

				// Get the last approval
				lastA := approvals[len(approvals)-1]
				f.ApprovalAction = lastA.ApprovalAction
				f.NewValue = lastA.NewValueDescription
				f.ChangedBy = lastA.ChangedBy
				f.ChangeDate = &lastA.ChangeDate
				f.ApprovalDate = lastA.ApprovalDate
				f.ApprovalBy = lastA.ApprovalBy
				f.ApprovalID = lastA.ID
				f.OldValue = lastA.OldValue

				// Remove required if the field has a pending approval
				f.Required = f.Required && (f.ApprovalAction != ApprovalAction(0))
			}
		}

		// Check the data type
		if f.Type == cID {
			m, ok := fieldValue.Interface().(Model)
			if !ok {
				Trail(ERROR, "Unable tp parse value of ID for %s.%s (%#v)", t.Name(), f.Name, fieldValue.Interface())
			}
			value = m.ID
		} else if f.Type == cNUMBER {
			if f.Format != "" {
				value = fmt.Sprintf(f.Format, fieldValue.Interface())
			} else {
				value = fieldValue.Interface()
			}
		} else if f.Type == cFK {
			// Get selected items's ID
			fkValue, _ := strconv.ParseUint(fmt.Sprint(fieldValue.Interface()), 10, 64)
			value = fkValue

			if f.LimitChoicesTo == nil {
				fkType := t.Field(index).Type.Name()
				if t.Field(index).Type.Kind() == reflect.Ptr {
					fkType = t.Field(index).Type.Elem().Name()
				}
				fkList, _ := NewModelArray(strings.ToLower(fkType), false)
				All(fkList.Addr().Interface())

				// Build choices
				f.Choices = []Choice{
					{
						K:        0,
						V:        "-",
						Selected: uint(fkValue) == 0,
					},
				}

				for i := 0; i < fkList.Len(); i++ {
					f.Choices = append(f.Choices, Choice{
						K:        GetID(fkList.Index(i)),
						V:        GetString(fkList.Index(i).Interface()),
						Selected: uint(fkValue) == GetID(fkList.Index(i)),
					})
				}
			} else {
				f.Choices = f.LimitChoicesTo(a, &session.User)

				for i := 0; i < len(f.Choices); i++ {
					f.Choices[i].Selected = uint(fkValue) == f.Choices[i].K
				}
			}

		} else if f.Type == cM2M {
			if fmt.Sprint(reflect.TypeOf(fieldValue.Interface())) == "string" {
				continue
			}
			fKType := reflect.TypeOf(fieldValue.Interface()).Elem()
			m, ok := NewModelArray(strings.ToLower(fKType.Name()), false)

			if !ok {
				Trail(ERROR, "GetListSchema.NewModelArray. No model name (%s)", s.ModelName)
			}
			if f.LimitChoicesTo == nil {
				All(m.Addr().Interface())
				f.Choices = []Choice{}
				for i := 0; i < m.Len(); i++ {
					item := m.Index(i).Interface()
					id := GetID(m.Index(i))
					// if id == myID {
					// 	continue
					// }
					f.Choices = append(f.Choices, Choice{
						K: id,
						V: GetString(item),
					})

				}
			} else {
				f.Choices = f.LimitChoicesTo(a, &session.User)
			}

			for i := 0; i < fieldValue.Len(); i++ {
				for counter, val := range f.Choices {
					itemID := GetID(fieldValue.Index(i))
					if val.K == itemID {
						f.Choices[counter].Selected = true
					}
				}
			}
		} else if f.Type == cDATE {
			if newForm && t.Field(index).Type.Kind() != reflect.Ptr {
				value = time.Now().Format("2006-01-02 15:04:05")
			} else {
				var d *time.Time
				// If the date is not a pointer to date make it a pointer
				if t.Field(index).Type.Kind() != reflect.Ptr {
					tempD, _ := fieldValue.Interface().(time.Time)
					d = &tempD
				} else {
					d, _ = fieldValue.Interface().(*time.Time)
				}
				if d == nil {
					value = ""
				} else {
					value = d.Format("2006-01-02 15:04:05") //2006-01-02 15:04:05
				}
			}
		} else if f.Type == cBOOL {
			d, ok := fieldValue.Interface().(bool)
			if !ok {
				Trail(ERROR, "Unable to parse bool value for %s.%s (%#v)", t.Name(), f.Name, fieldValue.Interface())
			}
			value = d
		} else if f.Type == cLIST {
			value = fieldValue.Int()
			if f.LimitChoicesTo != nil {
				f.Choices = append([]Choice{{"-", 0, false}}, f.LimitChoicesTo(a, &session.User)...)
			}
			for i := range f.Choices {
				f.Choices[i].Selected = f.Choices[i].K == uint(fieldValue.Int())
			}
		} else if f.Type == cMULTILINGUAL || f.Type == cHTML_MULTILINGUAL {
			value = fieldValue.Interface()
			for i := range ActiveLangs {
				val := helper.PrepForHTML(Translate(fmt.Sprint(value), ActiveLangs[i].Code, false))
				f.Translations[i].Value = val
				if f.ChangedBy != "" {
					newVal := helper.PrepForHTML(Translate(fmt.Sprint(f.NewValue), ActiveLangs[i].Code, false))
					f.Translations[i].NewValue = newVal
					oldVal := helper.PrepForHTML(Translate(fmt.Sprint(f.OldValue), ActiveLangs[i].Code, false))
					f.Translations[i].OldValue = oldVal
				}
			}
		} else if f.Type == cLINK {
			URL := fieldValue.String()
			if URL != "" {
				if strings.Contains(URL, "?") {
					URL += "&"
				} else {
					URL += "?"
				}
				URL += "x-csrf-token=" + session.Key
			}
			value = URL
		} else {
			value = fieldValue.Interface()
		}
		f.Value = value

	}

	// Get data from method fields
	for index := 0; index < t.NumMethod(); index++ {
		// Check if the method should be included in the field list
		if strings.Contains(t.Method(index).Name, "__Form") {
			if strings.ToLower(string(t.Method(index).Name[0])) == string(t.Method(index).Name[0]) {
				continue
			}

			in := []reflect.Value{}
			ret := modelValue.Method(index).Call(in)
			s.FieldByName(t.Method(index).Name).Value = template.HTML(stripHTMLScriptTag(fmt.Sprint(ret[0].Interface())))
		}
	}

	inlineData := []listData{}
	if uint(ModelID64) != 0 {
		for _, inlineS := range s.Inlines {
			inlineModel, _ := NewModel(strings.ToLower(inlineS.ModelName), false)
			//inlineQ := fmt.Sprintf("%s = %d", foreignKeys[s.ModelName][strings.ToLower(inlineS.ModelName)], ModelID64)
			//r.Form.Set("inline_id", inlineQ)

			// Check if there the inline has a ListModifier
			query := ""
			args := []interface{}{}
			if inlineS.ListModifier != nil {
				query, args = inlineS.ListModifier(inlineS, user)
			}
			// Add the fk for the inline
			if query != "" {
				query += " AND "
			}
			query += fmt.Sprintf("%s = ?", foreignKeys[s.ModelName][strings.ToLower(inlineS.ModelName)])
			args = append(args, ModelID64)

			rows := getListData(inlineModel.Interface(), PageLength, r, session, query, inlineS, args...)
			r.Form.Del("inline_id")
			if rows.Count == 0 {
				rows.Rows = [][]interface{}{}
			}
			inlineData = append(inlineData, *rows)
		}
	}
	s.InlinesData = inlineData
	s.ModelID = uint(ModelID64)
}