Cleaning up the code, adding lint and build workflows, adding makefil…

…es for build, lint, and test.

Signed-off-by: ytimocin <ytimocin@microsoft.com>

.github/scripts/install.sh

@@ -19,7 +19,7 @@ ARCH=$(uname -m)
 
 if [[ "$ARCH" == "x86_64" ]]; then
   ARCH="amd64"
-elif [[ "$ARCH" == "aarch64" ]]; then
+elif [[ "$ARCH" == "arm64" || "$ARCH" == "aarch64" ]]; then
   ARCH="arm64"
 else
   echo "Unsupported architecture: $ARCH"

.github/workflows/build.yml

@@ -0,0 +1,61 @@
+name: Build
+
+on:
+  push:
+    branches:
+      - main
+      - release/*
+    tags:
+      - "v*.*.*"
+  pull_request:
+    branches:
+      - main
+      - release/*
+
+permissions:
+  contents: write
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        goos: [linux, darwin, windows]
+        goarch: [amd64, arm64]
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Set up Go
+        uses: actions/setup-go@v5
+        with:
+          go-version: 1.22
+          cache: false
+
+      - name: Install dependencies
+        run: go mod tidy
+
+      - name: Run tests
+        run: go test -v ./...
+
+      - name: Build binaries
+        run: |
+          echo "Building for GOOS=${{ matrix.goos }} GOARCH=${{ matrix.goarch }}"
+          GOOS=${{ matrix.goos }} GOARCH=${{ matrix.goarch }} \
+          make build
+
+      - name: Rename Windows binaries with .exe
+        if: matrix.goos == 'windows'
+        run: mv dist/pops-windows-${{ matrix.goarch }} dist/pops-windows-${{ matrix.goarch }}.exe
+
+      - name: Validate binary
+        run: |
+          file dist/pops-${{ matrix.goos }}-${{ matrix.goarch }}*
+
+      - name: Upload binaries as artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: pops-${{ matrix.goos }}-${{ matrix.goarch }}
+          path: dist/*

.github/workflows/release.yml

@@ -30,14 +30,14 @@ jobs:
       - name: Install dependencies
         run: go mod tidy
 
-      - name: Create dist directory
-        run: mkdir -p dist
+      - name: Run tests
+        run: go test -v ./...
 
       - name: Build binaries
         run: |
           echo "Building for GOOS=${{ matrix.goos }} GOARCH=${{ matrix.goarch }}"
-          GOOS=${{ matrix.goos }} GOARCH=${{ matrix.goarch }} CGO_ENABLED=0 GO111MODULE=on \
-          go build -ldflags="-s -w" -o dist/pops-${{ matrix.goos }}-${{ matrix.goarch }}
+          GOOS=${{ matrix.goos }} GOARCH=${{ matrix.goarch }} \
+          make build
 
       - name: Rename Windows binaries with .exe
         if: matrix.goos == 'windows'
@@ -51,7 +51,7 @@ jobs:
         uses: actions/upload-artifact@v4
         with:
           name: pops-${{ matrix.goos }}-${{ matrix.goarch }}
-          path: dist/
+          path: dist/pops-${{ matrix.goos }}-${{ matrix.goarch }}*
 
   release:
     runs-on: ubuntu-latest
@@ -64,11 +64,11 @@ jobs:
       - name: Download build artifacts
         uses: actions/download-artifact@v4
         with:
-          path: release
+          path: dist/
           pattern: pops-*
 
-      - name: List release folder
-        run: ls -R release
+      - name: List dist folder
+        run: ls -R dist/
 
       - name: Create GitHub Release
         id: create_release
@@ -77,6 +77,6 @@ jobs:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         with:
           repository: prompt-ops/pops
-          files: release/*
+          files: dist/pops-*
           draft: false
           prerelease: false

.gitignore

@@ -35,3 +35,6 @@ Thumbs.db
 
 # `pops` binary
 pops
+
+# dist directory
+dist/

Makefile

@@ -0,0 +1,2 @@
+# Include all make files in the make directory
+include $(wildcard make/*.mk)

ai/ai.go

@@ -108,10 +108,11 @@ func parseResponse(response string) (ParsedResponse, error) {
 // parseSuggestions extracts the suggestions from the response.
 func parseSuggestions(lines []string) []string {
 	var suggestions []string
+	re := regexp.MustCompile(`^\d+\.\s+`)
 	for _, line := range lines {
 		line = strings.TrimSpace(line)
 		// Match numbered suggestions (e.g., "1. Describe one of the pods")
-		if matched, _ := regexp.MatchString(`^\d+\.\s+`, line); matched {
+		if matched := re.MatchString(line); matched {
 			suggestions = append(suggestions, line)
 		}
 	}

cmd/connection/cloud/create.go

@@ -27,9 +27,9 @@ func (m *createModel) Init() tea.Cmd {
 }
 
 func (m *createModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
-	switch msg.(type) {
+	switch msg := msg.(type) {
 	case ui.TransitionToShellMsg:
-		shellModel := ui.NewShellModel(msg.(ui.TransitionToShellMsg).Connection)
+		shellModel := ui.NewShellModel(msg.Connection)
 		m.current = shellModel
 		return m, shellModel.Init()
 	}

cmd/connection/cloud/open.go

@@ -28,9 +28,9 @@ func (m *openModel) Init() tea.Cmd {
 }
 
 func (m *openModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
-	switch msg.(type) {
+	switch msg := msg.(type) {
 	case ui.TransitionToShellMsg:
-		m.current = ui.NewShellModel(msg.(ui.TransitionToShellMsg).Connection)
+		m.current = ui.NewShellModel(msg.Connection)
 		return m, m.current.Init()
 	}
 	var cmd tea.Cmd

cmd/connection/create.go

@@ -104,10 +104,10 @@ func (m *createModel) Init() tea.Cmd {
 func (m *createModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	switch m.currentStep {
 	case createStepTypeSelection:
-		switch msg.(type) {
+		switch msg := msg.(type) {
 		case ui.TransitionToCreateMsg:
 			fmt.Println("Transitioning to create")
-			connectionType := msg.(ui.TransitionToCreateMsg).ConnectionType
+			connectionType := msg.ConnectionType
 			createModel, err := factory.GetCreateModel(connectionType)
 			if err != nil {
 				return m, tea.Quit

cmd/connection/db/create.go

@@ -27,9 +27,9 @@ func (m *createModel) Init() tea.Cmd {
 }
 
 func (m *createModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
-	switch msg.(type) {
+	switch msg := msg.(type) {
 	case ui.TransitionToShellMsg:
-		shellModel := ui.NewShellModel(msg.(ui.TransitionToShellMsg).Connection)
+		shellModel := ui.NewShellModel(msg.Connection)
 		m.current = shellModel
 		return m, shellModel.Init()
 	}

cmd/connection/db/open.go

@@ -27,9 +27,9 @@ func (m *openModel) Init() tea.Cmd {
 }
 
 func (m *openModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
-	switch msg.(type) {
+	switch msg := msg.(type) {
 	case ui.TransitionToShellMsg:
-		m.current = ui.NewShellModel(msg.(ui.TransitionToShellMsg).Connection)
+		m.current = ui.NewShellModel(msg.Connection)
 		return m, m.current.Init()
 	}
 	var cmd tea.Cmd

cmd/connection/kubernetes/create.go

@@ -27,9 +27,9 @@ func (m *createModel) Init() tea.Cmd {
 }
 
 func (m *createModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
-	switch msg.(type) {
+	switch msg := msg.(type) {
 	case ui.TransitionToShellMsg:
-		shellModel := ui.NewShellModel(msg.(ui.TransitionToShellMsg).Connection)
+		shellModel := ui.NewShellModel(msg.Connection)
 		return shellModel, shellModel.Init()
 	}
 	var cmd tea.Cmd

cmd/connection/kubernetes/open.go

@@ -27,9 +27,9 @@ func (m *openModel) Init() tea.Cmd {
 }
 
 func (m *openModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
-	switch msg.(type) {
+	switch msg := msg.(type) {
 	case ui.TransitionToShellMsg:
-		m.current = ui.NewShellModel(msg.(ui.TransitionToShellMsg).Connection)
+		m.current = ui.NewShellModel(msg.Connection)
 		return m, m.current.Init()
 	}
 	var cmd tea.Cmd