package exec

import (
  "context"
  "fmt"
  "github.com/RedTeamPentesting/adauth"
  "strings"
)

const (
  ConnectionMethodDCE = "dcerpc"
)

type ConnectionConfig struct {
  ConnectionMethod       string
  ConnectionMethodConfig interface{}
}

type CleanupConfig struct {
  CleanupMethod       string
  CleanupMethodConfig interface{}
}

type ExecutionConfig struct {
  ExecutableName string // ExecutableName represents the name of the executable; i.e. "notepad.exe", "calc"
  ExecutablePath string // ExecutablePath represents the full path to the executable; i.e. `C:\Windows\explorer.exe`
  ExecutableArgs string // ExecutableArgs represents the arguments to be passed to the executable during execution; i.e. "/C whoami"

  ExecutionMethod       string // ExecutionMethod represents the specific execution strategy used by the module.
  ExecutionMethodConfig interface{}
  ReturnOutput          bool
}

type ShellConfig struct {
  ShellName string // ShellName specifies the name of the shell executable; i.e. "cmd.exe", "powershell"
  ShellPath string // ShellPath is the full Windows path to the shell executable; i.e. `C:\Windows\System32\cmd.exe`
}

type Module interface {
  Connect(context.Context, *adauth.Credential, *adauth.Target, *ConnectionConfig) error
  Exec(context.Context, *ExecutionConfig) error
  Cleanup(context.Context, *CleanupConfig) error
}

func (cfg *ExecutionConfig) GetRawCommand() string {
  executable := cfg.ExecutablePath
  if strings.Contains(executable, " ") {
    executable = fmt.Sprintf("%q", executable)
  }
  if cfg.ExecutableArgs != "" {
    return executable + " " + cfg.ExecutableArgs
  }
  return executable
}