79 lines
2.9 KiB
C#
Raw Normal View History

2024-02-21 11:38:17 +01:00
using Orleans.Concurrency;
using Orleans.Runtime;
using Orleans.Streams;
using Orleans.Timers;
namespace Microsoft.AI.DevTeam;
[StatelessWorker]
[ImplicitStreamSubscription("Hubber")]
public class Hubber : Grain, IGrainWithStringKey
{
private readonly IManageGithub _ghService;
public Hubber( IManageGithub ghService)
{
_ghService = ghService;
}
public async override Task OnActivateAsync(CancellationToken cancellationToken)
{
var streamProvider = this.GetStreamProvider("StreamProvider");
var streamId = StreamId.Create("Hubber", this.GetPrimaryKeyString());
var stream = streamProvider.GetStream<Event>(streamId);
await stream.SubscribeAsync(HandleEvent);
}
public async Task HandleEvent(Event item, StreamSequenceToken? token)
{
switch (item.Type)
{
case EventType.NewAsk:
2024-02-22 11:17:31 +01:00
var org = item.Data["org"];
var repo = item.Data["repo"];
var input = item.Message;
var parentNumber = long.Parse(item.Data["parentNumber"]);
var pmIssue = await CreateIssue(org, repo, input, "", parentNumber);
var devLeadIssue = await CreateIssue(org, repo, input, "", parentNumber);
2024-02-22 13:09:48 +01:00
// TODO: store the mapping of parent/child?
break;
case EventType.ReadmeGenerated:
// _ghService.PostComment(item.Data["org"], item.Data["repo"], long.Parse(item.Data["issueNumber"]), item.Message);
2024-02-21 11:38:17 +01:00
break;
2024-02-22 13:09:48 +01:00
case EventType.DevPlanGenerated:
// _ghService.PostComment(item.Data["org"], item.Data["repo"], long.Parse(item.Data["issueNumber"]), item.Message);
2024-02-21 11:38:17 +01:00
break;
2024-02-22 13:09:48 +01:00
case EventType.CodeGenerated:
// _ghService.PostComment(item.Data["org"], item.Data["repo"], long.Parse(item.Data["issueNumber"]), item.Message);
break;
case EventType.DevPlanChainClosed:
// for each step, create Dev issue
//var devIssues = await CreateIssue(org, repo, input, "", parentNumber);
2024-02-21 11:38:17 +01:00
break;
default:
break;
}
}
2024-02-22 13:09:48 +01:00
2024-02-22 11:17:31 +01:00
public async Task<NewIssueResponse> CreateIssue(string org, string repo, string input, string function, long parentNumber)
2024-02-22 10:32:46 +01:00
{
2024-02-22 11:17:31 +01:00
return await _ghService.CreateIssue(org, repo,input,function,parentNumber);
2024-02-22 10:32:46 +01:00
}
public async Task PostComment(string org, string repo,long issueNumber, string comment )
{
await _ghService.PostComment(org, repo, issueNumber, comment);
}
public async Task CreateBranch()
{
// await _ghService.CreateBranch();
}
public async Task CreatePullRequest()
{
// await _ghService.CreatePR();
}
public async Task CommitToBranch()
2024-02-21 11:38:17 +01:00
{
2024-02-22 10:32:46 +01:00
// await _ghService.CommitToBranch()
2024-02-21 11:38:17 +01:00
}
}