dnnai/flutter_asr_client/lib/models/websocket_message.dart

import 'dart:convert';

sealed class WebSocketMessage {
  const WebSocketMessage();

  Map<String, dynamic> toJson();

  String toJsonString() => jsonEncode(toJson());
}

final class AudioMessage extends WebSocketMessage {
  const AudioMessage({required this.data});

  final String data;

  @override
  Map<String, dynamic> toJson() => {'type': 'audio', 'data': data};
}

final class TranscribeMessage extends WebSocketMessage {
  const TranscribeMessage();

  @override
  Map<String, dynamic> toJson() => {'type': 'transcribe'};
}

final class ClearMessage extends WebSocketMessage {
  const ClearMessage();

  @override
  Map<String, dynamic> toJson() => {'type': 'clear'};
}

final class FinalizeMessage extends WebSocketMessage {
  const FinalizeMessage();

  @override
  Map<String, dynamic> toJson() => {'type': 'finalize'};
}

sealed class ServerMessage {
  const ServerMessage();

  static ServerMessage fromJson(Map<String, dynamic> json) {
    final type = json['type'] as String?;
    switch (type) {
      case 'connected':
        return ConnectedMessage(
          model: json['model'] as Map<String, dynamic>? ?? const {},
        );
      case 'partial':
        return PartialMessage(
          text: json['text'] as String? ?? '',
          seq: json['seq'] as int? ?? 0,
        );
      case 'utterance':
        return UtteranceMessage(
          text: json['text'] as String? ?? '',
          seq: json['seq'] as int? ?? 0,
        );
      case 'transcript':
        return TranscriptMessage(
          text: json['text'] as String? ?? '',
          language: json['language'] as String? ?? '',
        );
      case 'reply':
        return ReplyMessage(
          text: json['text'] as String? ?? '',
        );
      case 'reply_partial':
        return ReplyPartialMessage(
          text: json['text'] as String? ?? '',
        );
      case 'cleared':
        return const ClearedMessage();
      case 'error':
        return ErrorMessage(
          message: json['message'] as String? ?? 'Unknown error',
        );
      default:
        return ErrorMessage(message: 'Unknown message type: $type');
    }
  }

  static ServerMessage fromJsonString(String source) {
    try {
      final json = jsonDecode(source) as Map<String, dynamic>;
      return fromJson(json);
    } on FormatException catch (e) {
      return ErrorMessage(message: 'Invalid JSON: $e');
    }
  }
}

final class ConnectedMessage extends ServerMessage {
  const ConnectedMessage({required this.model});

  final Map<String, dynamic> model;
}

final class PartialMessage extends ServerMessage {
  const PartialMessage({required this.text, this.seq = 0});

  final String text;
  final int seq;
}

final class UtteranceMessage extends ServerMessage {
  const UtteranceMessage({required this.text, this.seq = 0});

  final String text;
  final int seq;
}

final class ReplyMessage extends ServerMessage {
  const ReplyMessage({required this.text});

  final String text;
}

final class ReplyPartialMessage extends ServerMessage {
  const ReplyPartialMessage({required this.text});

  final String text;
}

final class ClearedMessage extends ServerMessage {
  const ClearedMessage();
}

final class TranscriptMessage extends ServerMessage {
  const TranscriptMessage({required this.text, this.language = ''});

  final String text;
  final String language;
}

final class ErrorMessage extends ServerMessage {
  const ErrorMessage({required this.message});

  final String message;
}