oef/orm/question.go

package orm

import (
	"net/http"

	"git.andreafazzi.eu/andrea/oef/renderer"
	"github.com/jinzhu/gorm"
)

type Question struct {
	gorm.Model

	Text string

	Answers []*Answer
}

func (q *Question) GetID() uint { return q.ID }

func (q *Question) String() string {
	return q.Text
}

func (q *Question) Create(args map[string]string, r *http.Request) (interface{}, error) {
	if r.Method == "GET" {
		return nil, nil
	} else {
		question := new(Question)
		err := renderer.Decode(question, r)
		if err != nil {
			return nil, err
		}
		question, err = CreateQuestion(question)
		if err != nil {
			return nil, err
		}
		return question, nil
	}
}

func (q *Question) Read(args map[string]string, r *http.Request) (interface{}, error) {
	var question Question

	id := args["id"]

	if err := DB().Preload("Answers").Where("id = ?", id).Find(&question).Error; err != nil {
		return nil, err
	}

	return &question, nil
}

func (q *Question) ReadAll(args map[string]string, r *http.Request) (interface{}, error) {
	var questions []*Question
	if err := DB().Order("created_at").Find(&questions).Error; err != nil {
		return nil, err
	}
	return questions, nil
}

func (q *Question) Update(args map[string]string, r *http.Request) (interface{}, error) {
	if r.Method == "GET" {
		return q.Read(args, r)
	} else {
		question, err := q.Read(args, nil)
		if err != nil {
			return nil, err
		}
		err = renderer.Decode(question, r)
		if err != nil {
			return nil, err
		}
		_, err = SaveQuestion(question)
		if err != nil {
			return nil, err
		}
		question, err = q.Read(args, nil)
		if err != nil {
			return nil, err
		}
		return question.(*Question), nil
	}
}

func (q *Question) Delete(args map[string]string, r *http.Request) (interface{}, error) {
	return nil, nil
}

func CreateQuestion(question *Question) (*Question, error) {
	if err := DB().Create(question).Error; err != nil {
		return nil, err
	}
	return question, nil
}

func SaveQuestion(question interface{}) (interface{}, error) {
	if err := DB().Omit("Answers").Save(question).Error; err != nil {
		return nil, err
	}
	return question, nil
}