-
-
Notifications
You must be signed in to change notification settings - Fork 2.4k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
a6461d2
commit 6bfa44b
Showing
8 changed files
with
203 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,65 @@ | ||
package segments | ||
|
||
import ( | ||
"path/filepath" | ||
"strings" | ||
|
||
"github.com/jandedobbeleer/oh-my-posh/src/properties" | ||
"github.com/jandedobbeleer/oh-my-posh/src/runtime" | ||
) | ||
|
||
const ( | ||
NONE = "none" | ||
) | ||
|
||
type NixShell struct { | ||
props properties.Properties | ||
env runtime.Environment | ||
|
||
Type string | ||
} | ||
|
||
func (n *NixShell) Template() string { | ||
return "via {{ .Type }}-shell" | ||
} | ||
|
||
func (n *NixShell) DetectType() string { | ||
shellType := n.env.Getenv("IN_NIX_SHELL") | ||
|
||
switch shellType { | ||
case "pure", "impure": | ||
return shellType | ||
default: | ||
if n.InNewNixShell() { | ||
return UNKNOWN | ||
} | ||
|
||
return NONE | ||
} | ||
} | ||
|
||
// Hack to detect if we're in a `nix shell` (in contrast to a `nix-shell`). | ||
// A better way to do this will be enabled by https://github.com/NixOS/nix/issues/6677 | ||
// so we check if the PATH contains a nix store. | ||
func (n *NixShell) InNewNixShell() bool { | ||
paths := filepath.SplitList(n.env.Getenv("PATH")) | ||
|
||
for _, p := range paths { | ||
if strings.Contains(p, "/nix/store") { | ||
return true | ||
} | ||
} | ||
|
||
return false | ||
} | ||
|
||
func (n *NixShell) Init(props properties.Properties, env runtime.Environment) { | ||
n.props = props | ||
n.env = env | ||
} | ||
|
||
func (n *NixShell) Enabled() bool { | ||
n.Type = n.DetectType() | ||
|
||
return n.Type != NONE | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,73 @@ | ||
package segments | ||
|
||
import ( | ||
"fmt" | ||
"testing" | ||
|
||
"github.com/alecthomas/assert" | ||
"github.com/jandedobbeleer/oh-my-posh/src/properties" | ||
"github.com/jandedobbeleer/oh-my-posh/src/runtime/mock" | ||
) | ||
|
||
const ( | ||
nixPath = "/nix/store/zznw8fnzss1vaqfg5hmv3y79s3hkqczi-devshell-dir/bin" | ||
defaultPath = "/users/xyz/testing" | ||
fullNixPath = defaultPath + ":" + nixPath | ||
) | ||
|
||
func TestNixShellSegment(t *testing.T) { | ||
cases := []struct { | ||
name string | ||
expectedString string | ||
shellType string | ||
enabled bool | ||
}{ | ||
{ | ||
name: "Pure Nix Shell", | ||
expectedString: "via pure-shell", | ||
shellType: "pure", | ||
enabled: true, | ||
}, | ||
{ | ||
name: "Impure Nix Shell", | ||
expectedString: "via impure-shell", | ||
shellType: "impure", | ||
enabled: true, | ||
}, | ||
{ | ||
name: "Unknown Nix Shell", | ||
expectedString: "via unknown-shell", | ||
shellType: "unknown", | ||
enabled: true, | ||
}, | ||
{ | ||
name: "No Nix Shell", | ||
expectedString: "", | ||
shellType: "", | ||
enabled: false, | ||
}, | ||
} | ||
|
||
for _, tc := range cases { | ||
t.Run(tc.name, func(t *testing.T) { | ||
env := new(mock.Environment) | ||
env.On("Getenv", "IN_NIX_SHELL").Return(tc.shellType) | ||
|
||
path := defaultPath | ||
if tc.shellType != "" { | ||
path = fullNixPath | ||
} | ||
|
||
env.On("Getenv", "PATH").Return(path) | ||
|
||
n := NixShell{} | ||
n.Init(properties.Map{}, env) | ||
|
||
assert.Equal(t, tc.enabled, n.Enabled(), fmt.Sprintf("Failed in case: %s", tc.name)) | ||
|
||
if tc.enabled { | ||
assert.Equal(t, tc.expectedString, renderTemplate(env, n.Template(), n), tc.name) | ||
} | ||
}) | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,42 @@ | ||
--- | ||
id: nix-shell | ||
title: Nix Shell | ||
sidebar_label: Nix Shell | ||
--- | ||
|
||
## What | ||
|
||
Displays the [nix shell] status if inside a nix-shell environment. | ||
|
||
## Sample Configuration | ||
|
||
import Config from "@site/src/components/Config.js"; | ||
|
||
<Config | ||
data={{ | ||
type: "nix-shell", | ||
style: "powerline", | ||
foreground: "blue", | ||
background: "transparent", | ||
template: "(nix-{{ .Type }})", | ||
}} | ||
/> | ||
|
||
## Template ([info][templates]) | ||
|
||
:::note default template | ||
|
||
```template | ||
via {{ .Type }}-shell" | ||
``` | ||
|
||
::: | ||
|
||
### Properties | ||
|
||
| Name | Type | Description | | ||
| ------- | -------- | ----------------------------------------------------------- | | ||
| `.Type` | `string` | the type of nix shell, can be `pure`, `impure` or `unknown` | | ||
|
||
[nix shell]: https://nixos.org/guides/nix-pills/developing-with-nix-shell.html | ||
[templates]: /docs/configuration/templates |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters