Merge pull request #1122 from wakatime/develop

Release v1.1.04.0

.github/workflows/on_push.yml

@@ -34,7 +34,7 @@ jobs:
         run: make test
       -
         name: Linter
-        uses: golangci/golangci-lint-action@v3
+        uses: golangci/golangci-lint-action@v6
         with:
           version: latest
           skip-cache: true
@@ -94,7 +94,7 @@ jobs:
         run: make test
       -
         name: Linter
-        uses: golangci/golangci-lint-action@v3
+        uses: golangci/golangci-lint-action@v6
         with:
           version: latest
           skip-cache: true
@@ -145,7 +145,7 @@ jobs:
         run: make test
       -
         name: Linter
-        uses: golangci/golangci-lint-action@v3
+        uses: golangci/golangci-lint-action@v6
         with:
           version: latest
           skip-cache: true
@@ -574,7 +574,7 @@ jobs:
           path: build/
       -
         name: Import Code-Signing Certificates
-        uses: Apple-Actions/import-codesign-certs@v1
+        uses: Apple-Actions/import-codesign-certs@v3
         with:
           # The certificates in a PKCS12 file encoded as a base64 string
           p12-file-base64: ${{ secrets.APPLE_DEVELOPER_CERTIFICATE_P12_BASE64 }}
@@ -644,7 +644,7 @@ jobs:
         # Run only for develop branch
         if: ${{ github.ref == 'refs/heads/develop' }}
         name: Changelog for develop
-        uses: gandarez/changelog-action@v1.2.0
+        uses: gandarez/changelog-action@v1.4.0
         id: changelog-develop
         with:
           current_tag: ${{ github.sha }}
@@ -655,7 +655,7 @@ jobs:
         # Run only for release branch
         if: ${{ github.ref == 'refs/heads/release' }}
         name: Get related pull request
-        uses: 8BitJonny/gh-get-current-pr@v2.2.0
+        uses: 8BitJonny/gh-get-current-pr@3.0.0
         id: changelog-release
         with:
           github-token: ${{ secrets.GITHUB_TOKEN }}
@@ -681,7 +681,7 @@ jobs:
         run: ./bin/prepare_assets.sh
       -
         name: "Create release"
-        uses: softprops/action-gh-release@master
+        uses: softprops/action-gh-release@v2
         with:
           name: ${{ needs.version.outputs.semver_tag }}
           tag_name: ${{ needs.version.outputs.semver_tag }}

cmd/api/api.go

