import { AcceptChannelMessage } from './messages/AcceptChannelMessage';
import { ChannelAnnouncementMessage } from './messages/ChannelAnnouncementMessage';
import { ChannelUpdateMessage } from './messages/ChannelUpdateMessage';
import { ErrorMessage } from './messages/ErrorMessage';
import { GossipTimestampFilterMessage } from './messages/GossipTimestampFilterMessage';
import { InitMessage } from './messages/InitMessage';
import { IWireMessage } from './messages/IWireMessage';
import { NodeAnnouncementMessage } from './messages/NodeAnnouncementMessage';
import { OpenChannelMessage } from './messages/OpenChannelMessage';
import { PingMessage } from './messages/PingMessage';
import { PongMessage } from './messages/PongMessage';
import { QueryChannelRangeMessage } from './messages/QueryChannelRangeMessage';
import { QueryShortChannelIdsMessage } from './messages/QueryShortChannelIdsMessage';
import { ReplyChannelRangeMessage } from './messages/ReplyChannelRangeMessage';
import { ReplyShortChannelIdsEndMessage } from './messages/ReplyShortChannelIdsEndMessage';
import { ShutdownMessage } from './messages/ShutdownMessage';
import { MessageType } from './MessageType';

export function deserialize(buffer: Buffer): IWireMessage {
  const type = buffer.readUInt16BE(0);

  switch (type) {
    // control messages
    case MessageType.Init:
      return InitMessage.deserialize(buffer);
    case MessageType.Error:
      return ErrorMessage.deserialize(buffer);
    case MessageType.Ping:
      return PingMessage.deserialize(buffer);
    case MessageType.Pong:
      return PongMessage.deserialize(buffer);

    // channel messages
    case MessageType.OpenChannel:
      return OpenChannelMessage.deserialize(buffer);
    case MessageType.AcceptChannel:
      return AcceptChannelMessage.deserialize(buffer);
    case MessageType.Shutdown:
      return ShutdownMessage.deserialize(buffer);

    // gossip messages
    case MessageType.NodeAnnouncement:
      return NodeAnnouncementMessage.deserialize(buffer);
    case MessageType.ChannelAnnouncement:
      return ChannelAnnouncementMessage.deserialize(buffer);
    case MessageType.ChannelUpdate:
      return ChannelUpdateMessage.deserialize(buffer);
    case MessageType.QueryShortChannelIds:
      return QueryShortChannelIdsMessage.deserialize(buffer);
    case MessageType.ReplyShortChannelIdsEnd:
      return ReplyShortChannelIdsEndMessage.deserialize(buffer);
    case MessageType.QueryChannelRange:
      return QueryChannelRangeMessage.deserialize(buffer);
    case MessageType.ReplyChannelRange:
      return ReplyChannelRangeMessage.deserialize(buffer);
    case MessageType.GossipTimestampFilter:
      return GossipTimestampFilterMessage.deserialize(buffer);
  }
}