package bug import ( "encoding/json" "fmt" "strings" "github.com/MichaelMure/git-bug/identity" "github.com/MichaelMure/git-bug/util/git" "github.com/MichaelMure/git-bug/util/text" "github.com/MichaelMure/git-bug/util/timestamp" ) var _ Operation = &CreateOperation{} // CreateOperation define the initial creation of a bug type CreateOperation struct { OpBase Title string Message string Files []git.Hash } func (op *CreateOperation) base() *OpBase { return &op.OpBase } func (op *CreateOperation) ID() string { return idOperation(op) } func (op *CreateOperation) Apply(snapshot *Snapshot) { snapshot.addActor(op.Author) snapshot.addParticipant(op.Author) snapshot.Title = op.Title comment := Comment{ id: op.ID(), Message: op.Message, Author: op.Author, UnixTime: timestamp.Timestamp(op.UnixTime), } snapshot.Comments = []Comment{comment} snapshot.Author = op.Author snapshot.CreatedAt = op.Time() snapshot.Timeline = []TimelineItem{ &CreateTimelineItem{ CommentTimelineItem: NewCommentTimelineItem(op.ID(), comment), }, } } func (op *CreateOperation) GetFiles() []git.Hash { return op.Files } func (op *CreateOperation) Validate() error { if err := opBaseValidate(op, CreateOp); err != nil { return err } if text.Empty(op.Title) { return fmt.Errorf("title is empty") } if strings.Contains(op.Title, "\n") { return fmt.Errorf("title should be a single line") } if !text.Safe(op.Title) { return fmt.Errorf("title is not fully printable") } if !text.Safe(op.Message) { return fmt.Errorf("message is not fully printable") } return nil } // Workaround to avoid the inner OpBase.MarshalJSON overriding the outer op // MarshalJSON func (op *CreateOperation) MarshalJSON() ([]byte, error) { base, err := json.Marshal(op.OpBase) if err != nil { return nil, err } // revert back to a flat map to be able to add our own fields var data map[string]interface{} if err := json.Unmarshal(base, &data); err != nil { return nil, err } data["title"] = op.Title data["message"] = op.Message data["files"] = op.Files return json.Marshal(data) } // Workaround to avoid the inner OpBase.MarshalJSON overriding the outer op // MarshalJSON func (op *CreateOperation) UnmarshalJSON(data []byte) error { // Unmarshal OpBase and the op separately base := OpBase{} err := json.Unmarshal(data, &base) if err != nil { return err } aux := struct { Title string `json:"title"` Message string `json:"message"` Files []git.Hash `json:"files"` }{} err = json.Unmarshal(data, &aux) if err != nil { return err } op.OpBase = base op.Title = aux.Title op.Message = aux.Message op.Files = aux.Files return nil } // Sign post method for gqlgen func (op *CreateOperation) IsAuthored() {} func NewCreateOp(author identity.Interface, unixTime int64, title, message string, files []git.Hash) *CreateOperation { return &CreateOperation{ OpBase: newOpBase(CreateOp, author, unixTime), Title: title, Message: message, Files: files, } } // CreateTimelineItem replace a Create operation in the Timeline and hold its edition history type CreateTimelineItem struct { CommentTimelineItem } // Sign post method for gqlgen func (c *CreateTimelineItem) IsAuthored() {} // Convenience function to apply the operation func Create(author identity.Interface, unixTime int64, title, message string) (*Bug, *CreateOperation, error) { return CreateWithFiles(author, unixTime, title, message, nil) } func CreateWithFiles(author identity.Interface, unixTime int64, title, message string, files []git.Hash) (*Bug, *CreateOperation, error) { newBug := NewBug() createOp := NewCreateOp(author, unixTime, title, message, files) if err := createOp.Validate(); err != nil { return nil, createOp, err } newBug.Append(createOp) return newBug, createOp, nil }