@@ -1,6 +1,7 @@
 package api
 
 import (
+	"context"
 	"fmt"
 	"strings"
 
@@ -13,31 +14,33 @@ import (
 
 // NewClient initializes a new api client with all options following the
 // passed in parameters.
-func NewClient(params paramscmd.API) (*api.Client, error) {
+func NewClient(ctx context.Context, params paramscmd.API) (*api.Client, error) {
 	withAuth, err := api.WithAuth(api.BasicAuth{
 		Secret: params.Key,
 	})
 	if err != nil {
 		return nil, fmt.Errorf("failed to set up auth option on api client: %w", err)
 	}
 
-	return newClient(params, withAuth)
+	return newClient(ctx, params, withAuth)
 }
 
 // NewClientWithoutAuth initializes a new api client with all options following the
 // passed in parameters and disabled authentication.
-func NewClientWithoutAuth(params paramscmd.API) (*api.Client, error) {
-	return newClient(params)
+func NewClientWithoutAuth(ctx context.Context, params paramscmd.API) (*api.Client, error) {
+	return newClient(ctx, params)
 }
 
 // newClient contains the logic of client initialization, except auth initialization.
-func newClient(params paramscmd.API, opts ...api.Option) (*api.Client, error) {
+func newClient(ctx context.Context, params paramscmd.API, opts ...api.Option) (*api.Client, error) {
 	opts = append(opts, api.WithTimeout(params.Timeout))
 	opts = append(opts, api.WithHostname(strings.TrimSpace(params.Hostname)))
 
+	logger := log.Extract(ctx)
+
 	tz, err := timezone()
 	if err != nil {
-		log.Debugf("failed to detect local timezone: %s", err)
+		logger.Debugf("failed to detect local timezone: %s", err)
 	} else {
 		opts = append(opts, api.WithTimezone(strings.TrimSpace(tz)))
 	}
@@ -54,7 +57,7 @@ func newClient(params paramscmd.API, opts ...api.Option) (*api.Client, error) {
 
 		opts = append(opts, withSSLCert)
 	} else if !params.DisableSSLVerify {
-		opts = append(opts, api.WithSSLCertPool(api.CACerts()))
+		opts = append(opts, api.WithSSLCertPool(api.CACerts(ctx)))
 	}
 
 	if params.ProxyURL != "" {
@@ -66,7 +69,7 @@ func newClient(params paramscmd.API, opts ...api.Option) (*api.Client, error) {
 		opts = append(opts, withProxy)
 
 		if strings.Contains(params.ProxyURL, `\\`) {
-			withNTLMRetry, err := api.WithNTLMRequestRetry(params.ProxyURL)
+			withNTLMRetry, err := api.WithNTLMRequestRetry(ctx, params.ProxyURL)
 			if err != nil {
 				return nil, fmt.Errorf("failed to set up ntlm request retry option on api client: %w", err)
 			}
@@ -75,7 +78,7 @@ func newClient(params paramscmd.API, opts ...api.Option) (*api.Client, error) {
 		}
 	}
 
-	opts = append(opts, api.WithUserAgent(params.Plugin))
+	opts = append(opts, api.WithUserAgent(ctx, params.Plugin))
 
 	return api.NewClient(params.URL, opts...), nil
 }

cmd/configread/configread.go

@@ -1,6 +1,7 @@
 package configread
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"strings"
@@ -18,7 +19,7 @@ type Params struct {
 }
 
 // Run prints the value for the given config key.
-func Run(v *viper.Viper) (int, error) {
+func Run(_ context.Context, v *viper.Viper) (int, error) {
 	output, err := Read(v)
 	if err != nil {
 		return exitcode.ErrConfigFileRead, fmt.Errorf(

cmd/configwrite/configwrite.go

@@ -1,6 +1,7 @@
 package configwrite
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"strings"
@@ -19,16 +20,16 @@ type Params struct {
 }
 
 // Run loads wakatime config file and call Write().
-func Run(v *viper.Viper) (int, error) {
-	w, err := ini.NewWriter(v, ini.FilePath)
+func Run(ctx context.Context, v *viper.Viper) (int, error) {
+	w, err := ini.NewWriter(ctx, v, ini.FilePath)
 	if err != nil {
 		return exitcode.ErrConfigFileParse, fmt.Errorf(
 			"failed to parse config file: %s",
 			err,
 		)
 	}
 
-	if err := Write(v, w); err != nil {
+	if err := Write(ctx, v, w); err != nil {
 		return exitcode.ErrGeneric, fmt.Errorf(
 			"failed to write to config file: %s",
 			err,
@@ -39,13 +40,13 @@ func Run(v *viper.Viper) (int, error) {
 }
 
 // Write writes value(s) to given config key(s) and persist on disk.
-func Write(v *viper.Viper, w ini.Writer) error {
+func Write(ctx context.Context, v *viper.Viper, w ini.Writer) error {
 	params, err := LoadParams(v)
 	if err != nil {
 		return fmt.Errorf("failed to load command parameters: %w", err)
 	}
 
-	return w.Write(params.Section, params.KeyValue)
+	return w.Write(ctx, params.Section, params.KeyValue)
 }
 
 // LoadParams loads needed data from the configuration file.

cmd/configwrite/configwrite_test.go

@@ -1,6 +1,7 @@
 package configwrite_test
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"os"
@@ -72,8 +73,10 @@ func TestWrite(t *testing.T) {
 
 	defer tmpFile.Close()
 
+	ctx := context.Background()
+
 	v := viper.New()
-	ini, err := ini.NewWriter(v, func(vp *viper.Viper) (string, error) {
+	ini, err := ini.NewWriter(ctx, v, func(_ context.Context, vp *viper.Viper) (string, error) {
 		assert.Equal(t, v, vp)
 		return tmpFile.Name(), nil
 	})
@@ -82,7 +85,7 @@ func TestWrite(t *testing.T) {
 	v.Set("config-section", "settings")
 	v.Set("config-write", map[string]string{"debug": "false"})
 
-	err = configwrite.Write(v, ini)
+	err = configwrite.Write(ctx, v, ini)
 	require.NoError(t, err)
 
 	err = ini.File.Reload()
@@ -117,7 +120,7 @@ func TestWriteErr(t *testing.T) {
 			v.Set("config-section", test.Section)
 			v.Set("config-write", test.Value)
 
-			err := configwrite.Write(v, w)
+			err := configwrite.Write(context.Background(), v, w)
 			require.Error(t, err)
 
 			assert.Equal(
@@ -133,7 +136,7 @@ func TestWriteErr(t *testing.T) {
 func TestWriteSaveErr(t *testing.T) {
 	v := viper.New()
 	w := &mockWriter{
-		WriteFn: func(section string, keyValue map[string]string) error {
+		WriteFn: func(_ context.Context, section string, keyValue map[string]string) error {
 			assert.Equal(t, "settings", section)
 			assert.Equal(t, map[string]string{"debug": "false"}, keyValue)
 
@@ -144,14 +147,14 @@ func TestWriteSaveErr(t *testing.T) {
 	v.Set("config-section", "settings")
 	v.Set("config-write", map[string]string{"debug": "false"})
 
-	err := configwrite.Write(v, w)
+	err := configwrite.Write(context.Background(), v, w)
 	assert.Error(t, err)
 }
 
 type mockWriter struct {
-	WriteFn func(section string, keyValue map[string]string) error
+	WriteFn func(ctx context.Context, section string, keyValue map[string]string) error
 }
 
-func (m *mockWriter) Write(section string, keyValue map[string]string) error {
-	return m.WriteFn(section, keyValue)
+func (m *mockWriter) Write(ctx context.Context, section string, keyValue map[string]string) error {
+	return m.WriteFn(ctx, section, keyValue)
 }

cmd/fileexperts/fileexperts.go

@@ -1,6 +1,7 @@
 package fileexperts
 
 import (
+	"context"
 	"fmt"
 
 	apicmd "github.com/wakatime/wakatime-cli/cmd/api"
@@ -18,8 +19,8 @@ import (
 )
 
 // Run executes the file-experts command.
-func Run(v *viper.Viper) (int, error) {
-	output, err := FileExperts(v)
+func Run(ctx context.Context, v *viper.Viper) (int, error) {
+	output, err := FileExperts(ctx, v)
 	if err != nil {
 		if errwaka, ok := err.(wakaerror.Error); ok {
 			return errwaka.ExitCode(), fmt.Errorf("file experts fetch failed: %s", errwaka.Message())
@@ -31,29 +32,31 @@ func Run(v *viper.Viper) (int, error) {
 		)
 	}
 
-	log.Debugln("successfully fetched file experts")
+	logger := log.Extract(ctx)
+	logger.Debugln("successfully fetched file experts")
+
 	fmt.Println(output)
 
 	return exitcode.Success, nil
 }
 
 // FileExperts returns a rendered file experts of todays coding activity.
-func FileExperts(v *viper.Viper) (string, error) {
-	params, err := LoadParams(v)
+func FileExperts(ctx context.Context, v *viper.Viper) (string, error) {
+	params, err := LoadParams(ctx, v)
 	if err != nil {
 		return "", fmt.Errorf("failed to load command parameters: %w", err)
 	}
 
 	handleOpts := initHandleOptions(params)
 
-	apiClient, err := apicmd.NewClientWithoutAuth(params.API)
+	apiClient, err := apicmd.NewClientWithoutAuth(ctx, params.API)
 	if err != nil {
 		return "", fmt.Errorf("failed to initialize api client: %w", err)
 	}
 
 	handle := fileexperts.NewHandle(apiClient, handleOpts...)
 
-	results, err := handle([]heartbeat.Heartbeat{{Entity: params.Heartbeat.Entity}})
+	results, err := handle(ctx, []heartbeat.Heartbeat{{Entity: params.Heartbeat.Entity}})
 	if err != nil {
 		return "", err
 	}
@@ -75,17 +78,17 @@ func FileExperts(v *viper.Viper) (string, error) {
 
 // LoadParams loads file-expert config params from viper.Viper instance. Returns ErrAuth
 // if failed to retrieve api key.
-func LoadParams(v *viper.Viper) (paramscmd.Params, error) {
+func LoadParams(ctx context.Context, v *viper.Viper) (paramscmd.Params, error) {
 	if v == nil {
 		return paramscmd.Params{}, fmt.Errorf("viper instance unset")
 	}
 
-	heartbeatParams, err := paramscmd.LoadHeartbeatParams(v)
+	heartbeatParams, err := paramscmd.LoadHeartbeatParams(ctx, v)
 	if err != nil {
 		return paramscmd.Params{}, fmt.Errorf("failed to load heartbeat params: %s", err)
 	}
 
-	apiParams, err := paramscmd.LoadAPIParams(v)
+	apiParams, err := paramscmd.LoadAPIParams(ctx, v)
 	if err != nil {
 		return paramscmd.Params{}, fmt.Errorf("failed to load API parameters: %w", err)
 	}