Skip to content

Commit

Permalink
Engine-Twitter: added index support for /reply and /retweet commands
Browse files Browse the repository at this point in the history
  • Loading branch information
@meebey
meebey committed Nov 27, 2013
1 parent e4b0e82 commit 046ba2c
Showing 1 changed file with 72 additions and 13 deletions.
85 changes: 72 additions & 13 deletions src/Engine-Twitter/Protocols/Twitter/TwitterProtocolManager.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -82,6 +82,9 @@ public class TwitterProtocolManager : ProtocolManagerBase
int ErrorResponseCount { get; set; }
const int MaxErrorResponseCount = 3;

TwitterStatus[] StatusIndex { get; set; }
int StatusIndexOffset { get; set; }

public override string NetworkID {
get {
if (f_TwitterUser == null) {
Expand Down Expand Up @@ -148,6 +151,8 @@ public TwitterProtocolManager(Session session) : base(session)
);
f_DirectMessagesChat.ApplyConfig(Session.UserConfig);
f_GroupChats.Add(f_DirectMessagesChat);

StatusIndex = new TwitterStatus[99];
}

public override void Connect(FrontendManager fm, ServerModel server)
Expand Down Expand Up @@ -697,8 +702,8 @@ public void CommandHelp(CommandModel cd)
"follow screen-name|user-id",
"unfollow screen-name|user-id",
"search keyword",
"retweet/rt tweet-id",
"reply tweet-id message",
"retweet/rt index-number|tweet-id",
"reply index-number|tweet-id message",
};

foreach (string line in help) {
Expand Down Expand Up @@ -924,6 +929,7 @@ public void CommandTimeline(CommandModel cmd)

var sortedStatuses = SortTimeline(statuses);
foreach (var status in sortedStatuses) {
AddIndexToStatus(status);
var msg = CreateMessageBuilder().
Append(status, GetPerson(status.User)).ToMessage();
chat.MessageBuffer.Add(msg);
Expand Down Expand Up @@ -1075,6 +1081,7 @@ public void CommandSearch(CommandModel cmd)
var search = response.ResponseObject;
var sortedSearch = SortTimeline(search);
foreach (var status in sortedSearch) {
AddIndexToStatus(status);
var msg = CreateMessageBuilder().
Append(status, GetPerson(status.User)).
ToMessage();
Expand All @@ -1094,13 +1101,25 @@ public void CommandRetweet(CommandModel cmd)
return;
}


TwitterStatus status = null;
int indexId;
if (Int32.TryParse(cmd.Parameter, out indexId)) {
status = GetStatusFromIndex(indexId);
}

decimal statusId;
if (!Decimal.TryParse(cmd.Parameter, out statusId)) {
return;
if (status == null) {
if (!Decimal.TryParse(cmd.Parameter, out statusId)) {
return;
}
} else {
statusId = status.Id;
}
var response = TwitterStatus.Retweet(f_OAuthTokens, statusId, f_OptionalProperties);
CheckResponse(response);
var status = response.ResponseObject;
status = response.ResponseObject;

var msg = CreateMessageBuilder().
Append(status, GetPerson(status.User)).
ToMessage();
Expand All @@ -1114,21 +1133,31 @@ public void CommandReply(CommandModel cmd)
return;
}

var id = cmd.DataArray[1];
TwitterStatus status = null;
int indexId;
if (Int32.TryParse(id, out indexId)) {
status = GetStatusFromIndex(indexId);
}

decimal statusId;
if (!Decimal.TryParse(cmd.DataArray[1], out statusId)) {
return;
if (status == null) {
if (!Decimal.TryParse(id, out statusId)) {
return;
}
var response = TwitterStatus.Show(f_OAuthTokens, statusId,
f_OptionalProperties);
CheckResponse(response);
status = response.ResponseObject;
}
var response = TwitterStatus.Show(f_OAuthTokens, statusId,
f_OptionalProperties);
CheckResponse(response);
var status = response.ResponseObject;

var text = String.Join(" ", cmd.DataArray.Skip(2).ToArray());
// the screen name must be somewhere in the message for replies
if (!text.Contains("@" + status.User.ScreenName)) {
text = String.Format("@{0} {1}", status.User.ScreenName, text);
}
var options = CreateOptions<StatusUpdateOptions>();
options.InReplyToStatusId = statusId;
options.InReplyToStatusId = status.Id;
PostUpdate(text, options);
}

Expand Down Expand Up @@ -1235,6 +1264,7 @@ private void UpdateFriendsTimeline()

List<TwitterStatus> sortedTimeline = SortTimeline(timeline);
foreach (TwitterStatus status in sortedTimeline) {
AddIndexToStatus(status);
var msg = CreateMessageBuilder().
Append(status, GetPerson(status.User)).
ToMessage();
Expand Down Expand Up @@ -1345,6 +1375,7 @@ private void UpdateReplies()
bool highlight = f_LastReplyStatusID != 0;
List<TwitterStatus> sortedTimeline = SortTimeline(timeline);
foreach (TwitterStatus status in sortedTimeline) {
AddIndexToStatus(status);
var msg = CreateMessageBuilder().
Append(status, GetPerson(status.User), highlight).
ToMessage();
Expand Down Expand Up @@ -1643,7 +1674,35 @@ private void SendMessage(string target, string text)
CheckResponse(res);
f_DirectMessageEvent.Set();
}


void AddIndexToStatus(TwitterStatus status)
{
lock (StatusIndex) {
var slot = ++StatusIndexOffset;
if (slot > StatusIndex.Length) {
StatusIndexOffset = 0;
slot = 1;
}
StatusIndex[slot - 1] = status;
status.Text = String.Format("[{0:00}] {1}", slot, status.Text);
var rtStatus = status.RetweetedStatus;
if (rtStatus != null) {
rtStatus.Text = String.Format("[{0:00}] {1}", slot,
rtStatus.Text);
}
}
}

TwitterStatus GetStatusFromIndex(int slot)
{
lock (StatusIndex) {
if (slot > StatusIndex.Length || slot < 1) {
return null;
}
return StatusIndex[slot - 1];
}
}

private void CheckTwitterizerException(TwitterizerException exception)
{
Trace.Call(exception == null ? null : exception.GetType());
Expand Down

0 comments on commit 046ba2c

Please sign in to comment.