wip

Author: ravilushqa

github.go

@@ -22,22 +22,32 @@ func NewGithubClient(ctx context.Context, token string) *GithubClient {
 	return &GithubClient{github.NewClient(tc)}
 }
 
-func (g *GithubClient) GetPullRequestFiles(ctx context.Context, owner, repo string, prNumber int) ([]*github.CommitFile, error) {
-	files, _, err := g.client.PullRequests.ListFiles(ctx, owner, repo, prNumber, nil)
+func (c *GithubClient) GetPullRequestFiles(ctx context.Context, owner, repo string, prNumber int) ([]*github.CommitFile, error) {
+	files, _, err := c.client.PullRequests.ListFiles(ctx, owner, repo, prNumber, nil)
 	return files, err
 }
 
-func (g *GithubClient) GetPullRequest(ctx context.Context, owner, repo string, number int) (*github.PullRequest, error) {
-	pr, _, err := g.client.PullRequests.Get(ctx, owner, repo, number)
+func (c *GithubClient) GetPullRequest(ctx context.Context, owner, repo string, number int) (*github.PullRequest, error) {
+	pr, _, err := c.client.PullRequests.Get(ctx, owner, repo, number)
 	return pr, err
 }
 
-func (g *GithubClient) GetPullRequestDiff(ctx context.Context, owner, repo string, number int) (string, error) {
-	diff, _, err := g.client.PullRequests.GetRaw(ctx, owner, repo, number, github.RawOptions{Type: github.Diff})
+func (c *GithubClient) GetPullRequestDiff(ctx context.Context, owner, repo string, number int) (string, error) {
+	diff, _, err := c.client.PullRequests.GetRaw(ctx, owner, repo, number, github.RawOptions{Type: github.Diff})
 	return diff, err
 }
 
-func (g *GithubClient) UpdatePullRequest(ctx context.Context, owner, repo string, number int, pr *github.PullRequest) (*github.PullRequest, error) {
-	updatedPR, _, err := g.client.PullRequests.Edit(ctx, owner, repo, number, pr)
+func (c *GithubClient) UpdatePullRequest(ctx context.Context, owner, repo string, number int, pr *github.PullRequest) (*github.PullRequest, error) {
+	updatedPR, _, err := c.client.PullRequests.Edit(ctx, owner, repo, number, pr)
 	return updatedPR, err
 }
+
+func (c *GithubClient) CreateComment(ctx context.Context, owner, repo string, number int, comment *github.PullRequestComment) (*github.PullRequestComment, error) {
+	createdComment, _, err := c.client.PullRequests.CreateComment(ctx, owner, repo, number, comment)
+	return createdComment, err
+}
+
+func (c *GithubClient) CompareCommits(ctx context.Context, owner, repo, base, head string) (*github.CommitsComparison, error) {
+	comp, _, err := c.client.Repositories.CompareCommits(ctx, owner, repo, base, head, nil)
+	return comp, err
+}

main.go

@@ -57,11 +57,14 @@ func main() {
 		fileName := getFilenameFromDiffHeader(fileDiff.Header)
 
 		prompt := fmt.Sprintf("File %s:\n%s\n%s\n", fileName, fileDiff.Header, fileDiff.Diff)
-		messages = append(messages, openai.ChatCompletionMessage{
+		messages = append(messages, openai.CompletionRequest{
 			Role:    openai.ChatMessageRoleUser,
 			Content: prompt,
 		})
+
+		fmt.Println(fileName, len(prompt))
 	}
+	os.Exit(0)
 	chatGPTDescription, err := openaiClient.ChatCompletion(context.Background(), messages)
 	if err != nil {
 		fmt.Printf("Error generating pull request description: %v\n", err)
@@ -74,6 +77,14 @@ func main() {
 		return
 	}
 
+	//compare base branch with head branch
+	commitsComparison, err := githubClient.CompareCommits(context.Background(), opts.Owner, opts.Repo, pr.GetBase().GetRef(), pr.GetHead().GetRef())
+	if err != nil {
+		return
+	}
+
+	_ = commitsComparison
+
 	jiraLink := generateJiraLinkByTitle(*pr.Title)
 
 	description := fmt.Sprintf("### Jira\n%s\n%s", jiraLink, chatGPTDescription)

openai.go

@@ -8,7 +8,8 @@ import (
 )
 
 const (
-	PromptUpdateDesription = "\"Generate a GitHub pull request description based on the following changes without basic prefix in markdown format with ###Description and ###Changes blocks:\\n\""
+	PromptUpdateDescription = "\"Generate a GitHub pull request description based on the following changes without basic prefix in markdown format with ###Description and ###Changes blocks:\\n\""
+	PromptDescribeChanges   = "Bellow is the code patch, please describe the changes in the following format: **file name**:\\n###Description\\n###Changes\\n"
 )
 
 type OpenAIClient struct {
@@ -22,8 +23,9 @@ func NewOpenAIClient(token string) *OpenAIClient {
 }
 
 func (o *OpenAIClient) ChatCompletion(ctx context.Context, messages []openai.ChatCompletionMessage) (string, error) {
+	return "", nil
 	resp, err := o.client.CreateChatCompletion(
-		context.Background(),
+		ctx,
 		openai.ChatCompletionRequest{
 			Model:    openai.GPT3Dot5Turbo,
 			Messages: messages,