siyuan/kernel/av/cell.go

// SiYuan - Refactor your thinking
// Copyright (c) 2020-present, b3log.org
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

package av

import (
	"github.com/88250/gulu"
	"github.com/88250/lute/ast"
	"strings"
)

type Cell struct {
	ID        string     `json:"id"`
	Value     *Value     `json:"value"`
	ValueType ColumnType `json:"valueType"`
	Color     string     `json:"color"`
	BgColor   string     `json:"bgColor"`
}

type Value struct {
	Block   *ValueBlock    `json:"block,omitempty"`
	Text    *ValueText     `json:"text,omitempty"`
	Number  *ValueNumber   `json:"number,omitempty"`
	Date    *ValueDate     `json:"date,omitempty"`
	Select  *ValueSelect   `json:"select,omitempty"`
	MSelect []*ValueSelect `json:"mSelect,omitempty"`
}

func (value *Value) ToJSONString() string {
	data, err := gulu.JSON.MarshalJSON(value)
	if nil != err {
		return ""
	}
	return string(data)
}

func (value *Value) Compare(other *Value) int {
	if nil == value {
		return -1
	}
	if nil == other {
		return 1
	}
	if nil != value.Block && nil != other.Block {
		return strings.Compare(value.Block.Content, other.Block.Content)
	}
	if nil != value.Text && nil != other.Text {
		return strings.Compare(value.Text.Content, other.Text.Content)
	}
	if nil != value.Number && nil != other.Number {
		if value.Number.Content > other.Number.Content {
			return 1
		} else if value.Number.Content < other.Number.Content {
			return -1
		} else {
			return 0
		}
	}
	if nil != value.Date && nil != other.Date {
		if value.Date.Content > other.Date.Content {
			return 1
		} else if value.Date.Content < other.Date.Content {
			return -1
		} else {
			return 0
		}
	}
	if nil != value.Select && nil != other.Select {
		return strings.Compare(value.Select.Content, other.Select.Content)
	}
	if nil != value.MSelect && nil != other.MSelect {
		var v1 string
		for _, v := range value.MSelect {
			v1 += v.Content
		}
		var v2 string
		for _, v := range other.MSelect {
			v2 += v.Content
		}
		return strings.Compare(v1, v2)
	}
	return 0
}

func NewCellBlock(blockID, blockContent string) *Cell {
	return &Cell{
		ID:        ast.NewNodeID(),
		Value:     &Value{Block: &ValueBlock{ID: blockID, Content: blockContent}},
		ValueType: ColumnTypeBlock,
	}
}

func NewCell(valueType ColumnType) *Cell {
	return &Cell{
		ID:        ast.NewNodeID(),
		ValueType: valueType,
	}
}

type ValueBlock struct {
	ID      string `json:"id"`
	Content string `json:"content"`
}

type ValueText struct {
	Content string `json:"content"`
}

type ValueNumber struct {
	Content float64 `json:"content"`
}

type ValueDate struct {
	Content int64 `json:"content"`
}

type ValueSelect struct {
	Content string `json:"content"`
}