102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package pluginctl
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"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 {
|
|
log.Printf("Connecting using local mode over %s", socketPath)
|
|
|
|
return client, nil
|
|
}
|
|
|
|
if os.Getenv("MM_LOCALSOCKETPATH") != "" {
|
|
log.Printf("No socket found at %s for local mode deployment. "+
|
|
"Attempting to authenticate with credentials.", 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 != "" {
|
|
log.Printf("Authenticating using token against %s.", siteURL)
|
|
client.SetToken(adminToken)
|
|
|
|
return client, nil
|
|
}
|
|
|
|
if adminUsername != "" && adminPassword != "" {
|
|
client := model.NewAPIv4Client(siteURL)
|
|
log.Printf("Authenticating as %s against %s.", adminUsername, 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)
|
|
}
|