use enum in chat member

DinoLeung · Mar 31, 2024 · 107eca1 · 107eca1
1 parent 8a8655b
commit 107eca1
Show file tree

Hide file tree

Showing 7 changed files with 33 additions and 13 deletions.
diff --git a/lib/src/telegram/models/chat_member.dart b/lib/src/telegram/models/chat_member.dart
@@ -18,6 +18,26 @@
 
 part of '../model.dart';
 
+@JsonEnum()
+enum ChatMemberStatus {
+  @JsonValue('creator')
+  creator,
+  @JsonValue('administrator')
+  administrator,
+  @JsonValue('member')
+  member,
+  @JsonValue('restricted')
+  restricted,
+  @JsonValue('left')
+  left,
+  @JsonValue('kicked')
+  kicked,
+}
+
+extension ChatMemberStatusExtenson on ChatMemberStatus {
+  String value() => _$ChatMemberStatusEnumMap[this]!;
+}
+
 /// This object contains information about one member of a chat.
 ///
 /// Currently, the following 6 types of chat members are supported:
@@ -32,7 +52,7 @@ part of '../model.dart';
 
 @JsonSerializable(fieldRename: FieldRename.snake)
 class ChatMember {
-  String status;
+  ChatMemberStatus status;
   User user;
 
   ChatMember({

diff --git a/lib/src/telegram/models/chat_member_administrator.dart b/lib/src/telegram/models/chat_member_administrator.dart
@@ -26,7 +26,7 @@ part of '../model.dart';
 @JsonSerializable(fieldRename: FieldRename.snake)
 class ChatMemberAdministrator implements ChatMember {
   @override
-  String status;
+  ChatMemberStatus status;
   @override
   User user;
   bool canBeEdited;
@@ -48,8 +48,8 @@ class ChatMemberAdministrator implements ChatMember {
   String? customTitle;
 
   ChatMemberAdministrator({
-    required this.status,
     required this.user,
+    this.status = ChatMemberStatus.administrator,
     this.canBeEdited = false,
     this.isAnonymous = false,
     this.canManageChat = false,

diff --git a/lib/src/telegram/models/chat_member_banned.dart b/lib/src/telegram/models/chat_member_banned.dart
@@ -27,15 +27,15 @@ part of '../model.dart';
 @JsonSerializable(fieldRename: FieldRename.snake)
 class ChatMemberBanned implements ChatMember {
   @override
-  String status;
+  ChatMemberStatus status;
   @override
   User user;
   int untilDate;
 
   ChatMemberBanned({
-    required this.status,
     required this.user,
     required this.untilDate,
+    this.status = ChatMemberStatus.kicked,
   });
 
   @JsonKey(includeFromJson: false, includeToJson: false)

diff --git a/lib/src/telegram/models/chat_member_left.dart b/lib/src/telegram/models/chat_member_left.dart
@@ -27,13 +27,13 @@ part of '../model.dart';
 @JsonSerializable(fieldRename: FieldRename.snake)
 class ChatMemberLeft implements ChatMember {
   @override
-  String status;
+  ChatMemberStatus status;
   @override
   User user;
 
   ChatMemberLeft({
-    required this.status,
     required this.user,
+    this.status = ChatMemberStatus.left,
   });
 
   factory ChatMemberLeft.fromJson(Map<String, dynamic> json) =>

diff --git a/lib/src/telegram/models/chat_member_member.dart b/lib/src/telegram/models/chat_member_member.dart
@@ -27,13 +27,13 @@ part of '../model.dart';
 @JsonSerializable(fieldRename: FieldRename.snake)
 class ChatMemberMember implements ChatMember {
   @override
-  String status;
+  ChatMemberStatus status;
   @override
   User user;
 
   ChatMemberMember({
-    required this.status,
     required this.user,
+    this.status = ChatMemberStatus.member,
   });
 
   factory ChatMemberMember.fromJson(Map<String, dynamic> json) =>

diff --git a/lib/src/telegram/models/chat_member_owner.dart b/lib/src/telegram/models/chat_member_owner.dart
@@ -26,16 +26,16 @@ part of '../model.dart';
 @JsonSerializable(fieldRename: FieldRename.snake)
 class ChatMemberOwner implements ChatMember {
   @override
-  String status;
+  ChatMemberStatus status;
   @override
   User user;
   bool isAnonymous;
   String? customTitle;
 
   ChatMemberOwner({
     required this.user,
-    required this.status,
     required this.isAnonymous,
+    this.status = ChatMemberStatus.creator,
     this.customTitle,
   });
 

diff --git a/lib/src/telegram/models/chat_member_restricted.dart b/lib/src/telegram/models/chat_member_restricted.dart
@@ -27,7 +27,7 @@ part of '../model.dart';
 @JsonSerializable(fieldRename: FieldRename.snake)
 class ChatMemberRestricted implements ChatMember {
   @override
-  String status;
+  ChatMemberStatus status;
   @override
   User user;
   bool isMember;
@@ -48,8 +48,8 @@ class ChatMemberRestricted implements ChatMember {
   int untilDate;
 
   ChatMemberRestricted({
-    required this.status,
     required this.user,
+    this.status = ChatMemberStatus.restricted,
     this.isMember = false,
     this.canChangeInfo = false,
     this.canInviteUsers = false,