Felipe Martin
73149001eb
- Add logger.go with public Logger variable and InitLogger function - Replace all fmt.Printf/fmt.Fprintf calls with structured Logger.Info/Logger.Error - Initialize logger in main function for consistent access across packages - Keep fmt.Errorf for proper error creation (standard Go practice) - Add tint dependency for colorized terminal output with timestamps - Convert user output to structured logging with key-value pairs - Update info command to use structured logging for plugin details - Update updateassets command to use structured progress logging - Update version command to use structured logging - Update authentication logging in client.go with structured fields - Update enable/disable commands to use structured logging - Remove unused fmt imports after conversion All output now uses slog with tint for beautiful, structured, colorized logging while maintaining proper error handling with fmt.Errorf for error creation. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
100 lines
2.5 KiB
Go
100 lines
2.5 KiB
Go
package pluginctl
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
)
|
|
|
|
const commandTimeout = 120 * time.Second
|
|
|
|
func getClient(ctx context.Context) (*model.Client4, error) {
|
|
socketPath := os.Getenv("MM_LOCALSOCKETPATH")
|
|
if socketPath == "" {
|
|
socketPath = model.LocalModeSocketPath
|
|
}
|
|
|
|
client, connected := getUnixClient(socketPath)
|
|
if connected {
|
|
Logger.Info("Connecting using local mode", "socket_path", socketPath)
|
|
|
|
return client, nil
|
|
}
|
|
|
|
if os.Getenv("MM_LOCALSOCKETPATH") != "" {
|
|
Logger.Info("No socket found for local mode deployment. Attempting to authenticate with credentials.", "socket_path", socketPath)
|
|
}
|
|
|
|
siteURL := os.Getenv("MM_SERVICESETTINGS_SITEURL")
|
|
adminToken := os.Getenv("MM_ADMIN_TOKEN")
|
|
adminUsername := os.Getenv("MM_ADMIN_USERNAME")
|
|
adminPassword := os.Getenv("MM_ADMIN_PASSWORD")
|
|
|
|
if siteURL == "" {
|
|
return nil, errors.New("MM_SERVICESETTINGS_SITEURL is not set")
|
|
}
|
|
|
|
client = model.NewAPIv4Client(siteURL)
|
|
|
|
if adminToken != "" {
|
|
Logger.Info("Authenticating using token", "site_url", siteURL)
|
|
client.SetToken(adminToken)
|
|
|
|
return client, nil
|
|
}
|
|
|
|
if adminUsername != "" && adminPassword != "" {
|
|
client := model.NewAPIv4Client(siteURL)
|
|
Logger.Info("Authenticating with credentials", "username", adminUsername, "site_url", siteURL)
|
|
_, _, err := client.Login(ctx, adminUsername, adminPassword)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to login as %s: %w", adminUsername, err)
|
|
}
|
|
|
|
return client, nil
|
|
}
|
|
|
|
return nil, errors.New("one of MM_ADMIN_TOKEN or MM_ADMIN_USERNAME/MM_ADMIN_PASSWORD must be defined")
|
|
}
|
|
|
|
func getUnixClient(socketPath string) (*model.Client4, bool) {
|
|
_, err := net.Dial("unix", socketPath)
|
|
if err != nil {
|
|
return nil, false
|
|
}
|
|
|
|
return model.NewAPIv4SocketClient(socketPath), true
|
|
}
|
|
|
|
// runPluginCommand executes a plugin command using the plugin from the current folder.
|
|
func runPluginCommand(
|
|
_ []string,
|
|
pluginPath string,
|
|
action func(context.Context, *model.Client4, string) error,
|
|
) error {
|
|
// Load plugin manifest to get the plugin ID
|
|
manifest, err := LoadPluginManifestFromPath(pluginPath)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to load plugin manifest: %w", err)
|
|
}
|
|
|
|
pluginID := manifest.Id
|
|
if pluginID == "" {
|
|
return errors.New("plugin ID not found in manifest")
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), commandTimeout)
|
|
defer cancel()
|
|
|
|
client, err := getClient(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return action(ctx, client, pluginID)
|
|
}
|