models.go

package main

import (
	"encoding/json"
	"fmt"
	"time"

	"github.com/hashicorp/jsonapi"
)

type UnsetableTime struct {
	Value *time.Time
}

func (t *UnsetableTime) MarshalAttribute() (interface{}, error) {
	if t == nil {
		return nil, nil
	}

	if t.Value == nil {
		return json.RawMessage(nil), nil
	} else {
		return t.Value, nil
	}
}

// Blog is a model representing a blog site
type Blog struct {
	ID            int            `jsonapi:"primary,blogs"`
	Title         string         `jsonapi:"attr,title"`
	Posts         []*Post        `jsonapi:"relation,posts"`
	CurrentPost   *Post          `jsonapi:"relation,current_post"`
	CurrentPostID int            `jsonapi:"attr,current_post_id"`
	CreatedAt     *UnsetableTime `jsonapi:"attr,created_at,omitempty,iso8601"`
	ViewCount     int            `jsonapi:"attr,view_count"`
}

// Post is a model representing a post on a blog
type Post struct {
	ID       int        `jsonapi:"primary,posts"`
	BlogID   int        `jsonapi:"attr,blog_id"`
	Title    string     `jsonapi:"attr,title"`
	Body     string     `jsonapi:"attr,body"`
	Comments []*Comment `jsonapi:"relation,comments"`
}

// Comment is a model representing a user submitted comment
type Comment struct {
	ID     int    `jsonapi:"primary,comments"`
	PostID int    `jsonapi:"attr,post_id"`
	Body   string `jsonapi:"attr,body"`
}

// JSONAPILinks implements the Linkable interface for a blog
func (blog Blog) JSONAPILinks() *jsonapi.Links {
	return &jsonapi.Links{
		"self": fmt.Sprintf("https://example.com/blogs/%d", blog.ID),
	}
}

// JSONAPIRelationshipLinks implements the RelationshipLinkable interface for a blog
func (blog Blog) JSONAPIRelationshipLinks(relation string) *jsonapi.Links {
	if relation == "posts" {
		return &jsonapi.Links{
			"related": fmt.Sprintf("https://example.com/blogs/%d/posts", blog.ID),
		}
	}
	if relation == "current_post" {
		return &jsonapi.Links{
			"related": fmt.Sprintf("https://example.com/blogs/%d/current_post", blog.ID),
		}
	}
	return nil
}

// JSONAPIMeta implements the Metable interface for a blog
func (blog Blog) JSONAPIMeta() *jsonapi.Meta {
	return &jsonapi.Meta{
		"detail": "extra details regarding the blog",
	}
}

// JSONAPIRelationshipMeta implements the RelationshipMetable interface for a blog
func (blog Blog) JSONAPIRelationshipMeta(relation string) *jsonapi.Meta {
	if relation == "posts" {
		return &jsonapi.Meta{
			"detail": "posts meta information",
		}
	}
	if relation == "current_post" {
		return &jsonapi.Meta{
			"detail": "current post meta information",
		}
	}
	return nil
}