websocket

type Link struct {
	ID   string
	Type string
	URL  string
	Name string
}

type Message struct {
	Type    MessageType     `json:"type,omitempty"`
	Payload json.RawMessage `json:"payload"`
	Raw     []byte          `json:"--"`

	// internal information
	DocumentID string            `json:"-"` // current document id
	UserID     string            `json:"-"` // current connected user
	Permission domain.Permission `json:"-"` // permission of the client

	// channel to reply to the sender
	Client *WebsocketClient `json:"-"`
	Reply  chan []byte      `json:"-"`
}

type MessageType string

type NotifyMessage struct {
	Client  *WebsocketClient
	Payload json.RawMessage
}

type ProsemirrorCustomStep struct {
	Type    string          `json:"type"`
	Payload json.RawMessage `json:"payload"`
}

type ProsemirrorImageStep struct {
	Slice struct {
		Content []struct {
			Type  string `json:"type"`
			Attrs struct {
				ImageID string `json:"imageId"`
				Width   int    `json:"width"`
				Height  int    `json:"height"`
				Copy    bool   `json:"copy"`
			} `json:"attrs"`
		}
	}
}

type ProsemirrorInfoResponse struct {
	Type    MessageType `json:"type"`
	Payload struct {
		BaseVersion int64 `json:"base_version"`
		Version     int64 `json:"version"`
	} `json:"payload"`
}

type ProsemirrorInitMessage struct {
	DocumentID      string          `json:"documentid,omitempty"`
	DocumentSchema  json.RawMessage `json:"schema,omitempty"`
	DocumentVersion int64           `json:"version,omitempty"`
}

type ProsemirrorMark struct {
	Type  string `json:"type"`
	Attrs struct {
		ID   string `json:"id"`
		File bool   `json:"file"`
		Link bool   `json:"link"`

		Name string `json:"name"`
		Url  string `json:"url"`

		Process   bool   `json:"process"`
		ProcessID string `json:"processId"`
	} `json:"attrs"`
}

type ProsemirrorMarkStep struct {
	Type string          `json:"stepType"`
	From int             `json:"from"`
	To   int             `json:"to"`
	Mark ProsemirrorMark `json:"mark"`
}

type ProsemirrorReplaceStep struct {
	Slice struct {
		Content []ProsemirrorStepContent `json:"content"`
	}
}

type ProsemirrorStepContent struct {
	Type  string `json:"type"`
	Attrs struct {
		// attributes for pdf files
		DocumentID string `json:"documentId"`
		FileName   string `json:"fileName"`
		Src        string `json:"src"`

		// attributes for images
		ImageID string `json:"imageId"`
		Width   int    `json:"width"`
		Height  int    `json:"height"`
		Copy    bool   `json:"copy"`
	} `json:"attrs"`
	Marks   []*ProsemirrorMark        `json:"marks,omitempty"`
	Content []*ProsemirrorStepContent `json:"content,omitempty"`
}

type ProsemirrorStepMessage struct {
	DocumentID      string            `json:"documentid,omitempty"`
	DocumentVersion int64             `json:"version,omitempty"`
	ClientID        int               `json:"clientID,omitempty"`
	Steps           []json.RawMessage `json:"steps,omitempty"`
	SaveImmediate   bool              `json:"save_immediate"`
}

type ProsemirrorStepResponse struct {
	Type    MessageType `json:"type"`
	Payload struct {
		BaseVersion   int64             `json:"base_version"`
		Version       int64             `json:"version"`
		ClientIDs     []int             `json:"clientIds,omitempty"`
		Steps         []json.RawMessage `json:"steps,omitempty"`
		FromInit      bool              `json:"from_init"`
		SaveImmediate bool              `json:"save_immediate"`
	} `json:"payload"`
}

type Registration struct {
	Client *WebsocketClient
	Done   chan bool
}

type Response struct {
	Type    MessageType `json:"type,omitempty"`
	Payload interface{} `json:"payload"`
}

type WebsocketClient struct {
	// reference the websocket connection
	Conn *websocket.Conn

	// document is the unique id of the block that the client is working on
	DocumentID     string
	DocumentSchema json.RawMessage

	// unique id of the respective user
	UserID string

	// Document permissions of the respective client (read, comment, edit)
	Permission domain.Permission

	// use a channel to send messages
	Send chan []byte

	// reference to the handler that will manage the message
	MessageHandler chan Message
}

type WebsocketHub struct {
	Srv        *environment.Services     // external services
	Rooms      map[string]*WebsocketRoom // room for documents
	Register   chan *Registration        // register a new client
	Unregister chan *Registration        // deregister a client
}

type WebsocketRoom struct {
	Clients    map[*WebsocketClient]bool
	Register   chan *Registration // register a new client
	Unregister chan *Registration // deregister a client

	Notify    chan NotifyMessage // send message to all clients except the sender
	Broadcast chan []byte        // broadcast messages to all clients

	Handler chan Message // handle incoming messages

	DocumentID      string          // unique id of the respective editor content
	DocumentSchema  json.RawMessage // schema of the respective document
	DocumentVersion int64           // current document version on the server
}