mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2024-12-27 03:49:40 -08:00
refactor: extract environment
This commit is contained in:
Jan De Dobbeleer
Jan De Dobbeleer
parent
41268e832a
commit
53f313999a
180
src/runtime/environment.go
Normal file
180
src/runtime/environment.go
Normal file
|
|
@ -0,0 +1,180 @@
|
|||
package runtime
|
||||
|
||||
import (
|
||||
"io"
|
||||
"io/fs"
|
||||
"time"
|
||||
|
||||
"github.com/jandedobbeleer/oh-my-posh/src/cache"
|
||||
"github.com/jandedobbeleer/oh-my-posh/src/runtime/battery"
|
||||
"github.com/jandedobbeleer/oh-my-posh/src/runtime/http"
|
||||
|
||||
disk "github.com/shirou/gopsutil/v3/disk"
|
||||
)
|
||||
|
||||
const (
|
||||
UNKNOWN = "unknown"
|
||||
WINDOWS = "windows"
|
||||
DARWIN = "darwin"
|
||||
LINUX = "linux"
|
||||
CMD = "cmd"
|
||||
)
|
||||
|
||||
type Environment interface {
|
||||
Getenv(key string) string
|
||||
Pwd() string
|
||||
Home() string
|
||||
User() string
|
||||
Root() bool
|
||||
Host() (string, error)
|
||||
GOOS() string
|
||||
Shell() string
|
||||
Platform() string
|
||||
StatusCodes() (int, string)
|
||||
PathSeparator() string
|
||||
HasFiles(pattern string) bool
|
||||
HasFilesInDir(dir, pattern string) bool
|
||||
HasFolder(folder string) bool
|
||||
HasParentFilePath(path string, followSymlinks bool) (fileInfo *FileInfo, err error)
|
||||
HasFileInParentDirs(pattern string, depth uint) bool
|
||||
ResolveSymlink(path string) (string, error)
|
||||
DirMatchesOneOf(dir string, regexes []string) bool
|
||||
DirIsWritable(path string) bool
|
||||
CommandPath(command string) string
|
||||
HasCommand(command string) bool
|
||||
FileContent(file string) string
|
||||
LsDir(path string) []fs.DirEntry
|
||||
RunCommand(command string, args ...string) (string, error)
|
||||
RunShellCommand(shell, command string) string
|
||||
ExecutionTime() float64
|
||||
Flags() *Flags
|
||||
BatteryState() (*battery.Info, error)
|
||||
QueryWindowTitles(processName, windowTitleRegex string) (string, error)
|
||||
WindowsRegistryKeyValue(path string) (*WindowsRegistryValue, error)
|
||||
HTTPRequest(url string, body io.Reader, timeout int, requestModifiers ...http.RequestModifier) ([]byte, error)
|
||||
IsWsl() bool
|
||||
IsWsl2() bool
|
||||
IsCygwin() bool
|
||||
StackCount() int
|
||||
TerminalWidth() (int, error)
|
||||
CachePath() string
|
||||
Cache() cache.Cache
|
||||
Session() cache.Cache
|
||||
Close()
|
||||
Logs() string
|
||||
InWSLSharedDrive() bool
|
||||
ConvertToLinuxPath(path string) string
|
||||
ConvertToWindowsPath(path string) string
|
||||
Connection(connectionType ConnectionType) (*Connection, error)
|
||||
TemplateCache() *cache.Template
|
||||
LoadTemplateCache()
|
||||
SetPromptCount()
|
||||
CursorPosition() (row, col int)
|
||||
SystemInfo() (*SystemInfo, error)
|
||||
Debug(message string)
|
||||
DebugF(format string, a ...any)
|
||||
Error(err error)
|
||||
Trace(start time.Time, args ...string)
|
||||
}
|
||||
|
||||
type Flags struct {
|
||||
ErrorCode int
|
||||
PipeStatus string
|
||||
Config string
|
||||
Shell string
|
||||
ShellVersion string
|
||||
PWD string
|
||||
PSWD string
|
||||
ExecutionTime float64
|
||||
Eval bool
|
||||
StackCount int
|
||||
Migrate bool
|
||||
TerminalWidth int
|
||||
Strict bool
|
||||
Debug bool
|
||||
Manual bool
|
||||
Plain bool
|
||||
Primary bool
|
||||
HasTransient bool
|
||||
PromptCount int
|
||||
Cleared bool
|
||||
Cached bool
|
||||
NoExitCode bool
|
||||
Column int
|
||||
JobCount int
|
||||
}
|
||||
|
||||
type CommandError struct {
|
||||
Err string
|
||||
ExitCode int
|
||||
}
|
||||
|
||||
func (e *CommandError) Error() string {
|
||||
return e.Err
|
||||
}
|
||||
|
||||
type FileInfo struct {
|
||||
ParentFolder string
|
||||
Path string
|
||||
IsDir bool
|
||||
}
|
||||
|
||||
type WindowsRegistryValueType string
|
||||
|
||||
const (
|
||||
DWORD = "DWORD"
|
||||
QWORD = "QWORD"
|
||||
BINARY = "BINARY"
|
||||
STRING = "STRING"
|
||||
)
|
||||
|
||||
type WindowsRegistryValue struct {
|
||||
ValueType WindowsRegistryValueType
|
||||
DWord uint64
|
||||
QWord uint64
|
||||
String string
|
||||
}
|
||||
|
||||
type NotImplemented struct{}
|
||||
|
||||
func (n *NotImplemented) Error() string {
|
||||
return "not implemented"
|
||||
}
|
||||
|
||||
type ConnectionType string
|
||||
|
||||
const (
|
||||
ETHERNET ConnectionType = "ethernet"
|
||||
WIFI ConnectionType = "wifi"
|
||||
CELLULAR ConnectionType = "cellular"
|
||||
BLUETOOTH ConnectionType = "bluetooth"
|
||||
)
|
||||
|
||||
type Connection struct {
|
||||
Name string
|
||||
Type ConnectionType
|
||||
TransmitRate uint64
|
||||
ReceiveRate uint64
|
||||
SSID string // Wi-Fi only
|
||||
}
|
||||
|
||||
type Memory struct {
|
||||
PhysicalTotalMemory uint64
|
||||
PhysicalAvailableMemory uint64
|
||||
PhysicalFreeMemory uint64
|
||||
PhysicalPercentUsed float64
|
||||
SwapTotalMemory uint64
|
||||
SwapFreeMemory uint64
|
||||
SwapPercentUsed float64
|
||||
}
|
||||
|
||||
type SystemInfo struct {
|
||||
// mem
|
||||
Memory
|
||||
// load
|
||||
Load1 float64
|
||||
Load5 float64
|
||||
Load15 float64
|
||||
// disk
|
||||
Disks map[string]disk.IOCountersStat
|
||||
}
|
||||
|
|
@ -21,7 +21,6 @@ import (
|
|||
"github.com/jandedobbeleer/oh-my-posh/src/log"
|
||||
"github.com/jandedobbeleer/oh-my-posh/src/maps"
|
||||
"github.com/jandedobbeleer/oh-my-posh/src/regex"
|
||||
"github.com/jandedobbeleer/oh-my-posh/src/runtime/battery"
|
||||
"github.com/jandedobbeleer/oh-my-posh/src/runtime/cmd"
|
||||
"github.com/jandedobbeleer/oh-my-posh/src/runtime/config"
|
||||
"github.com/jandedobbeleer/oh-my-posh/src/runtime/http"
|
||||
|
|
@ -31,173 +30,6 @@ import (
|
|||
process "github.com/shirou/gopsutil/v3/process"
|
||||
)
|
||||
|
||||
const (
|
||||
UNKNOWN = "unknown"
|
||||
WINDOWS = "windows"
|
||||
DARWIN = "darwin"
|
||||
LINUX = "linux"
|
||||
CMD = "cmd"
|
||||
)
|
||||
|
||||
type Flags struct {
|
||||
ErrorCode int
|
||||
PipeStatus string
|
||||
Config string
|
||||
Shell string
|
||||
ShellVersion string
|
||||
PWD string
|
||||
PSWD string
|
||||
ExecutionTime float64
|
||||
Eval bool
|
||||
StackCount int
|
||||
Migrate bool
|
||||
TerminalWidth int
|
||||
Strict bool
|
||||
Debug bool
|
||||
Manual bool
|
||||
Plain bool
|
||||
Primary bool
|
||||
HasTransient bool
|
||||
PromptCount int
|
||||
Cleared bool
|
||||
Cached bool
|
||||
NoExitCode bool
|
||||
Column int
|
||||
JobCount int
|
||||
}
|
||||
|
||||
type CommandError struct {
|
||||
Err string
|
||||
ExitCode int
|
||||
}
|
||||
|
||||
func (e *CommandError) Error() string {
|
||||
return e.Err
|
||||
}
|
||||
|
||||
type FileInfo struct {
|
||||
ParentFolder string
|
||||
Path string
|
||||
IsDir bool
|
||||
}
|
||||
|
||||
type WindowsRegistryValueType string
|
||||
|
||||
const (
|
||||
DWORD = "DWORD"
|
||||
QWORD = "QWORD"
|
||||
BINARY = "BINARY"
|
||||
STRING = "STRING"
|
||||
)
|
||||
|
||||
type WindowsRegistryValue struct {
|
||||
ValueType WindowsRegistryValueType
|
||||
DWord uint64
|
||||
QWord uint64
|
||||
String string
|
||||
}
|
||||
|
||||
type NotImplemented struct{}
|
||||
|
||||
func (n *NotImplemented) Error() string {
|
||||
return "not implemented"
|
||||
}
|
||||
|
||||
type ConnectionType string
|
||||
|
||||
const (
|
||||
ETHERNET ConnectionType = "ethernet"
|
||||
WIFI ConnectionType = "wifi"
|
||||
CELLULAR ConnectionType = "cellular"
|
||||
BLUETOOTH ConnectionType = "bluetooth"
|
||||
)
|
||||
|
||||
type Connection struct {
|
||||
Name string
|
||||
Type ConnectionType
|
||||
TransmitRate uint64
|
||||
ReceiveRate uint64
|
||||
SSID string // Wi-Fi only
|
||||
}
|
||||
|
||||
type Memory struct {
|
||||
PhysicalTotalMemory uint64
|
||||
PhysicalAvailableMemory uint64
|
||||
PhysicalFreeMemory uint64
|
||||
PhysicalPercentUsed float64
|
||||
SwapTotalMemory uint64
|
||||
SwapFreeMemory uint64
|
||||
SwapPercentUsed float64
|
||||
}
|
||||
|
||||
type SystemInfo struct {
|
||||
// mem
|
||||
Memory
|
||||
// load
|
||||
Load1 float64
|
||||
Load5 float64
|
||||
Load15 float64
|
||||
// disk
|
||||
Disks map[string]disk.IOCountersStat
|
||||
}
|
||||
|
||||
type Environment interface {
|
||||
Getenv(key string) string
|
||||
Pwd() string
|
||||
Home() string
|
||||
User() string
|
||||
Root() bool
|
||||
Host() (string, error)
|
||||
GOOS() string
|
||||
Shell() string
|
||||
Platform() string
|
||||
StatusCodes() (int, string)
|
||||
PathSeparator() string
|
||||
HasFiles(pattern string) bool
|
||||
HasFilesInDir(dir, pattern string) bool
|
||||
HasFolder(folder string) bool
|
||||
HasParentFilePath(path string, followSymlinks bool) (fileInfo *FileInfo, err error)
|
||||
HasFileInParentDirs(pattern string, depth uint) bool
|
||||
ResolveSymlink(path string) (string, error)
|
||||
DirMatchesOneOf(dir string, regexes []string) bool
|
||||
DirIsWritable(path string) bool
|
||||
CommandPath(command string) string
|
||||
HasCommand(command string) bool
|
||||
FileContent(file string) string
|
||||
LsDir(path string) []fs.DirEntry
|
||||
RunCommand(command string, args ...string) (string, error)
|
||||
RunShellCommand(shell, command string) string
|
||||
ExecutionTime() float64
|
||||
Flags() *Flags
|
||||
BatteryState() (*battery.Info, error)
|
||||
QueryWindowTitles(processName, windowTitleRegex string) (string, error)
|
||||
WindowsRegistryKeyValue(path string) (*WindowsRegistryValue, error)
|
||||
HTTPRequest(url string, body io.Reader, timeout int, requestModifiers ...http.RequestModifier) ([]byte, error)
|
||||
IsWsl() bool
|
||||
IsWsl2() bool
|
||||
IsCygwin() bool
|
||||
StackCount() int
|
||||
TerminalWidth() (int, error)
|
||||
CachePath() string
|
||||
Cache() cache.Cache
|
||||
Session() cache.Cache
|
||||
Close()
|
||||
Logs() string
|
||||
InWSLSharedDrive() bool
|
||||
ConvertToLinuxPath(path string) string
|
||||
ConvertToWindowsPath(path string) string
|
||||
Connection(connectionType ConnectionType) (*Connection, error)
|
||||
TemplateCache() *cache.Template
|
||||
LoadTemplateCache()
|
||||
SetPromptCount()
|
||||
CursorPosition() (row, col int)
|
||||
SystemInfo() (*SystemInfo, error)
|
||||
Debug(message string)
|
||||
DebugF(format string, a ...any)
|
||||
Error(err error)
|
||||
Trace(start time.Time, args ...string)
|
||||
}
|
||||
|
||||
type Terminal struct {
|
||||
CmdFlags *Flags
|
||||
Var maps.Simple
|
||||
|
|
|
|||
Loading…
Reference in a new issue