@@ -62,6 +62,7 @@ func Answer(s *discordgo.Session, m *discordgo.MessageReactionAdd) {
62
62
log .Error (err )
63
63
}
64
64
Register .ChannelState .TypeTag = def
65
+ Register .Clear ()
65
66
}
66
67
67
68
NillType := func () {
@@ -96,57 +97,75 @@ func Answer(s *discordgo.Session, m *discordgo.MessageReactionAdd) {
96
97
97
98
if m .Emoji .MessageFormat () == config .Art {
98
99
if Register .ChannelState .TypeTag == config .LiveType {
100
+ Message ("[Info] you enable Livestream and Art type on this channel" )
99
101
Register .UpdateType (config .ArtNLiveType )
102
+ err := Register .UpdateChannel ("type" )
103
+ if err != nil {
104
+ s .ChannelMessageSend (m .ChannelID , err .Error ())
105
+ log .Error (err )
106
+ }
107
+ Register .Clear ()
108
+ return
100
109
} else if Register .ChannelState .TypeTag == config .LewdType {
110
+ Message ("[Info] you enable Lewd and Art type on this channel" )
101
111
Register .UpdateType (config .LewdNArtType )
102
112
} else if Register .ChannelState .TypeTag == config .ArtType {
103
113
NillType ()
104
114
return
105
115
} else if Register .ChannelState .TypeTag == config .ArtNLiveType {
106
116
Message ("[Info] you disable Art type on this channel" )
107
117
Register .UpdateType (config .LiveType )
108
- err := Register .UpdateChannel ()
118
+ err := Register .UpdateChannel ("type" )
109
119
if err != nil {
120
+ s .ChannelMessageSend (m .ChannelID , err .Error ())
110
121
log .Error (err )
111
122
}
123
+ Register .Clear ()
112
124
return
113
125
} else {
126
+ Message ("[Info] you enable Art type on this channel" )
114
127
Register .UpdateType (config .ArtType )
115
128
}
116
129
} else if m .Emoji .MessageFormat () == config .Live {
117
130
if Register .ChannelState .TypeTag == config .ArtType {
131
+ Message ("[Info] you enable Livestream and Art type on this channel" )
118
132
Register .UpdateType (config .ArtNLiveType )
119
133
} else if Register .ChannelState .TypeTag == config .LewdType {
120
134
LewdLive (config .LewdType )
135
+ return
121
136
} else if Register .ChannelState .TypeTag == config .LiveType {
122
137
NillType ()
123
138
return
124
139
} else if Register .ChannelState .TypeTag == config .ArtNLiveType {
125
140
Message ("[Info] you disable Live type on this channel" )
126
141
Register .UpdateType (config .ArtType )
127
- err := Register .UpdateChannel ()
142
+ err := Register .UpdateChannel ("type" )
128
143
if err != nil {
144
+ s .ChannelMessageSend (m .ChannelID , err .Error ())
129
145
log .Error (err )
130
146
}
131
147
return
132
148
} else {
149
+ Message ("[Info] you enable Livestream type on this channel" )
133
150
Register .UpdateType (config .LiveType )
134
151
}
135
152
} else if m .Emoji .MessageFormat () == config .Lewd {
136
153
if Register .ChannelState .TypeTag == config .LiveType {
137
154
LewdLive (config .LiveType )
155
+ return
138
156
} else if Register .ChannelState .TypeTag == config .ArtType {
139
157
Register .UpdateType (config .LewdNArtType )
140
158
} else if Register .ChannelState .TypeTag == config .ArtNLiveType {
141
159
LewdLive (config .ArtNLiveType )
160
+ return
142
161
} else if Register .ChannelState .TypeTag == config .LewdType {
143
162
NillType ()
144
163
return
145
164
} else if Register .ChannelState .TypeTag == config .LewdNArtType {
146
165
Message ("[Info] you disable Lewd type on this channel" )
147
166
Register .UpdateType (config .ArtType )
148
167
} else {
149
- Register .UpdateType (config .LewdType ) //nice
168
+ Register .UpdateType (config .LewdType )
150
169
}
151
170
}
152
171
@@ -168,8 +187,12 @@ func Answer(s *discordgo.Session, m *discordgo.MessageReactionAdd) {
168
187
}
169
188
}
170
189
171
- Register .UpdateChannel ()
172
- _ , err := s .ChannelMessageSend (m .ChannelID , "Done" )
190
+ err := Register .UpdateChannel ("all" )
191
+ if err != nil {
192
+ s .ChannelMessageSend (m .ChannelID , err .Error ())
193
+ log .Error (err )
194
+ }
195
+ _ , err = s .ChannelMessageSend (m .ChannelID , "Done" )
173
196
if err != nil {
174
197
log .Error (err )
175
198
}
@@ -183,8 +206,12 @@ func Answer(s *discordgo.Session, m *discordgo.MessageReactionAdd) {
183
206
Register .DelRegion (s )
184
207
} else if m .Emoji .MessageFormat () == config .Four {
185
208
LiveChange ()
186
- Register .UpdateChannel ()
187
- _ , err := s .ChannelMessageSend (m .ChannelID , "Done" )
209
+ err := Register .UpdateChannel ("all" )
210
+ if err != nil {
211
+ s .ChannelMessageSend (m .ChannelID , err .Error ())
212
+ log .Error (err )
213
+ }
214
+ _ , err = s .ChannelMessageSend (m .ChannelID , "Done" )
188
215
if err != nil {
189
216
log .Error (err )
190
217
}
@@ -367,11 +394,12 @@ func RegisterFunc(s *discordgo.Session, m *discordgo.MessageCreate) {
367
394
Register .SetAdmin (m .Author .ID ).SetChannel (m .ChannelID )
368
395
Register .ChannelStates = ChannelData
369
396
370
- Register .UpdateState ("SelectChannel" )
371
397
_ , err = s .ChannelMessageSend (m .ChannelID , "Select ID : " )
372
398
if err != nil {
373
399
log .Error (err )
374
400
}
401
+
402
+ Register .UpdateState ("SelectChannel" )
375
403
} else {
376
404
_ , err := s .ChannelMessageSend (m .ChannelID , "You don't have permission to enable/disable/update" )
377
405
if err != nil {
@@ -702,61 +730,42 @@ func (Data *Regis) IndieNotif(s *discordgo.Session) *Regis {
702
730
return Data
703
731
}
704
732
705
- func (Data * Regis ) UpdateChannel () error {
706
- ChannelID := Data .ChannelState .ChannelID
707
-
708
- err := Data .ChannelState .UpdateChannel ("Type" )
709
- if err != nil {
710
- _ , err := Bot .ChannelMessageSend (ChannelID , err .Error ())
733
+ func (Data * Regis ) UpdateChannel (s string ) error {
734
+ if s == "type" {
735
+ err := Data .ChannelState .UpdateChannel ("Type" )
711
736
if err != nil {
712
- log . Error ( err )
737
+ return err
713
738
}
714
- }
715
-
716
- err = Data .ChannelState .UpdateChannel ("LiveOnly" )
717
- if err != nil {
718
- _ , err := Bot .ChannelMessageSend (ChannelID , err .Error ())
739
+ } else {
740
+ err := Data .ChannelState .UpdateChannel ("LiveOnly" )
719
741
if err != nil {
720
- log . Error ( err )
742
+ return err
721
743
}
722
- }
723
744
724
- err = Data .ChannelState .UpdateChannel ("Dynamic" )
725
- if err != nil {
726
- _ , err := Bot .ChannelMessageSend (ChannelID , err .Error ())
745
+ err = Data .ChannelState .UpdateChannel ("Dynamic" )
727
746
if err != nil {
728
- log . Error ( err )
747
+ return err
729
748
}
730
- }
731
749
732
- if ! Register .ChannelState .LiveOnly {
733
- err = Data .ChannelState .UpdateChannel ("NewUpcoming" )
734
- if err != nil {
735
- _ , err := Bot .ChannelMessageSend (ChannelID , err .Error ())
750
+ if ! Register .ChannelState .LiveOnly {
751
+ err = Data .ChannelState .UpdateChannel ("NewUpcoming" )
736
752
if err != nil {
737
- log . Error ( err )
753
+ return err
738
754
}
739
755
}
740
- }
741
756
742
- err = Data .ChannelState .UpdateChannel ("LiteMode" )
743
- if err != nil {
744
- _ , err := Bot .ChannelMessageSend (ChannelID , err .Error ())
757
+ err = Data .ChannelState .UpdateChannel ("LiteMode" )
745
758
if err != nil {
746
- log . Error ( err )
759
+ return err
747
760
}
748
- }
749
761
750
- if Register .ChannelState .Group .GroupName == config .Indie {
751
- err = Data .ChannelState .UpdateChannel ("IndieNotif" )
752
- if err != nil {
753
- _ , err := Bot .ChannelMessageSend (ChannelID , err .Error ())
762
+ if Register .ChannelState .Group .GroupName == config .Indie {
763
+ err = Data .ChannelState .UpdateChannel ("IndieNotif" )
754
764
if err != nil {
755
- log . Error ( err )
765
+ return err
756
766
}
757
767
}
758
768
}
759
-
760
769
return nil
761
770
}
762
771
0 commit comments