mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2024-11-13 14:34:06 -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/log"
|
||||||
"github.com/jandedobbeleer/oh-my-posh/src/maps"
|
"github.com/jandedobbeleer/oh-my-posh/src/maps"
|
||||||
"github.com/jandedobbeleer/oh-my-posh/src/regex"
|
"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/cmd"
|
||||||
"github.com/jandedobbeleer/oh-my-posh/src/runtime/config"
|
"github.com/jandedobbeleer/oh-my-posh/src/runtime/config"
|
||||||
"github.com/jandedobbeleer/oh-my-posh/src/runtime/http"
|
"github.com/jandedobbeleer/oh-my-posh/src/runtime/http"
|
||||||
|
|
@ -31,173 +30,6 @@ import (
|
||||||
process "github.com/shirou/gopsutil/v3/process"
|
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 {
|
type Terminal struct {
|
||||||
CmdFlags *Flags
|
CmdFlags *Flags
|
||||||
Var maps.Simple
|
Var maps.Simple
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue