✨ parametrize labels

Author: ad2ien

.github/workflows/build.yml

@@ -11,7 +11,7 @@ jobs:
       contents: read
       packages: write
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v6
 
       - name: lint
         uses: golangci/golangci-lint-action@v8

README.md

@@ -10,11 +10,27 @@
 
 Given a github token and matrix credentials
 
-- Once a day, check `help-wanted` issues of the repo stared by the user
+- Once a day, check configured labels `help-wanted` issues of the repo stared by the user
 - Update info in an sqlite database
 - Send message to a matrix room with link and title
 - Serves an interface listing open help wanting issues
 
+## Configuration
+
+```.env
+GITHUB_TOKEN=
+
+MATRIX_HOMESERVER=
+MATRIX_USERNAME=
+MATRIX_PASSWORD=
+MATRIX_ROOMID = 
+
+DB_FILE=
+
+# coma separated labels with quotes to look for
+LABELS='"help-wanted","junior friendly","good first issue"'
+```
+
 ## How
 
 Github graphql request
@@ -27,7 +43,7 @@ Github graphql request
         nameWithOwner
         description
         stargazerCount
-        issues(states: OPEN, labels: ["help-wanted"], first: 5) {
+        issues(states: OPEN, labels: ["$labels"], first: 5) {
           nodes {
             title
             url
@@ -74,8 +90,3 @@ Create migration
 ```bash
 docker run -v $(pwd)/migrations:/migrations --network host migrate/migrate -path=/migrations -database "sqlite://db/help-the-stars.db" create -ext sql -dir /migrations -seq MIGRATION_NAME
 ```
-
-## todo
-
-- [ ] parametrize labels Junior friendly, good first issue
-- [ ] golangci-lint

internal/datacontroller.go

@@ -66,7 +66,7 @@ func (d *DataController) Worker() {
 			d.GetAndSaveIssues()
 
 		} else {
-			fmt.Print(".")
+			log.Debug(".")
 			time.Sleep(time.Duration(internalSeconds) * time.Millisecond)
 		}
 	}

internal/stars.go

@@ -14,6 +14,10 @@ import (
 	"golang.org/x/oauth2"
 )
 
+const MAX_RETRY = 3
+const BACKOFF_DELAY = 10 * time.Second
+const MAX_ISSUES_PER_REPO = 5
+
 type GhIssue struct {
 	Title     string
 	Url       string
@@ -44,17 +48,17 @@ type GhQuery struct {
 	}
 }
 
-const graphqlUrl = "https://api.github.com/graphql"
+const GRAPHQL_URL = "https://api.github.com/graphql"
 
-const graphqlTemplate = `
+const GRAPHQL_TEMPLATE = `
 {
   viewer {
     starredRepositories(first: 50, after: "{{.RepoCursor}}") {
       nodes {
         nameWithOwner
         description
         stargazerCount
-        issues(states: OPEN, labels: ["help-wanted"], first: 5) {
+        issues(states: OPEN, labels: [{{.Labels}}], first: {{.MaxIssues}}) {
           nodes {
             title
             labels(first: 5) {
@@ -81,14 +85,19 @@ const graphqlTemplate = `
   }
 }
 `
+
 // Parse the template once at package initialization
-var tmpl = template.Must(template.New("graphql").Parse(graphqlTemplate))
+var tmpl = template.Must(template.New("graphql").Parse(GRAPHQL_TEMPLATE))
 
 func buildQueryFromTemplate(repoCursor string) (string, error) {
 	data := struct {
 		RepoCursor string
+		Labels     string
+		MaxIssues  int
 	}{
 		RepoCursor: repoCursor,
+		Labels:     GetSetting("LABELS"),
+		MaxIssues:  MAX_ISSUES_PER_REPO,
 	}
 
 	var query bytes.Buffer
@@ -145,7 +154,7 @@ func fetchQueryResults(cursor string) (GhQuery, error) {
 	}
 
 	// Create the HTTP request
-	req, err := http.NewRequest("POST", graphqlUrl, bytes.NewBuffer(requestBody))
+	req, err := http.NewRequest("POST", GRAPHQL_URL, bytes.NewBuffer(requestBody))
 	if err != nil {
 		log.Fatalf("Error creating request: %v", err)
 	}
@@ -154,11 +163,11 @@ func fetchQueryResults(cursor string) (GhQuery, error) {
 	req.Header.Set("Content-Type", "application/json")
 
 	// Send the request
-	resp, err := httpClient.Do(req)
+	resp, err := doWithRetry(httpClient, req)
 	if err != nil {
 		log.Fatalf("Error sending request: %v", err)
 	}
-	defer resp.Body.Close()
+	defer closeBody(resp.Body)
 
 	// Read the response
 	body, err := io.ReadAll(resp.Body)
@@ -174,3 +183,29 @@ func fetchQueryResults(cursor string) (GhQuery, error) {
 
 	return queryResult, nil
 }
+
+func doWithRetry(http *http.Client, req *http.Request) (*http.Response, error) {
+	i := 1
+	for {
+		resp, err := http.Do(req)
+		if err == nil {
+			if resp.StatusCode >= 500 {
+				log.Fatalf("Github server error: %d", resp.StatusCode)
+			}
+			return resp, nil
+		}
+
+		i++
+		if i >= MAX_RETRY {
+			return nil, err
+		}
+		time.Sleep(BACKOFF_DELAY * time.Duration(i))
+
+	}
+}
+
+func closeBody(body io.ReadCloser) {
+	if err := body.Close(); err != nil {
+		log.Warn("Error closing response body: %v", err)
+	}
+}

main.go

@@ -20,6 +20,12 @@ func main() {
 	flag.Usage = func() {
 		fmt.Println("Usage of Help the start ⭐")
 		fmt.Println("  https://github.com/ad2ien/help-the-stars/")
+		fmt.Println(".env example :")
+		fmt.Println("\tMATRIX_TOKEN=your-matrix-token")
+		fmt.Println("\tMATRIX_USERID=your-matrix-userid")
+		fmt.Println("\tMATRIX_ROOMID=your-matrix-roomid")
+		fmt.Println("\tDB_FILE=db/help-the-stars-dev.db")
+		fmt.Println("\tLABELS='\"help-wanted\",\"junior friendly\",\"good first issue\"'")
 		fmt.Println("\nFlags:")
 		flag.PrintDefaults()
 	}