FilesExpand file tree

main

/

category.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

104 lines (87 loc) · 2.68 KB

main

/

category.go

Copy path

Top

File metadata and controls

• Code
• Blame

104 lines (87 loc) · 2.68 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

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

/*

* Copyright (c) 2021-present Fabien Potencier <fabien@symfony.com>

*

* This file is part of Symfony CLI project

*

* This program is free software: you can redistribute it and/or modify

* it under the terms of the GNU Affero General Public License as

* published by the Free Software Foundation, either version 3 of the

* License, or (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

* GNU Affero General Public License for more details.

*

* You should have received a copy of the GNU Affero General Public License

* along with this program. If not, see <http://www.gnu.org/licenses/>.

*/

package console

import "sort"

type CommandCategories interface {

// AddCommand adds a command to a category, creating a new category if necessary.

AddCommand(category string, command *Command)

// Categories returns a copy of the category slice

Categories() []CommandCategory

}

type commandCategories []*commandCategory

func newCommandCategories() CommandCategories {

ret := commandCategories([]*commandCategory{})

return &ret

}

func (c *commandCategories) Less(i, j int) bool {

return (*c)[i].Name() < (*c)[j].Name()

}

func (c *commandCategories) Len() int {

return len(*c)

}

func (c *commandCategories) Swap(i, j int) {

(*c)[i], (*c)[j] = (*c)[j], (*c)[i]

}

func (c *commandCategories) AddCommand(category string, command *Command) {

for _, commandCategory := range []*commandCategory(*c) {

if commandCategory.name == category {

commandCategory.commands = append(commandCategory.commands, command)

return

}

}

newVal := commandCategories(append(*c,

&commandCategory{name: category, commands: []*Command{command}}))

(*c) = newVal

}

func (c *commandCategories) Categories() []CommandCategory {

ret := make([]CommandCategory, len(*c))

for i, cat := range *c {

ret[i] = cat

}

return ret

}

// CommandCategory is a category containing commands.

type CommandCategory interface {

// Name returns the category name string

Name() string

// VisibleCommands returns a slice of the Commands with Hidden=false

VisibleCommands() []*Command

}

type commandCategory struct {

name string

commands []*Command

}

func (c *commandCategory) Name() string {

return c.name

}

func (c *commandCategory) VisibleCommands() []*Command {

if c.commands == nil {

return []*Command{}

}

ret := []*Command{}

for _, command := range c.commands {

if command.Hidden == nil || !command.Hidden() {

ret = append(ret, command)

}

}

sort.Slice(ret, func(i, j int) bool {

return ret[i].Name < ret[j].Name

})

return ret

}