FilesExpand file tree

main

/

flag_value_source.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

76 lines (62 loc) · 1.65 KB

main

/

flag_value_source.go

Copy path

Top

File metadata and controls

• Code
• Blame

76 lines (62 loc) · 1.65 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

package cli

import (

"fmt"

"os"

"strings"

"syscall"

)

// FlagValueSource encapsulates a source which can be used to

// fetch a value

type FlagValueSource interface {

// Returns the value from the source and if it was found

// otherwise returns an empty string & found is set to false

Get() (string, bool)

// The identifier for this source

Identifier() string

}

// ValueSources encapsulates all value sources

type ValueSources []FlagValueSource

func (v ValueSources) Get() (string, string, bool) {

for _, src := range v {

if value, found := src.Get(); found {

return value, src.Identifier(), true

}

}

return "", "", false

}

// EnvSource encapsulates an env

type EnvSource string

func (e EnvSource) Get() (string, bool) {

envVar := strings.TrimSpace(string(e))

return syscall.Getenv(envVar)

}

func (e EnvSource) Identifier() string {

return fmt.Sprintf("environment variable %q", string(e))

}

// EnvVars is a helper function to encapsulate a bunch

// of envs together as ValueSources

func EnvVars(envs ...string) ValueSources {

vs := []FlagValueSource{}

for _, env := range envs {

vs = append(vs, EnvSource(env))

}

return vs

}

// FileSource encapsulates an file source

type FileSource string

func (f FileSource) Get() (string, bool) {

data, err := os.ReadFile(string(f))

return string(data), err == nil

}

func (f FileSource) Identifier() string {

return fmt.Sprintf("file %q", string(f))

}

// FilePaths is a helper function to encapsulate a bunch

// of file paths together as ValueSources

func FilePaths(paths ...string) ValueSources {

vs := []FlagValueSource{}

for _, path := range paths {

vs = append(vs, FileSource(path))

}

return vs

}