go-proton-api/label_types.go

package proton

import (
	"encoding/json"
	"strings"
)

const (
	InboxLabel     = "0"
	AllDraftsLabel = "1"
	AllSentLabel   = "2"
	TrashLabel     = "3"
	SpamLabel      = "4"
	AllMailLabel   = "5"
	ArchiveLabel   = "6"
	SentLabel      = "7"
	DraftsLabel    = "8"
	OutboxLabel    = "9"
	StarredLabel   = "10"
)

type Label struct {
	ID    string
	Name  string
	Path  []string
	Color string
	Type  LabelType
}

func (label *Label) UnmarshalJSON(data []byte) error {
	type Alias Label

	aux := &struct {
		Path string

		*Alias
	}{
		Alias: (*Alias)(label),
	}

	if err := json.Unmarshal(data, &aux); err != nil {
		return err
	}

	label.Path = strings.Split(aux.Path, "/")

	return nil
}

func (label Label) MarshalJSON() ([]byte, error) {
	type Alias Label

	aux := &struct {
		Path string

		*Alias
	}{
		Path:  strings.Join(label.Path, "/"),
		Alias: (*Alias)(&label),
	}

	return json.Marshal(aux)
}

type CreateLabelReq struct {
	Name  string
	Color string
	Type  LabelType

	ParentID string `json:",omitempty"`
}

type UpdateLabelReq struct {
	Name  string
	Color string

	ParentID string `json:",omitempty"`
}

type LabelType int

const (
	LabelTypeLabel LabelType = iota + 1
	LabelTypeContactGroup
	LabelTypeFolder
	LabelTypeSystem
)