wer
This commit is contained in:
@@ -0,0 +1 @@
|
||||
v1.0.0
|
||||
@@ -0,0 +1 @@
|
||||
{"Version":"v1.0.0","Time":"2020-03-29T21:11:09Z"}
|
||||
@@ -0,0 +1,8 @@
|
||||
module github.com/daviddengcn/go-colortext
|
||||
|
||||
go 1.14
|
||||
|
||||
require (
|
||||
github.com/golangplus/bytes v1.0.0 // indirect
|
||||
github.com/golangplus/testing v1.0.0
|
||||
)
|
||||
Binary file not shown.
@@ -0,0 +1 @@
|
||||
h1:ANqDyC0ys6qCSvuEK7l3g5RaehL/Xck9EX8ATG8oKsE=
|
||||
@@ -0,0 +1,2 @@
|
||||
v1.0.6
|
||||
v4.0.2+incompatible
|
||||
@@ -0,0 +1 @@
|
||||
module github.com/dixonwille/wmenu
|
||||
+1
@@ -0,0 +1 @@
|
||||
{"Version":"v4.0.2+incompatible","Time":"2018-01-05T18:19:49Z"}
|
||||
@@ -0,0 +1 @@
|
||||
module github.com/dixonwille/wmenu
|
||||
BIN
Binary file not shown.
+1
@@ -0,0 +1 @@
|
||||
h1:lxrPJsx9LpdUFD5T+dOfl6gPKLbBmiAtEdACLT1I2/w=
|
||||
@@ -0,0 +1,2 @@
|
||||
v0.0.0-20160111154220-45c989fe5450
|
||||
v1.0.0
|
||||
Vendored
+1
@@ -0,0 +1 @@
|
||||
module github.com/golangplus/bytes
|
||||
@@ -0,0 +1,5 @@
|
||||
module github.com/golangplus/bytes
|
||||
|
||||
go 1.14
|
||||
|
||||
require github.com/golangplus/testing v1.0.0
|
||||
@@ -0,0 +1 @@
|
||||
v1.0.0
|
||||
@@ -0,0 +1,3 @@
|
||||
module github.com/golangplus/fmt
|
||||
|
||||
go 1.14
|
||||
@@ -0,0 +1 @@
|
||||
v1.0.0
|
||||
@@ -0,0 +1,8 @@
|
||||
module github.com/golangplus/testing
|
||||
|
||||
go 1.14
|
||||
|
||||
require (
|
||||
github.com/golangplus/bytes v0.0.0-20160111154220-45c989fe5450
|
||||
github.com/golangplus/fmt v1.0.0
|
||||
)
|
||||
@@ -0,0 +1 @@
|
||||
v0.0.20
|
||||
@@ -0,0 +1 @@
|
||||
{"Version":"v0.0.20","Time":"2023-10-17T07:28:21Z","Origin":{"VCS":"git","URL":"https://github.com/mattn/go-isatty","Hash":"a7c02353c47bc4ec6b30dc9628154ae4fe760c11","Ref":"refs/tags/v0.0.20"}}
|
||||
@@ -0,0 +1,5 @@
|
||||
module github.com/mattn/go-isatty
|
||||
|
||||
go 1.15
|
||||
|
||||
require golang.org/x/sys v0.6.0
|
||||
Binary file not shown.
@@ -0,0 +1 @@
|
||||
h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
@@ -0,0 +1,2 @@
|
||||
v0.6.0
|
||||
v0.35.0
|
||||
@@ -0,0 +1 @@
|
||||
{"Version":"v0.35.0","Time":"2025-08-06T21:03:43Z","Origin":{"VCS":"git","URL":"https://go.googlesource.com/sys","Hash":"5b936e1f126baa13682eff91c2e4d5d9e3a0b71d","Ref":"refs/tags/v0.35.0"}}
|
||||
@@ -0,0 +1,3 @@
|
||||
module golang.org/x/sys
|
||||
|
||||
go 1.23.0
|
||||
@@ -0,0 +1,3 @@
|
||||
module golang.org/x/sys
|
||||
|
||||
go 1.17
|
||||
Binary file not shown.
@@ -0,0 +1 @@
|
||||
h1:MVltZSvRTcU2ljQOhs94SXPftV6DCNnZViHeQps87pQ=
|
||||
@@ -0,0 +1 @@
|
||||
v2.0.0
|
||||
@@ -0,0 +1 @@
|
||||
{"Version":"v2.0.0","Time":"2017-02-22T03:43:20Z"}
|
||||
@@ -0,0 +1 @@
|
||||
module gopkg.in/dixonwille/wlog.v2
|
||||
Binary file not shown.
@@ -0,0 +1 @@
|
||||
h1:TbGWtD8ahWVSihKKr+z2Dw7Cv/7IrfN6dwrcrre17pU=
|
||||
Vendored
+9
@@ -0,0 +1,9 @@
|
||||
1005358
|
||||
github.com/daviddengcn/go-colortext v1.0.0 h1:ANqDyC0ys6qCSvuEK7l3g5RaehL/Xck9EX8ATG8oKsE=
|
||||
github.com/daviddengcn/go-colortext v1.0.0/go.mod h1:zDqEI5NVUop5QPpVJUxE9UO10hRnmkD5G4Pmri9+m4c=
|
||||
|
||||
go.sum database tree
|
||||
43191231
|
||||
qHm0EQ3KgyaK0OqkvYk3oQYgEdXK/VSsNYlbZRCb3Pw=
|
||||
|
||||
— sum.golang.org Az3grtfyLGjokX0/tVEfOzpGVUN+QbORZDiM3OQLiEhL2tPBek1hNZ5pL2xWN480RcOycWNpRJ6Y2VZGFE7rW9sv/Qs=
|
||||
+9
@@ -0,0 +1,9 @@
|
||||
200592
|
||||
github.com/dixonwille/wmenu v1.0.6 h1:LGglnOOozO+lLyh38JcUAYEMal4n7qU4mfUmkB5Tewo=
|
||||
github.com/dixonwille/wmenu v1.0.6/go.mod h1:DnajdZEKFQksxBctWekpWaQXQrDUHRBco6b8MyZnR1s=
|
||||
|
||||
go.sum database tree
|
||||
43191517
|
||||
7EyyR27fHHZvBXkJPRXdQwdS4rdhTkZYoT5hKpGlnnc=
|
||||
|
||||
— sum.golang.org Az3grtQelJ8vIsOksGofUjCjg30znYS84OnPtg45zIGkgsYtRvVvixn/AMUG1ZJbvI/au4bh0r9HnjvoptRWwOWr9Q8=
|
||||
Vendored
+9
@@ -0,0 +1,9 @@
|
||||
166897
|
||||
github.com/dixonwille/wmenu v4.0.2+incompatible h1:lxrPJsx9LpdUFD5T+dOfl6gPKLbBmiAtEdACLT1I2/w=
|
||||
github.com/dixonwille/wmenu v4.0.2+incompatible/go.mod h1:DnajdZEKFQksxBctWekpWaQXQrDUHRBco6b8MyZnR1s=
|
||||
|
||||
go.sum database tree
|
||||
43191376
|
||||
DIOQQw4mW9S2sQW62NxA334oBLi5bVg6Nj117r5CumM=
|
||||
|
||||
— sum.golang.org Az3grm/s8odMDAP2t4Kt0zSlfanJpM/KZl4AIHNW7gKuFvkJtyMyuvnYD556OUPLChd1N/oPhBgfUpczsHCXHDgTigI=
|
||||
+9
@@ -0,0 +1,9 @@
|
||||
1688
|
||||
github.com/golangplus/bytes v0.0.0-20160111154220-45c989fe5450 h1:7xqw01UYS+KCI25bMrPxwNYkSns2Db1ziQPpVq99FpE=
|
||||
github.com/golangplus/bytes v0.0.0-20160111154220-45c989fe5450/go.mod h1:Bk6SMAONeMXrxql8uvOKuAZSu8aM5RUGv+1C6IJaEho=
|
||||
|
||||
go.sum database tree
|
||||
43191621
|
||||
CPpOpnVoazY4kNwzZTjKVhSXeOozossXQNa2UaER874=
|
||||
|
||||
— sum.golang.org Az3grs07kGysb2oy7poMGiocHtxStJiWkIY5aXWTE/8z587hOHyTt7/2zEwxImAeHn6zBMh7pk4WYTlw2h0A3GS/PA4=
|
||||
+9
@@ -0,0 +1,9 @@
|
||||
1005311
|
||||
github.com/golangplus/bytes v1.0.0 h1:YQKBijBVMsBxIiXT4IEhlKR2zHohjEqPole4umyDX+c=
|
||||
github.com/golangplus/bytes v1.0.0/go.mod h1:AdRaCFwmc/00ZzELMWb01soso6W1R/++O1XL80yAn+A=
|
||||
|
||||
go.sum database tree
|
||||
43191434
|
||||
uUgGgJJ11/evLXIVRDwX7BtLpBGIWimWLbbQ9Q68BDk=
|
||||
|
||||
— sum.golang.org Az3grjxycAWElvF12OWPfckTbbldhvtCrWLR2kvTEYuM+1NByk0Ljje8r7BXdaxTmC5gosnwsAqlIWaAD0upft68mwg=
|
||||
+9
@@ -0,0 +1,9 @@
|
||||
1005266
|
||||
github.com/golangplus/fmt v1.0.0 h1:FnUKtw86lXIPfBMc3FimNF3+ABcV+aH5F17OOitTN+E=
|
||||
github.com/golangplus/fmt v1.0.0/go.mod h1:zpM0OfbMCjPtd2qkTD/jX2MgiFCqklhSUFyDW44gVQE=
|
||||
|
||||
go.sum database tree
|
||||
43191559
|
||||
edMTV+ivJ5B37CQXVJCLAAjeMST4od34ze2/+pBcoVo=
|
||||
|
||||
— sum.golang.org Az3gru0G0VTkySWK+XC96DWVPPkh3rc+0hFRm3pXrv8oUaVqcv9NrDkfxGyxQPYeLYCccAhsQ8sq4uzk5fc1YK+E4g0=
|
||||
+9
@@ -0,0 +1,9 @@
|
||||
1005305
|
||||
github.com/golangplus/testing v1.0.0 h1:+ZeeiKZENNOMkTTELoSySazi+XaEhVO0mb+eanrSEUQ=
|
||||
github.com/golangplus/testing v1.0.0/go.mod h1:ZDreixUV3YzhoVraIDyOzHrr76p6NUh6k/pPg/Q3gYA=
|
||||
|
||||
go.sum database tree
|
||||
43191533
|
||||
maZW25aF1f5oVvbLBzQe0YSd+DCY1S1KxqIf7RdEO3c=
|
||||
|
||||
— sum.golang.org Az3grsaj+H35f61nbG8uSkJKc/SP53ijXTRhpNfFl8pZpm5F3q1XHgm6fLHEEBfdnAsdXbdRJ336ezAQuvKae+pYMgA=
|
||||
+9
@@ -0,0 +1,9 @@
|
||||
20103885
|
||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
|
||||
go.sum database tree
|
||||
43190277
|
||||
3sXu5/jRWbnpx0n8SDcDfFJz4DEw13hovHu+DNv47bo=
|
||||
|
||||
— sum.golang.org Az3grrx6yeO58h59oW1E7xQm+Fovz9EiK94MrFF3TkHviAe6RxYSm36W5fRLvXFgjuDPtFMkRROmualxlJpdxS4E9QI=
|
||||
@@ -0,0 +1,9 @@
|
||||
41721156
|
||||
golang.org/x/sys v0.35.0 h1:vz1N37gP5bs89s7He8XuIYXpyY0+QlsKmzipCbUtyxI=
|
||||
golang.org/x/sys v0.35.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
|
||||
|
||||
go.sum database tree
|
||||
43188730
|
||||
J2dNtgSfeQdU383Ai26yrXFKwdgxrnH2RL54WrtVYes=
|
||||
|
||||
— sum.golang.org Az3grr7Cs/piAs7SsfsJwBOk3neQJID0R85JZulrLK7ZhMlUfib7p1XECPtvCe8G84QYqbSHaZ/8V6sMW2FccU966QI=
|
||||
@@ -0,0 +1,9 @@
|
||||
16177722
|
||||
golang.org/x/sys v0.6.0 h1:MVltZSvRTcU2ljQOhs94SXPftV6DCNnZViHeQps87pQ=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
|
||||
go.sum database tree
|
||||
43189576
|
||||
bKD0/PXJfMy7gcRUqBwYlQeA7Z1dKAXz832FirMsEaQ=
|
||||
|
||||
— sum.golang.org Az3grvfUZuwRiHMSJf+TUZHhcCWe0sGshW5QRBx5ZMZC+4yX9BzrS4ADJHHJxM2LwKYETPsF3CxXRmkipno1vIsrQw4=
|
||||
+9
@@ -0,0 +1,9 @@
|
||||
166898
|
||||
gopkg.in/dixonwille/wlog.v2 v2.0.0 h1:TbGWtD8ahWVSihKKr+z2Dw7Cv/7IrfN6dwrcrre17pU=
|
||||
gopkg.in/dixonwille/wlog.v2 v2.0.0/go.mod h1:JYQHRnhGPLno/iATOiGkEXoRanJXqdz9Qo6/QwfARUc=
|
||||
|
||||
go.sum database tree
|
||||
43191692
|
||||
qWxr/VHR84emiDefVcES1LiNYT1rge7571whsp167Bs=
|
||||
|
||||
— sum.golang.org Az3grgO12q3fua7W82xGucR3qBMahgw66LQXZJCESkk+zk7OXVESW3LIqjVMTKqA9jLEBm7DlBUpAU61MnGbrTUp7wg=
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -0,0 +1,5 @@
|
||||
Éþ¼„˜–”;_Txl%šëŽÍHÐsR¨wj
|
||||
^³’ù¢ŸÎ7§ªä0˜Õ·] �O²�1á³ÁÉYÍù
|
||||
…l¨†»Ölœ~Ù8×0Ðnœ&âiáåçÈbIaïÐbÇ[yÖ½V=¼KÞ kà Ë(]Ñé½®\‚`Éè"!tÆ"'�ºõÑ~¾APˆ©?Û]J>= ™îíª¡èŒ±ò•+¥ùË·ÏgJ1«%û3¢TŸZ7&ñ4‘W™~–ðg£÷‘¬y‚-�*?¦ˆpc4®ÐçÌ “ìie"Ñ f¼ËÞÒ¡{!5Ÿ·"«¡Ï-=m0žFJ) dª¾5°â·␍oÛ�¶\ÆÄ}PÓj$Ãȃ<Â1œ¨s[2 øà"ÌBÆ^ÂÑPf␍�&´|Tï±³§ l*tŒŽ'a§SA™C=Æ1‡üM +þrXßéyºZÅ1�•™<¹íV<
|
||||
¸¶®BP”½t»)µ
|
||||
}©`A¶2
|
||||
@@ -0,0 +1,5 @@
|
||||
Éþ¼„˜–”;_Txl%šëŽÍHÐsR¨wj
|
||||
^³’ù¢ŸÎ7§ªä0˜Õ·] �O²�1á³ÁÉYÍù
|
||||
…l¨†»Ölœ~Ù8×0Ðnœ&âiáåçÈbIaïÐbÇ[yÖ½V=¼KÞ kà Ë(]Ñé½®\‚`Éè"!tÆ"'�ºõÑ~¾APˆ©?Û]J>= ™îíª¡èŒ±ò•+¥ùË·ÏgJ1«%û3¢TŸZ7&ñ4‘W™~–ðg£÷‘¬y‚-�*?¦ˆpc4®ÐçÌ “ìie"Ñ f¼ËÞÒ¡{!5Ÿ·"«¡Ï-=m0žFJ) dª¾5°â·␍oÛ�¶\ÆÄ}PÓj$Ãȃ<Â1œ¨s[2 øà"ÌBÆ^ÂÑPf␍�&´|Tï±³§ l*tŒŽ'a§SA™C=Æ1‡üM +þrXßéyºZÅ1�•™<¹íV<
|
||||
¸¶®BP”½t»)µ
|
||||
}©`A¶2c’ŠÕ¿Ô'ë¹$\xÎæ¦5ô•£í3%üe -ì‹…
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -0,0 +1,3 @@
|
||||
o
|
||||
”@m°%q±ŁT`…ow!^ńľZ—{§�ËŤéĐvÍëąś5Ń1 "sJ®YFŘń Ç
|
||||
iÂéč¶9Z•Ć9Ć
|
||||
@@ -0,0 +1,22 @@
|
||||
# Compiled Object files, Static and Dynamic libs (Shared Objects)
|
||||
*.o
|
||||
*.a
|
||||
*.so
|
||||
|
||||
# Folders
|
||||
_obj
|
||||
_test
|
||||
|
||||
# Architecture specific extensions/prefixes
|
||||
*.[568vq]
|
||||
[568vq].out
|
||||
|
||||
*.cgo1.go
|
||||
*.cgo2.c
|
||||
_cgo_defun.c
|
||||
_cgo_gotypes.go
|
||||
_cgo_export.*
|
||||
|
||||
_testmain.go
|
||||
|
||||
*.exe
|
||||
@@ -0,0 +1,54 @@
|
||||
BSD License
|
||||
===========
|
||||
|
||||
Copyright (c) 2016, David Deng
|
||||
All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
|
||||
* Redistributions of source code must retain the above copyright notice, this
|
||||
list of conditions and the following disclaimer.
|
||||
|
||||
* Redistributions in binary form must reproduce the above copyright notice,
|
||||
this list of conditions and the following disclaimer in the documentation
|
||||
and/or other materials provided with the distribution.
|
||||
|
||||
* Neither the name of go-colortext nor the names of its
|
||||
contributors may be used to endorse or promote products derived from
|
||||
this software without specific prior written permission.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
||||
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
||||
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
|
||||
MIT License
|
||||
===========
|
||||
|
||||
Copyright (c) 2016 David Deng
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
@@ -0,0 +1,21 @@
|
||||
go-colortext package [](http://go-search.org/view?id=github.com%2Fdaviddengcn%2Fgo-colortext)
|
||||
====================
|
||||
|
||||
This is a package to change the color of the text and background in the console, working both under Windows and other systems.
|
||||
|
||||
Under Windows, the console APIs are used. Otherwise, ANSI texts are output.
|
||||
|
||||
Docs: http://godoc.org/github.com/daviddengcn/go-colortext ([packages that import ct](http://go-search.org/view?id=github.com%2fdaviddengcn%2fgo-colortext))
|
||||
|
||||
Usage:
|
||||
```go
|
||||
ct.Foreground(Green, false)
|
||||
fmt.Println("Green text starts here...")
|
||||
ct.ChangeColor(Red, true, White, false)
|
||||
fmt.Println(...)
|
||||
ct.ResetColor()
|
||||
```
|
||||
|
||||
LICENSE
|
||||
=======
|
||||
BSD/MIT license
|
||||
@@ -0,0 +1,53 @@
|
||||
/*
|
||||
ct package provides functions to change the color of console text.
|
||||
|
||||
Under windows platform, the Console API is used. Under other systems, ANSI text mode is used.
|
||||
*/
|
||||
package ct
|
||||
|
||||
import (
|
||||
"io"
|
||||
"os"
|
||||
)
|
||||
|
||||
// Color is the type of color to be set.
|
||||
type Color int
|
||||
|
||||
const (
|
||||
// No change of color
|
||||
None = Color(iota)
|
||||
Black
|
||||
Red
|
||||
Green
|
||||
Yellow
|
||||
Blue
|
||||
Magenta
|
||||
Cyan
|
||||
White
|
||||
)
|
||||
|
||||
// Writer is the io.Writer where ANSI escape codes will be written to
|
||||
var Writer io.Writer = os.Stdout
|
||||
|
||||
// ResetColor resets the foreground and background to original colors
|
||||
func ResetColor() {
|
||||
resetColor()
|
||||
}
|
||||
|
||||
// ChangeColor sets the foreground and background colors. If the value of the color is None,
|
||||
// the corresponding color keeps unchanged.
|
||||
// If fgBright or bgBright is set true, corresponding color use bright color. bgBright may be
|
||||
// ignored in some OS environment.
|
||||
func ChangeColor(fg Color, fgBright bool, bg Color, bgBright bool) {
|
||||
changeColor(fg, fgBright, bg, bgBright)
|
||||
}
|
||||
|
||||
// Foreground changes the foreground color.
|
||||
func Foreground(cl Color, bright bool) {
|
||||
ChangeColor(cl, bright, None, false)
|
||||
}
|
||||
|
||||
// Background changes the background color.
|
||||
func Background(cl Color, bright bool) {
|
||||
ChangeColor(None, false, cl, bright)
|
||||
}
|
||||
@@ -0,0 +1,51 @@
|
||||
// +build !windows
|
||||
|
||||
package ct
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
func isDumbTerm() bool {
|
||||
return os.Getenv("TERM") == "dumb"
|
||||
}
|
||||
|
||||
func resetColor() {
|
||||
if isDumbTerm() {
|
||||
return
|
||||
}
|
||||
fmt.Fprint(Writer, "\x1b[0m")
|
||||
}
|
||||
|
||||
func ansiText(fg Color, fgBright bool, bg Color, bgBright bool) string {
|
||||
if fg == None && bg == None {
|
||||
return ""
|
||||
}
|
||||
s := []byte("\x1b[0")
|
||||
if fg != None {
|
||||
s = strconv.AppendUint(append(s, ";"...), 30+(uint64)(fg-Black), 10)
|
||||
if fgBright {
|
||||
s = append(s, ";1"...)
|
||||
}
|
||||
}
|
||||
if bg != None {
|
||||
s = strconv.AppendUint(append(s, ";"...), 40+(uint64)(bg-Black), 10)
|
||||
if bgBright {
|
||||
s = append(s, ";1"...)
|
||||
}
|
||||
}
|
||||
s = append(s, "m"...)
|
||||
return string(s)
|
||||
}
|
||||
|
||||
func changeColor(fg Color, fgBright bool, bg Color, bgBright bool) {
|
||||
if isDumbTerm() {
|
||||
return
|
||||
}
|
||||
if fg == None && bg == None {
|
||||
return
|
||||
}
|
||||
fmt.Fprint(Writer, ansiText(fg, fgBright, bg, bgBright))
|
||||
}
|
||||
@@ -0,0 +1,18 @@
|
||||
// +build !windows
|
||||
|
||||
package ct
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/golangplus/testing/assert"
|
||||
)
|
||||
|
||||
func TestAnsiText(t *testing.T) {
|
||||
assert.Equal(t, "ansiText", ansiText(None, false, None, false), "")
|
||||
assert.Equal(t, "ansiText", ansiText(Red, false, None, false), "\x1b[0;31m")
|
||||
assert.Equal(t, "ansiText", ansiText(Red, true, None, false), "\x1b[0;31;1m")
|
||||
assert.Equal(t, "ansiText", ansiText(None, false, Green, false), "\x1b[0;42m")
|
||||
assert.Equal(t, "ansiText", ansiText(Red, false, Green, false), "\x1b[0;31;42m")
|
||||
assert.Equal(t, "ansiText", ansiText(Red, true, Green, false), "\x1b[0;31;1;42m")
|
||||
}
|
||||
@@ -0,0 +1,82 @@
|
||||
package ct
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestChangeColor(t *testing.T) {
|
||||
defer ResetColor()
|
||||
fmt.Println("Normal text...")
|
||||
text := "This is an demo of using ChangeColor to output colorful texts"
|
||||
i := 1
|
||||
for _, c := range text {
|
||||
ChangeColor(Color(i/2%8)+Black, i%2 == 1, Color((i+2)/2%8)+Black, false)
|
||||
fmt.Print(string(c))
|
||||
i++
|
||||
}
|
||||
fmt.Println()
|
||||
ChangeColor(Red, true, White, false)
|
||||
fmt.Println("Before reset.")
|
||||
ChangeColor(Red, false, White, true)
|
||||
fmt.Println("Before reset.")
|
||||
ResetColor()
|
||||
fmt.Println("After reset.")
|
||||
fmt.Println("After reset.")
|
||||
}
|
||||
|
||||
func TestForeground(t *testing.T) {
|
||||
ResetColor()
|
||||
defer ResetColor()
|
||||
|
||||
fmt.Println("Please check the words under the following text shows with the corresponding front color:")
|
||||
|
||||
colorToText := [...]string{
|
||||
Black: "black",
|
||||
Red: "red",
|
||||
Green: "green",
|
||||
Yellow: "yellow",
|
||||
Blue: "blue",
|
||||
Magenta: "magenta",
|
||||
Cyan: "cyan",
|
||||
White: "white",
|
||||
}
|
||||
for i, txt := range colorToText {
|
||||
cl := Color(i)
|
||||
if cl != None {
|
||||
Foreground(cl, false)
|
||||
fmt.Print(txt, ",")
|
||||
Foreground(cl, true)
|
||||
fmt.Print(txt, ",")
|
||||
}
|
||||
}
|
||||
fmt.Println()
|
||||
}
|
||||
|
||||
func TestBackground(t *testing.T) {
|
||||
ResetColor()
|
||||
defer ResetColor()
|
||||
|
||||
fmt.Println("Please check the words under the following text shows with the corresponding background color:")
|
||||
|
||||
colorToText := [...]string{
|
||||
Black: "black",
|
||||
Red: "red",
|
||||
Green: "green",
|
||||
Yellow: "yellow",
|
||||
Blue: "blue",
|
||||
Magenta: "magenta",
|
||||
Cyan: "cyan",
|
||||
White: "white",
|
||||
}
|
||||
for i, txt := range colorToText {
|
||||
cl := Color(i)
|
||||
if cl != None {
|
||||
Background(cl, false)
|
||||
fmt.Print(txt, ",")
|
||||
Background(cl, true)
|
||||
fmt.Print(txt, ",")
|
||||
}
|
||||
}
|
||||
fmt.Println()
|
||||
}
|
||||
@@ -0,0 +1,133 @@
|
||||
// +build windows
|
||||
|
||||
package ct
|
||||
|
||||
import (
|
||||
"syscall"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
var fg_colors = []uint16{
|
||||
0,
|
||||
0,
|
||||
foreground_red,
|
||||
foreground_green,
|
||||
foreground_red | foreground_green,
|
||||
foreground_blue,
|
||||
foreground_red | foreground_blue,
|
||||
foreground_green | foreground_blue,
|
||||
foreground_red | foreground_green | foreground_blue}
|
||||
|
||||
var bg_colors = []uint16{
|
||||
0,
|
||||
0,
|
||||
background_red,
|
||||
background_green,
|
||||
background_red | background_green,
|
||||
background_blue,
|
||||
background_red | background_blue,
|
||||
background_green | background_blue,
|
||||
background_red | background_green | background_blue}
|
||||
|
||||
const (
|
||||
foreground_blue = uint16(0x0001)
|
||||
foreground_green = uint16(0x0002)
|
||||
foreground_red = uint16(0x0004)
|
||||
foreground_intensity = uint16(0x0008)
|
||||
background_blue = uint16(0x0010)
|
||||
background_green = uint16(0x0020)
|
||||
background_red = uint16(0x0040)
|
||||
background_intensity = uint16(0x0080)
|
||||
|
||||
foreground_mask = foreground_blue | foreground_green | foreground_red | foreground_intensity
|
||||
background_mask = background_blue | background_green | background_red | background_intensity
|
||||
)
|
||||
|
||||
var (
|
||||
kernel32 = syscall.NewLazyDLL("kernel32.dll")
|
||||
|
||||
procGetStdHandle = kernel32.NewProc("GetStdHandle")
|
||||
procSetConsoleTextAttribute = kernel32.NewProc("SetConsoleTextAttribute")
|
||||
procGetConsoleScreenBufferInfo = kernel32.NewProc("GetConsoleScreenBufferInfo")
|
||||
|
||||
hStdout uintptr
|
||||
initScreenInfo *console_screen_buffer_info
|
||||
)
|
||||
|
||||
func setConsoleTextAttribute(hConsoleOutput uintptr, wAttributes uint16) bool {
|
||||
ret, _, _ := procSetConsoleTextAttribute.Call(
|
||||
hConsoleOutput,
|
||||
uintptr(wAttributes))
|
||||
return ret != 0
|
||||
}
|
||||
|
||||
type coord struct {
|
||||
X, Y int16
|
||||
}
|
||||
|
||||
type small_rect struct {
|
||||
Left, Top, Right, Bottom int16
|
||||
}
|
||||
|
||||
type console_screen_buffer_info struct {
|
||||
DwSize coord
|
||||
DwCursorPosition coord
|
||||
WAttributes uint16
|
||||
SrWindow small_rect
|
||||
DwMaximumWindowSize coord
|
||||
}
|
||||
|
||||
func getConsoleScreenBufferInfo(hConsoleOutput uintptr) *console_screen_buffer_info {
|
||||
var csbi console_screen_buffer_info
|
||||
if ret, _, _ := procGetConsoleScreenBufferInfo.Call(hConsoleOutput, uintptr(unsafe.Pointer(&csbi))); ret == 0 {
|
||||
return nil
|
||||
}
|
||||
return &csbi
|
||||
}
|
||||
|
||||
const (
|
||||
std_output_handle = uint32(-11 & 0xFFFFFFFF)
|
||||
)
|
||||
|
||||
func init() {
|
||||
kernel32 := syscall.NewLazyDLL("kernel32.dll")
|
||||
|
||||
procGetStdHandle = kernel32.NewProc("GetStdHandle")
|
||||
|
||||
hStdout, _, _ = procGetStdHandle.Call(uintptr(std_output_handle))
|
||||
|
||||
initScreenInfo = getConsoleScreenBufferInfo(hStdout)
|
||||
|
||||
syscall.LoadDLL("")
|
||||
}
|
||||
|
||||
func resetColor() {
|
||||
if initScreenInfo == nil { // No console info - Ex: stdout redirection
|
||||
return
|
||||
}
|
||||
setConsoleTextAttribute(hStdout, initScreenInfo.WAttributes)
|
||||
}
|
||||
|
||||
func changeColor(fg Color, fgBright bool, bg Color, bgBright bool) {
|
||||
attr := uint16(0)
|
||||
if fg == None || bg == None {
|
||||
cbufinfo := getConsoleScreenBufferInfo(hStdout)
|
||||
if cbufinfo == nil { // No console info - Ex: stdout redirection
|
||||
return
|
||||
}
|
||||
attr = cbufinfo.WAttributes
|
||||
}
|
||||
if fg != None {
|
||||
attr = attr & ^foreground_mask | fg_colors[fg]
|
||||
if fgBright {
|
||||
attr |= foreground_intensity
|
||||
}
|
||||
}
|
||||
if bg != None {
|
||||
attr = attr & ^background_mask | bg_colors[bg]
|
||||
if bgBright {
|
||||
attr |= background_intensity
|
||||
}
|
||||
}
|
||||
setConsoleTextAttribute(hStdout, attr)
|
||||
}
|
||||
@@ -0,0 +1,40 @@
|
||||
// Package ctfmt is a handful wrapping of go-colortext (ct) package and fmt package.
|
||||
package ctfmt
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/daviddengcn/go-colortext"
|
||||
)
|
||||
|
||||
// Print calls fmt.Print with foreground color set.
|
||||
func Print(cl ct.Color, bright bool, a ...interface{}) (n int, err error) {
|
||||
ct.Foreground(cl, bright)
|
||||
defer ct.ResetColor()
|
||||
|
||||
return fmt.Print(a...)
|
||||
}
|
||||
|
||||
// Println calls fmt.Println with foreground color set.
|
||||
func Println(cl ct.Color, bright bool, a ...interface{}) (n int, err error) {
|
||||
ct.Foreground(cl, bright)
|
||||
defer ct.ResetColor()
|
||||
|
||||
return fmt.Println(a...)
|
||||
}
|
||||
|
||||
// Printf calls fmt.Printf with foreground color set.
|
||||
func Printf(cl ct.Color, bright bool, format string, a ...interface{}) (n int, err error) {
|
||||
ct.Foreground(cl, bright)
|
||||
defer ct.ResetColor()
|
||||
|
||||
return fmt.Printf(format, a...)
|
||||
}
|
||||
|
||||
// Printfln calls fmt.Printf and add an extra new-line char with foreground color set.
|
||||
func Printfln(cl ct.Color, bright bool, format string, a ...interface{}) (n int, err error) {
|
||||
ct.Foreground(cl, bright)
|
||||
defer ct.ResetColor()
|
||||
|
||||
return fmt.Printf(format+"\n", a...)
|
||||
}
|
||||
@@ -0,0 +1,8 @@
|
||||
module github.com/daviddengcn/go-colortext
|
||||
|
||||
go 1.14
|
||||
|
||||
require (
|
||||
github.com/golangplus/bytes v1.0.0 // indirect
|
||||
github.com/golangplus/testing v1.0.0
|
||||
)
|
||||
@@ -0,0 +1,8 @@
|
||||
github.com/golangplus/bytes v0.0.0-20160111154220-45c989fe5450 h1:7xqw01UYS+KCI25bMrPxwNYkSns2Db1ziQPpVq99FpE=
|
||||
github.com/golangplus/bytes v0.0.0-20160111154220-45c989fe5450/go.mod h1:Bk6SMAONeMXrxql8uvOKuAZSu8aM5RUGv+1C6IJaEho=
|
||||
github.com/golangplus/bytes v1.0.0 h1:YQKBijBVMsBxIiXT4IEhlKR2zHohjEqPole4umyDX+c=
|
||||
github.com/golangplus/bytes v1.0.0/go.mod h1:AdRaCFwmc/00ZzELMWb01soso6W1R/++O1XL80yAn+A=
|
||||
github.com/golangplus/fmt v1.0.0 h1:FnUKtw86lXIPfBMc3FimNF3+ABcV+aH5F17OOitTN+E=
|
||||
github.com/golangplus/fmt v1.0.0/go.mod h1:zpM0OfbMCjPtd2qkTD/jX2MgiFCqklhSUFyDW44gVQE=
|
||||
github.com/golangplus/testing v1.0.0 h1:+ZeeiKZENNOMkTTELoSySazi+XaEhVO0mb+eanrSEUQ=
|
||||
github.com/golangplus/testing v1.0.0/go.mod h1:ZDreixUV3YzhoVraIDyOzHrr76p6NUh6k/pPg/Q3gYA=
|
||||
@@ -0,0 +1,4 @@
|
||||
.imdone/
|
||||
*.out
|
||||
vendor/**
|
||||
!vendor/vendor.json
|
||||
@@ -0,0 +1,9 @@
|
||||
language: go
|
||||
go:
|
||||
- tip
|
||||
before_install:
|
||||
- go get -t -v ./...
|
||||
script:
|
||||
- go test -race -coverprofile=coverage.txt -covermode=atomic
|
||||
after_success:
|
||||
- bash <(curl -s https://codecov.io/bash)
|
||||
@@ -0,0 +1,21 @@
|
||||
The MIT License (MIT)
|
||||
|
||||
Copyright (c) 2016 Will Dixon
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
@@ -0,0 +1,167 @@
|
||||
# WMenu[](https://travis-ci.org/dixonwille/wmenu) [](https://codecov.io/gh/dixonwille/wmenu)
|
||||
|
||||
Package wmenu creates menus for cli programs. It uses wlog for it's interface
|
||||
with the command line. It uses os.Stdin, os.Stdout, and os.Stderr with
|
||||
concurrency by default. wmenu allows you to change the color of the different
|
||||
parts of the menu. This package also creates it's own error structure so you can
|
||||
type assert if you need to. wmenu will validate all responses before calling any function. It will also figure out which function should be called so you don't have to.
|
||||
|
||||
[](https://asciinema.org/a/4lv3ash3ubtnsclindvzdf320)
|
||||
|
||||
## Import
|
||||
|
||||
I try and keep up with my tags. To use the version and stable it is recommended to use `govendor` or another vendoring tool that allows you to build your project for specific tags.
|
||||
|
||||
```sh
|
||||
govendor fetch github.com/dixonwille/wmenu@v4
|
||||
```
|
||||
|
||||
The above will grab the latest v4 at that time and mark it. It will then be stable for you to use.
|
||||
|
||||
I will try to support as many versions as possable but please be patient.
|
||||
|
||||
### V1.0.0 - Major Release [](https://goreportcard.com/report/gopkg.in/dixonwille/wmenu.v1) [](https://godoc.org/gopkg.in/dixonwille/wmenu.v1)
|
||||
|
||||
### V2.0.0 - Allowing an interface to be passed in for options [](https://goreportcard.com/report/gopkg.in/dixonwille/wmenu.v2) [](https://godoc.org/gopkg.in/dixonwille/wmenu.v2)
|
||||
|
||||
### V3.0.0 - Pass in the option to that option's function [](https://goreportcard.com/report/gopkg.in/dixonwille/wmenu.v3) [](https://godoc.org/gopkg.in/dixonwille/wmenu.v3)
|
||||
|
||||
### V4.0.0 - Now have an Action that supports multiple options [](https://goreportcard.com/report/gopkg.in/dixonwille/wmenu.v4) [](https://godoc.org/gopkg.in/dixonwille/wmenu.v4)
|
||||
|
||||
|
||||
## Features
|
||||
|
||||
* Force single selection
|
||||
* Allow multiple selection
|
||||
* Change the delimiter
|
||||
* Change the color of different parts of the menu
|
||||
* Easily see which option(s) are default
|
||||
* Change the symbol used for default option(s)
|
||||
* Ask simple yes and no questions
|
||||
* Validate all responses before calling any functions
|
||||
* With yes and no can accept:
|
||||
* yes, Yes, YES, y, Y
|
||||
* no, No, NO, n, N
|
||||
* Figure out which Action should be called (Options, Default, or Multiple Action)
|
||||
* Re-ask question if invalid response up to a certain number of times
|
||||
* Can change max number of times to ask before failing output
|
||||
* Change reader and writer
|
||||
* Clear the screen whenever the menu is brought up
|
||||
* Has its own error structure so you can type assert menu errors
|
||||
|
||||
### V2 - Adds these Features
|
||||
|
||||
* Allowing any interface to be passed through for the options.
|
||||
|
||||
### V3 - Adds these Features
|
||||
|
||||
* Pass the option chosen to that options function
|
||||
|
||||
### V4 - Adds these Features
|
||||
|
||||
* Have one function for both single and multiple select. Allowing the user to an easier way of handeling the request.
|
||||
|
||||
## Usage
|
||||
This is a simple use of the package. (**NOTE: THIS IS A V4 SAMPLE**)
|
||||
|
||||
``` go
|
||||
menu := wmenu.NewMenu("What is your favorite food?")
|
||||
menu.Action(func (opts []wmenu.Opt) error {fmt.Printf(opts[0].Text + " is your favorite food."); return nil})
|
||||
menu.Option("Pizza", nil, true, nil)
|
||||
menu.Option("Ice Cream", nil, false, nil)
|
||||
menu.Option("Tacos", nil, false, func(opt wmenu.Opt) error {
|
||||
fmt.Printf("Tacos are great")
|
||||
})
|
||||
err := menu.Run()
|
||||
if err != nil{
|
||||
log.Fatal(err)
|
||||
}
|
||||
```
|
||||
|
||||
The output would look like this:
|
||||
|
||||
```
|
||||
0) *Pizza
|
||||
1) Ice Cream
|
||||
2) Tacos
|
||||
What is your favorite food?
|
||||
```
|
||||
|
||||
If the user just presses `[Enter]` then the option(s) with the `*` will be selected. This indicates that it is a default function. If they choose `1` then they would see `Ice Cream is your favorite food.`. This used the Action's function because the option selected didn't have a function along with it. But if they choose `2` they would see `Tacos are great`. That option did have a function with it which take precedence over Action.
|
||||
|
||||
You can you also use:
|
||||
|
||||
``` go
|
||||
menu.AllowMultiple()
|
||||
```
|
||||
|
||||
This will allow the user to select multiple options. The default delimiter is a `[space]`, but can be changed by using:
|
||||
|
||||
``` go
|
||||
menu.SetSeperator("some string")
|
||||
```
|
||||
|
||||
Another feature is the ability to ask yes or no questions.
|
||||
|
||||
``` go
|
||||
menu.IsYesNo(0)
|
||||
```
|
||||
|
||||
This will remove any options previously added options and hide the ones used for the menu. It will simply just ask yes or no. Menu will parse and validate the response for you. This option will always call the Action's function and pass in the option that was selected.
|
||||
|
||||
## V3+ - Release
|
||||
Allows the user to pass anything for the value so it can be retrieved later in the function. The following is to show case the power of this.
|
||||
|
||||
> The following was written in V3 but the concept holds for V4. V4 just changed `actFunc` to be `func([]wmenu.Opt) error` instead.
|
||||
|
||||
```go
|
||||
type NameEntity struct {
|
||||
FirstName string
|
||||
LastName string
|
||||
}
|
||||
|
||||
optFunc := func(opt wmenu.Opt) error {
|
||||
fmt.Println("Option 0 was chosen.")
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opt wmenu.Opt) error {
|
||||
name, ok := opt.Value.(NameEntity)
|
||||
if !ok {
|
||||
log.Fatal("Could not cast option's value to NameEntity")
|
||||
}
|
||||
fmt.Printf("%s has an id of %d.\n", opt.Text, opt.ID)
|
||||
fmt.Printf("Hello, %s %s.\n", name.FirstName, name.LastName)
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, os.Stdout, os.Stderr)
|
||||
menu.Action(actFunc)
|
||||
menu.Option("Option 0", NameEntity{"Bill", "Bob"}, true, optFunc)
|
||||
menu.Option("Option 1", NameEntity{"John", "Doe"}, false, nil)
|
||||
menu.Option("Option 2", NameEntity{"Jane", "Doe"}, false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
```
|
||||
|
||||
The immediate output would be:
|
||||
|
||||
```
|
||||
Output:
|
||||
0) *Option 0
|
||||
1) Option 1
|
||||
2) Option 2
|
||||
Choose an option.
|
||||
```
|
||||
|
||||
Now if the user pushes `[ENTER]` the output would be `Options 0 was chosen.`. But now if either option 1 or 2 were chosen it would cast the options value to a NameEntity allowing the function to be able to gather both the first name and last name of the NameEntity. If you want though you can just pass in `nil` as the value or even a string (`"hello"`) since both of these implement the empty interface required by value. Just make sure to cast the values so you can use them appropriately.
|
||||
|
||||
## Further Reading
|
||||
This whole package has been documented and has a few examples in:
|
||||
* [godocs V1](https://godoc.org/gopkg.in/dixonwille/wmenu.v1)
|
||||
* [godocs V2](https://godoc.org/gopkg.in/dixonwille/wmenu.v2)
|
||||
* [godocs V3](https://godoc.org/gopkg.in/dixonwille/wmenu.v3)
|
||||
* [godocs V4](https://godoc.org/gopkg.in/dixonwille/wmenu.v4)
|
||||
|
||||
You should read the docs to find all functions and structures at your finger tips.
|
||||
@@ -0,0 +1,36 @@
|
||||
package wmenu
|
||||
|
||||
import (
|
||||
"os"
|
||||
"os/exec"
|
||||
"runtime"
|
||||
)
|
||||
|
||||
var clear map[string]func()
|
||||
|
||||
func init() {
|
||||
clear = make(map[string]func())
|
||||
clear["linux"] = func() {
|
||||
cmd := exec.Command("clear")
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Run()
|
||||
}
|
||||
clear["darwin"] = func() {
|
||||
cmd := exec.Command("clear")
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Run()
|
||||
}
|
||||
clear["windows"] = func() {
|
||||
cmd := exec.Command("cmd", "/c", "cls")
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Run()
|
||||
}
|
||||
}
|
||||
|
||||
//Clear simply clears the command line interface (os.Stdout only).
|
||||
func Clear() {
|
||||
value, ok := clear[runtime.GOOS]
|
||||
if ok {
|
||||
value()
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,26 @@
|
||||
package wmenu
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestClear(t *testing.T) {
|
||||
Clear()
|
||||
}
|
||||
|
||||
func TestClearLinux(t *testing.T) {
|
||||
clearOs("linux")
|
||||
}
|
||||
|
||||
func TestClearDarwin(t *testing.T) {
|
||||
clearOs("darwin")
|
||||
}
|
||||
|
||||
func TestClearWindows(t *testing.T) {
|
||||
clearOs("windows")
|
||||
}
|
||||
|
||||
func clearOs(os string) {
|
||||
value, ok := clear[os]
|
||||
if ok {
|
||||
value()
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,86 @@
|
||||
package wmenu
|
||||
|
||||
import "errors"
|
||||
|
||||
var (
|
||||
//ErrInvalid is returned if a response from user was an invalid option
|
||||
ErrInvalid = errors.New("Invalid response")
|
||||
|
||||
//ErrTooMany is returned if multiSelect is false and a user tries to select multiple options
|
||||
ErrTooMany = errors.New("Too many responses")
|
||||
|
||||
//ErrNoResponse is returned if there were no responses and no action to call
|
||||
ErrNoResponse = errors.New("No response")
|
||||
|
||||
//ErrDuplicate is returned is a user selects an option twice
|
||||
ErrDuplicate = errors.New("Duplicated response")
|
||||
)
|
||||
|
||||
//MenuError records menu errors
|
||||
type MenuError struct {
|
||||
Err error
|
||||
Res string
|
||||
TriesLeft int
|
||||
}
|
||||
|
||||
//Error prints the error in an easy to read string.
|
||||
func (e *MenuError) Error() string {
|
||||
if e.Res != "" {
|
||||
return e.Err.Error() + ": " + e.Res
|
||||
}
|
||||
return e.Err.Error()
|
||||
}
|
||||
|
||||
func newMenuError(err error, res string, tries int) *MenuError {
|
||||
return &MenuError{
|
||||
Err: err,
|
||||
Res: res,
|
||||
TriesLeft: tries,
|
||||
}
|
||||
}
|
||||
|
||||
//IsInvalidErr checks to see if err is of type invalid error returned by menu.
|
||||
func IsInvalidErr(err error) bool {
|
||||
e, ok := err.(*MenuError)
|
||||
if ok && e.Err == ErrInvalid {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
//IsNoResponseErr checks to see if err is of type no response returned by menu.
|
||||
func IsNoResponseErr(err error) bool {
|
||||
e, ok := err.(*MenuError)
|
||||
if ok && e.Err == ErrNoResponse {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
//IsTooManyErr checks to see if err is of type too many returned by menu.
|
||||
func IsTooManyErr(err error) bool {
|
||||
e, ok := err.(*MenuError)
|
||||
if ok && e.Err == ErrTooMany {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
//IsDuplicateErr checks to see if err is of type duplicate returned by menu.
|
||||
func IsDuplicateErr(err error) bool {
|
||||
e, ok := err.(*MenuError)
|
||||
if ok && e.Err == ErrDuplicate {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
//IsMenuErr checks to see if it is a menu err.
|
||||
//This is a general check not a specific one.
|
||||
func IsMenuErr(err error) bool {
|
||||
_, ok := err.(*MenuError)
|
||||
if ok {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
@@ -0,0 +1,56 @@
|
||||
package wmenu
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
var (
|
||||
testInvalid = newMenuError(ErrInvalid, "1", 0)
|
||||
testDuplicate = newMenuError(ErrDuplicate, "2", 0)
|
||||
testTooMany = newMenuError(ErrTooMany, "", 0)
|
||||
testNoResponse = newMenuError(ErrNoResponse, "", 0)
|
||||
errNormal = errors.New("Opps")
|
||||
errMenu = newMenuError(errors.New("General"), "", 0)
|
||||
)
|
||||
|
||||
func TestIsInvalidErr(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
assert.True(IsInvalidErr(testInvalid))
|
||||
assert.False(IsInvalidErr(testDuplicate))
|
||||
assert.False(IsInvalidErr(errNormal))
|
||||
}
|
||||
|
||||
func TestIsDuplicateErr(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
assert.True(IsDuplicateErr(testDuplicate))
|
||||
assert.False(IsDuplicateErr(testInvalid))
|
||||
assert.False(IsDuplicateErr(errNormal))
|
||||
}
|
||||
|
||||
func TestIsTooManyErr(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
assert.True(IsTooManyErr(testTooMany))
|
||||
assert.False(IsTooManyErr(testDuplicate))
|
||||
assert.False(IsTooManyErr(errNormal))
|
||||
}
|
||||
|
||||
func TestIsNoResponseErr(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
assert.True(IsNoResponseErr(testNoResponse))
|
||||
assert.False(IsNoResponseErr(testDuplicate))
|
||||
assert.False(IsNoResponseErr(errNormal))
|
||||
}
|
||||
|
||||
func TestIsMenuError(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
assert.True(IsMenuErr(testNoResponse))
|
||||
assert.True(IsMenuErr(testDuplicate))
|
||||
assert.True(IsMenuErr(testInvalid))
|
||||
assert.True(IsMenuErr(testTooMany))
|
||||
assert.True(IsMenuErr(testNoResponse))
|
||||
assert.True(IsMenuErr(errMenu))
|
||||
assert.False(IsMenuErr(errNormal))
|
||||
}
|
||||
@@ -0,0 +1,76 @@
|
||||
//package main is to show how multiple menus can be used with eachother
|
||||
package main
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"github.com/dixonwille/wmenu"
|
||||
)
|
||||
|
||||
type menuItem int
|
||||
|
||||
const (
|
||||
pizza menuItem = iota
|
||||
iceCream
|
||||
tacos
|
||||
)
|
||||
|
||||
var menuItemStrings = map[menuItem]string{
|
||||
pizza: "Pizza",
|
||||
iceCream: "Ice Cream",
|
||||
tacos: "Tacos",
|
||||
}
|
||||
|
||||
func main() {
|
||||
mm := mainMenu()
|
||||
err := mm.Run()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
func mainMenu() *wmenu.Menu {
|
||||
menu := wmenu.NewMenu("What is your favorite food?")
|
||||
menu.Option(menuItemStrings[pizza], pizza, true, nil)
|
||||
menu.Option(menuItemStrings[iceCream], iceCream, false, nil)
|
||||
menu.Option(menuItemStrings[tacos], tacos, false, func(opt wmenu.Opt) error {
|
||||
fmt.Printf("Tacos are great!\n")
|
||||
return nil
|
||||
})
|
||||
menu.Action(func(opts []wmenu.Opt) error {
|
||||
if len(opts) != 1 {
|
||||
return errors.New("wrong number of options chosen")
|
||||
}
|
||||
|
||||
tm := toppingsMenu(opts[0].Value.(menuItem))
|
||||
return tm.Run()
|
||||
})
|
||||
return menu
|
||||
}
|
||||
|
||||
func toppingsMenu(favorite menuItem) *wmenu.Menu {
|
||||
menu := wmenu.NewMenu(fmt.Sprintf("What is your favorite topping for %s?", menuItemStrings[favorite]))
|
||||
if favorite == pizza {
|
||||
menu.Option("Meat", nil, true, nil)
|
||||
menu.Option("Cheese", nil, false, nil)
|
||||
menu.Option("Vegitables", nil, false, nil)
|
||||
}
|
||||
if favorite == iceCream {
|
||||
menu.Option("Fruit", nil, true, nil)
|
||||
menu.Option("Chocolate Syrup", nil, false, nil)
|
||||
menu.Option("Caramel Syrup", nil, false, nil)
|
||||
}
|
||||
menu.Action(func(opts []wmenu.Opt) error {
|
||||
if len(opts) != 1 {
|
||||
return errors.New("wrong number of options chosen")
|
||||
}
|
||||
|
||||
fmt.Printf("Your favorite food is %s with %s on top.\n", menuItemStrings[favorite], opts[0].Text)
|
||||
|
||||
return nil
|
||||
})
|
||||
return menu
|
||||
}
|
||||
@@ -0,0 +1,387 @@
|
||||
//Package wmenu creates menus for cli programs.
|
||||
//It uses wlog for it's interface with the command line.
|
||||
//It uses os.Stdin, os.Stdout, and os.Stderr with concurrency by default.
|
||||
//wmenu allows you to change the color of the different parts of the menu.
|
||||
//This package also creates it's own error structure so you can type assert if you need to.
|
||||
//wmenu will validate all responses before calling any function.
|
||||
//It will also figure out which function should be called so you don't have to.
|
||||
package wmenu
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/mattn/go-isatty"
|
||||
wlog "gopkg.in/dixonwille/wlog.v2"
|
||||
)
|
||||
|
||||
const (
|
||||
y = iota
|
||||
n
|
||||
)
|
||||
|
||||
var (
|
||||
NoColor = os.Getenv("TERM") == "dumb" ||
|
||||
(!isatty.IsTerminal(os.Stdout.Fd()) && !isatty.IsCygwinTerminal(os.Stdout.Fd()))
|
||||
)
|
||||
|
||||
//Menu is used to display options to a user.
|
||||
//A user can then select options and Menu will validate the response and perform the correct action.
|
||||
type Menu struct {
|
||||
question string
|
||||
function func([]Opt) error
|
||||
options []Opt
|
||||
ui wlog.UI
|
||||
multiSeparator string
|
||||
allowMultiple bool
|
||||
loopOnInvalid bool
|
||||
clear bool
|
||||
tries int
|
||||
defIcon string
|
||||
isYN bool
|
||||
ynDef int
|
||||
}
|
||||
|
||||
//NewMenu creates a menu with a wlog.UI as the writer.
|
||||
func NewMenu(question string) *Menu {
|
||||
//Create a default ui to use for menu
|
||||
var ui wlog.UI
|
||||
ui = wlog.New(os.Stdin, os.Stdout, os.Stderr)
|
||||
ui = wlog.AddConcurrent(ui)
|
||||
|
||||
return &Menu{
|
||||
question: question,
|
||||
function: nil,
|
||||
options: nil,
|
||||
ui: ui,
|
||||
multiSeparator: " ",
|
||||
allowMultiple: false,
|
||||
loopOnInvalid: false,
|
||||
clear: false,
|
||||
tries: 3,
|
||||
defIcon: "*",
|
||||
isYN: false,
|
||||
ynDef: 0,
|
||||
}
|
||||
}
|
||||
|
||||
//AddColor will change the color of the menu items.
|
||||
//optionColor changes the color of the options.
|
||||
//questionColor changes the color of the questions.
|
||||
//errorColor changes the color of the question.
|
||||
//Use wlog.None if you do not want to change the color.
|
||||
func (m *Menu) AddColor(optionColor, questionColor, responseColor, errorColor wlog.Color) {
|
||||
if !NoColor {
|
||||
m.ui = wlog.AddColor(questionColor, errorColor, wlog.None, wlog.None, optionColor, responseColor, wlog.None, wlog.None, wlog.None, m.ui)
|
||||
}
|
||||
}
|
||||
|
||||
//ClearOnMenuRun will clear the screen when a menu is ran.
|
||||
//This is checked when LoopOnInvalid is activated.
|
||||
//Meaning if an error occurred then it will clear the screen before asking again.
|
||||
func (m *Menu) ClearOnMenuRun() {
|
||||
m.clear = true
|
||||
}
|
||||
|
||||
//SetSeparator sets the separator to use when multiple options are valid responses.
|
||||
//Default value is a space.
|
||||
func (m *Menu) SetSeparator(sep string) {
|
||||
m.multiSeparator = sep
|
||||
}
|
||||
|
||||
//SetTries sets the number of tries on the loop before failing out.
|
||||
//Default is 3.
|
||||
//Negative values act like 0.
|
||||
func (m *Menu) SetTries(i int) {
|
||||
m.tries = i
|
||||
}
|
||||
|
||||
//LoopOnInvalid is used if an invalid option was given then it will prompt the user again.
|
||||
func (m *Menu) LoopOnInvalid() {
|
||||
m.loopOnInvalid = true
|
||||
}
|
||||
|
||||
//SetDefaultIcon sets the icon used to identify which options will be selected by default
|
||||
func (m *Menu) SetDefaultIcon(icon string) {
|
||||
m.defIcon = icon
|
||||
}
|
||||
|
||||
//IsYesNo sets the menu to a yes/no state.
|
||||
//Does not show options but does ask question.
|
||||
//Will also parse the answer to allow for all variants of yes/no (IE Y yes No ...)
|
||||
//Specify the default value using def. 0 is for yes and 1 is for no.
|
||||
//Both will call the Action function you specified.
|
||||
// Opt{ID: 0, Text: "y"} for yes and Opt{ID: 1, Text: "n"} for no will be passed to the function.
|
||||
func (m *Menu) IsYesNo(def int) {
|
||||
m.isYN = true
|
||||
m.ynDef = def
|
||||
}
|
||||
|
||||
//Option adds an option to the menu for the user to select from.
|
||||
//value is an empty interface that can be used to pass anything through to the function.
|
||||
//title is the string the user will select
|
||||
//isDefault is whether this option is a default option (IE when no options are selected).
|
||||
//function is what is called when only this option is selected.
|
||||
//If function is nil then it will default to the menu's Action.
|
||||
func (m *Menu) Option(title string, value interface{}, isDefault bool, function func(Opt) error) {
|
||||
option := newOption(len(m.options), title, value, isDefault, function)
|
||||
m.options = append(m.options, *option)
|
||||
}
|
||||
|
||||
//Action adds a default action to use in certain scenarios.
|
||||
//If the selected option (by default or user selected) does not have a function applied to it this will be called.
|
||||
//If there are no default options and no option was selected this will be called with an option that has an ID of -1.
|
||||
func (m *Menu) Action(function func([]Opt) error) {
|
||||
m.function = function
|
||||
}
|
||||
|
||||
//AllowMultiple will tell the menu to allow multiple selections.
|
||||
//The menu will fail if this is not called and mulple selections were selected.
|
||||
func (m *Menu) AllowMultiple() {
|
||||
m.allowMultiple = true
|
||||
}
|
||||
|
||||
//ChangeReaderWriter changes where the menu listens and writes to.
|
||||
//reader is where user input is collected.
|
||||
//writer and errorWriter is where the menu should write to.
|
||||
func (m *Menu) ChangeReaderWriter(reader io.Reader, writer, errorWriter io.Writer) {
|
||||
var ui wlog.UI
|
||||
ui = wlog.New(reader, writer, errorWriter)
|
||||
m.ui = ui
|
||||
}
|
||||
|
||||
//Run is used to execute the menu.
|
||||
//It will print to options and question to the screen.
|
||||
//It will only clear the screen if ClearOnMenuRun is activated.
|
||||
//This will validate all responses.
|
||||
//Errors are of type MenuError.
|
||||
func (m *Menu) Run() error {
|
||||
if m.clear {
|
||||
Clear()
|
||||
}
|
||||
valid := false
|
||||
var options []Opt
|
||||
//Loop and on error check if loopOnInvalid is enabled.
|
||||
//If it is Clear the screen and write error.
|
||||
//Then ask again
|
||||
for !valid {
|
||||
//step 1 print options to screen
|
||||
m.print()
|
||||
//step 2 ask question, get and validate response
|
||||
opt, err := m.ask()
|
||||
if err != nil {
|
||||
m.tries = m.tries - 1
|
||||
if !IsMenuErr(err) {
|
||||
err = newMenuError(err, "", m.triesLeft())
|
||||
}
|
||||
if m.loopOnInvalid && m.tries > 0 {
|
||||
if m.clear {
|
||||
Clear()
|
||||
}
|
||||
m.ui.Error(err.Error())
|
||||
} else {
|
||||
return err
|
||||
}
|
||||
} else {
|
||||
options = opt
|
||||
valid = true
|
||||
}
|
||||
}
|
||||
//step 3 call appropriate action with the responses
|
||||
return m.callAppropriate(options)
|
||||
}
|
||||
|
||||
func (m *Menu) callAppropriate(options []Opt) (err error) {
|
||||
if len(options) == 0 {
|
||||
return m.callAppropriateNoOptions()
|
||||
}
|
||||
if len(options) == 1 && options[0].function != nil {
|
||||
return options[0].function(options[0])
|
||||
}
|
||||
return m.function(options)
|
||||
}
|
||||
|
||||
func (m *Menu) callAppropriateNoOptions() (err error) {
|
||||
options := m.getDefault()
|
||||
if len(options) == 0 {
|
||||
return m.function([]Opt{{ID: -1}})
|
||||
}
|
||||
if len(options) == 1 && options[0].function != nil {
|
||||
return options[0].function(options[0])
|
||||
}
|
||||
return m.function(options)
|
||||
}
|
||||
|
||||
//hide options when this is a yes or no
|
||||
func (m *Menu) print() {
|
||||
if !m.isYN {
|
||||
for _, opt := range m.options {
|
||||
icon := m.defIcon
|
||||
if !opt.isDefault {
|
||||
icon = ""
|
||||
}
|
||||
m.ui.Output(fmt.Sprintf("%d) %s%s", opt.ID, icon, opt.Text))
|
||||
}
|
||||
} else {
|
||||
//TODO Allow user to specify what to use as value for YN options
|
||||
m.options = []Opt{}
|
||||
m.Option("y", "yes", m.ynDef == y, nil)
|
||||
m.Option("n", "no", m.ynDef == n, nil)
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Menu) ask() ([]Opt, error) {
|
||||
if m.isYN {
|
||||
if m.ynDef == y {
|
||||
m.question += " (Y/n)"
|
||||
} else {
|
||||
m.question += " (y/N)"
|
||||
}
|
||||
}
|
||||
trim := ""
|
||||
if m.multiSeparator == " " {
|
||||
trim = m.multiSeparator
|
||||
} else {
|
||||
trim = m.multiSeparator + " "
|
||||
}
|
||||
res, err := m.ui.Ask(m.question, trim)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
//Validate responses
|
||||
//Check if no responses are returned and no action to call
|
||||
if res == "" {
|
||||
//get default options
|
||||
opt := m.getDefault()
|
||||
if !m.validOptAndFunc(opt) {
|
||||
return nil, newMenuError(ErrNoResponse, "", m.triesLeft())
|
||||
}
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
var responses []int
|
||||
if !m.isYN {
|
||||
responses, err = m.resToInt(res)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = m.validateResponses(responses)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
} else {
|
||||
responses, err = m.ynResParse(res)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
//Parse responses and return them as options
|
||||
var finalOptions []Opt
|
||||
for _, response := range responses {
|
||||
finalOptions = append(finalOptions, m.options[response])
|
||||
}
|
||||
|
||||
return finalOptions, nil
|
||||
}
|
||||
|
||||
//Converts the response string to a slice of ints, also validates along the way.
|
||||
func (m *Menu) resToInt(res string) ([]int, error) {
|
||||
resStrings := strings.Split(res, m.multiSeparator)
|
||||
//Check if we don't want multiple responses
|
||||
if !m.allowMultiple && len(resStrings) > 1 {
|
||||
return nil, newMenuError(ErrTooMany, "", m.triesLeft())
|
||||
}
|
||||
|
||||
//Convert responses to intigers
|
||||
var responses []int
|
||||
for _, response := range resStrings {
|
||||
//Check if it is an intiger
|
||||
response = strings.Trim(response, " ")
|
||||
r, err := strconv.Atoi(response)
|
||||
if err != nil {
|
||||
return nil, newMenuError(ErrInvalid, response, m.triesLeft())
|
||||
}
|
||||
responses = append(responses, r)
|
||||
}
|
||||
return responses, nil
|
||||
}
|
||||
|
||||
func (m *Menu) ynResParse(res string) ([]int, error) {
|
||||
resStrings := strings.Split(res, m.multiSeparator)
|
||||
if len(resStrings) > 1 {
|
||||
return nil, newMenuError(ErrTooMany, "", m.triesLeft())
|
||||
}
|
||||
re := regexp.MustCompile("^\\s*(?:([Yy])(?:es|ES)?|([Nn])(?:o|O)?)\\s*$")
|
||||
matches := re.FindStringSubmatch(res)
|
||||
if len(matches) < 2 {
|
||||
return nil, newMenuError(ErrInvalid, res, m.triesLeft())
|
||||
}
|
||||
if strings.ToLower(matches[1]) == "y" {
|
||||
return []int{y}, nil
|
||||
}
|
||||
return []int{n}, nil
|
||||
}
|
||||
|
||||
//Check if response is in the range of options
|
||||
//If it is make sure it is not duplicated
|
||||
func (m *Menu) validateResponses(responses []int) error {
|
||||
var tmp []int
|
||||
for _, response := range responses {
|
||||
if response < 0 || len(m.options)-1 < response {
|
||||
return newMenuError(ErrInvalid, strconv.Itoa(response), m.triesLeft())
|
||||
}
|
||||
|
||||
if exist(tmp, response) {
|
||||
return newMenuError(ErrDuplicate, strconv.Itoa(response), m.triesLeft())
|
||||
}
|
||||
|
||||
tmp = append(tmp, response)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
//Simply checks if number exists in the slice
|
||||
func exist(slice []int, number int) bool {
|
||||
for _, s := range slice {
|
||||
if number == s {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
//gets a list of default options
|
||||
func (m *Menu) getDefault() []Opt {
|
||||
var opt []Opt
|
||||
for _, o := range m.options {
|
||||
if o.isDefault {
|
||||
opt = append(opt, o)
|
||||
}
|
||||
}
|
||||
return opt
|
||||
}
|
||||
|
||||
//make sure that there is an action available to be called in certain cases
|
||||
//returns false if it chould not find an action for the number options available
|
||||
func (m *Menu) validOptAndFunc(opt []Opt) bool {
|
||||
if m.function == nil {
|
||||
if len(opt) == 1 && opt[0].function != nil {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func (m *Menu) triesLeft() int {
|
||||
if m.loopOnInvalid && m.tries > 0 {
|
||||
return m.tries
|
||||
}
|
||||
return 0
|
||||
}
|
||||
@@ -0,0 +1,710 @@
|
||||
package wmenu
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"errors"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
wlog "gopkg.in/dixonwille/wlog.v2"
|
||||
)
|
||||
|
||||
var newMenuCases = []string{"Testing this menu.", "", "!@#$%^&*()"}
|
||||
var setSeperatorCases = []string{"", ".", ",", "~"}
|
||||
var defaultIconCases = []string{"", ",", "*", "~", "!", "@"}
|
||||
var setTriesCases = []int{0, -4, 5}
|
||||
var optionCases = []struct {
|
||||
name string
|
||||
value interface{}
|
||||
def bool
|
||||
function func(Opt) error
|
||||
}{
|
||||
{"Options", "Options", true, func(Opt) error { fmt.Println("testing option"); return nil }},
|
||||
{"", "Options", true, func(Opt) error { fmt.Println("testing option"); return nil }},
|
||||
{"Options", "", false, func(Opt) error { fmt.Println("testing option"); return nil }},
|
||||
{"", "", false, nil},
|
||||
{"", nil, false, nil},
|
||||
}
|
||||
var actionCases = []func([]Opt) error{
|
||||
func(opts []Opt) error { fmt.Println(opts); return nil },
|
||||
nil,
|
||||
}
|
||||
var addColorCases = []struct {
|
||||
opt wlog.Color
|
||||
que wlog.Color
|
||||
res wlog.Color
|
||||
err wlog.Color
|
||||
}{
|
||||
{wlog.None, wlog.None, wlog.None, wlog.None},
|
||||
{wlog.Red, wlog.Green, wlog.Blue, wlog.Black},
|
||||
{wlog.Yellow, wlog.White, wlog.Cyan, wlog.Magenta},
|
||||
{wlog.BrightBlack, wlog.BrightBlue, wlog.BrightCyan, wlog.BrightGreen},
|
||||
{wlog.BrightRed, wlog.BrightWhite, wlog.BrightYellow, wlog.BrightMagenta},
|
||||
}
|
||||
var errorCases = []struct {
|
||||
input string
|
||||
optFunction func(option Opt) error
|
||||
defFunction func([]Opt) error
|
||||
expected string
|
||||
singDef bool
|
||||
multiDef bool
|
||||
}{
|
||||
{"0\r\n", func(option Opt) error { return errors.New("Oops") }, nil, "Oops", false, false},
|
||||
{"0\r\n", nil, func(opts []Opt) error { return errors.New("Oops") }, "Oops", false, false},
|
||||
{"0 1\r\n", nil, func(opts []Opt) error { return errors.New("Oops") }, "Too many responses", false, false},
|
||||
{"\r\n", func(option Opt) error { return errors.New("Oops") }, nil, "Oops", true, false},
|
||||
{"\r\n", nil, func(opts []Opt) error { return errors.New("Oops") }, "Oops", true, false},
|
||||
{"\r\n", nil, func(opts []Opt) error { return errors.New("Oops") }, "Oops", false, false},
|
||||
{"\r\n", nil, func(opts []Opt) error { return errors.New("Oops") }, "Oops", true, true},
|
||||
}
|
||||
var ynCases = []struct {
|
||||
input string
|
||||
defFunc func([]Opt) error
|
||||
expected string
|
||||
def int
|
||||
}{
|
||||
{"\r\n", func(opts []Opt) error { return errors.New(opts[0].Text) }, "y", 0},
|
||||
{"\r\n", func(opts []Opt) error { return errors.New(opts[0].Text) }, "n", 1},
|
||||
{"YES\r\n", func(opts []Opt) error { return errors.New(opts[0].Text) }, "y", 0},
|
||||
{"n\r\n", func(opts []Opt) error { return errors.New(opts[0].Text) }, "n", 0},
|
||||
{" Yes \r\n", func(opts []Opt) error { return errors.New(opts[0].Text) }, "y", 0},
|
||||
{"ahh\r\n", func(opts []Opt) error { return errors.New(opts[0].Text) }, "Invalid response: ahh", 0},
|
||||
{"boo ahh\r\n", func(opts []Opt) error { return errors.New(opts[0].Text) }, "Too many responses", 0},
|
||||
}
|
||||
var trimCases = []struct {
|
||||
input string
|
||||
del string
|
||||
expected string
|
||||
}{
|
||||
{"0 \r\n", " ", "0\r\n"},
|
||||
{" 0\r\n", " ", "0\r\n"},
|
||||
{" 0 \r\n", " ", "0\r\n"},
|
||||
{"0 1 \r\n", " ", "0 1\r\n"},
|
||||
{" 0 1\r\n", " ", "0 1\r\n"},
|
||||
{" 0 1 \r\n", " ", "0 1\r\n"},
|
||||
{" 0, 1 \r\n", ",", "0 1\r\n"},
|
||||
{"0, 1,\r\n", ",", "0 1\r\n"},
|
||||
{"0, 1, \r\n", ",", "0 1\r\n"},
|
||||
}
|
||||
|
||||
func Example_optionValue() {
|
||||
type NameEntity struct {
|
||||
FirstName string
|
||||
LastName string
|
||||
}
|
||||
|
||||
reader := strings.NewReader("1\r\n") //Simulates the user typing "1" and hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
fmt.Println("Option 0 was chosen.")
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opts []Opt) error {
|
||||
name, ok := opts[0].Value.(NameEntity)
|
||||
if !ok {
|
||||
log.Fatal("Could not cast option's value to NameEntity")
|
||||
}
|
||||
fmt.Printf("%s has an id of %d.\n", opts[0].Text, opts[0].ID)
|
||||
fmt.Printf("Hello, %s %s.\n", name.FirstName, name.LastName)
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, os.Stdout, os.Stderr)
|
||||
menu.Action(actFunc)
|
||||
menu.Option("Option 0", NameEntity{"Bill", "Bob"}, true, optFunc)
|
||||
menu.Option("Option 1", NameEntity{"John", "Doe"}, false, nil)
|
||||
menu.Option("Option 2", NameEntity{"Jane", "Doe"}, false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
//Output:
|
||||
//0) *Option 0
|
||||
//1) Option 1
|
||||
//2) Option 2
|
||||
//Choose an option.
|
||||
//Option 1 has an id of 1.
|
||||
//Hello, John Doe.
|
||||
}
|
||||
|
||||
func Example_yesNo() {
|
||||
reader := strings.NewReader("y\r\n") //Simulates the user typing "y" and hitting the [enter] key
|
||||
actFunc := func(opts []Opt) error {
|
||||
fmt.Printf("%s has an id of %d.\n", opts[0].Text, opts[0].ID)
|
||||
fmt.Printf("But has a value of %s.\n", opts[0].Value.(string))
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Would you like to start?")
|
||||
menu.ChangeReaderWriter(reader, os.Stdout, os.Stderr)
|
||||
menu.Action(actFunc)
|
||||
menu.IsYesNo(0)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
//Output:
|
||||
//Would you like to start? (Y/n)
|
||||
//y has an id of 0.
|
||||
//But has a value of yes.
|
||||
}
|
||||
|
||||
func Example_simpleDefault() {
|
||||
reader := strings.NewReader("\r\n") //Simulates the user hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
fmt.Fprint(os.Stdout, "Option 0 was chosen.")
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, os.Stdout, os.Stderr)
|
||||
menu.Option("Option 0", "0", true, optFunc)
|
||||
menu.Option("Option 1", "1", false, nil)
|
||||
menu.Option("Option 2", "2", false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
//Output:
|
||||
//0) *Option 0
|
||||
//1) Option 1
|
||||
//2) Option 2
|
||||
//Choose an option.
|
||||
//Option 0 was chosen.
|
||||
}
|
||||
|
||||
func Example_multiple() {
|
||||
reader := strings.NewReader("1,2\r\n") //Simulates the user typing "1,2" and hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
fmt.Println("Option 0 was chosen.")
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opts []Opt) error {
|
||||
for _, opt := range opts {
|
||||
fmt.Printf("%s has an id of %d.\n", opt.Text, opt.ID)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, os.Stdout, os.Stderr)
|
||||
menu.Action(actFunc)
|
||||
menu.AllowMultiple()
|
||||
menu.SetSeparator(",")
|
||||
menu.Option("Option 0", "0", true, optFunc)
|
||||
menu.Option("Option 1", "1", false, nil)
|
||||
menu.Option("Option 2", "2", true, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
//Output:
|
||||
//0) *Option 0
|
||||
//1) Option 1
|
||||
//2) *Option 2
|
||||
//Choose an option.
|
||||
//Option 1 has an id of 1.
|
||||
//Option 2 has an id of 2.
|
||||
}
|
||||
|
||||
func Example_multipleDefault() {
|
||||
reader := strings.NewReader("\r\n") //Simulates the user hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
fmt.Println("Option 0 was chosen.")
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opts []Opt) error {
|
||||
for _, opt := range opts {
|
||||
fmt.Printf("%s has an id of %d.\n", opt.Text, opt.ID)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, os.Stdout, os.Stderr)
|
||||
menu.Action(actFunc)
|
||||
menu.AllowMultiple()
|
||||
menu.Option("Option 0", "0", true, optFunc)
|
||||
menu.Option("Option 1", "1", false, nil)
|
||||
menu.Option("Option 2", "2", true, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
//Output:
|
||||
//0) *Option 0
|
||||
//1) Option 1
|
||||
//2) *Option 2
|
||||
//Choose an option.
|
||||
//Option 0 has an id of 0.
|
||||
//Option 2 has an id of 2.
|
||||
}
|
||||
|
||||
func Example_errorNoResponse() {
|
||||
reader := strings.NewReader("\r\n") //Simulates the user hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
fmt.Println("Option 0 was chosen.")
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, os.Stdout, os.Stderr)
|
||||
menu.Option("Option 0", "0", false, optFunc)
|
||||
menu.Option("Option 1", "1", false, nil)
|
||||
menu.Option("Option 2", "2", false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
if IsNoResponseErr(err) {
|
||||
fmt.Println("We caught the err: " + err.Error())
|
||||
} else {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
//Output:
|
||||
//0) Option 0
|
||||
//1) Option 1
|
||||
//2) Option 2
|
||||
//Choose an option.
|
||||
//We caught the err: No response
|
||||
}
|
||||
|
||||
func Example_errorInvalid() {
|
||||
reader := strings.NewReader("3\r\n") //Simulates the user typing "3" and hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
fmt.Println("Option 0 was chosen.")
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, os.Stdout, os.Stderr)
|
||||
menu.Option("Option 0", "0", false, optFunc)
|
||||
menu.Option("Option 1", "1", false, nil)
|
||||
menu.Option("Option 2", "2", false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
if IsInvalidErr(err) {
|
||||
fmt.Println("We caught the err: " + err.Error())
|
||||
} else {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
//Output:
|
||||
//0) Option 0
|
||||
//1) Option 1
|
||||
//2) Option 2
|
||||
//Choose an option.
|
||||
//We caught the err: Invalid response: 3
|
||||
}
|
||||
|
||||
func Example_errorTooMany() {
|
||||
reader := strings.NewReader("1 2\r\n") //Simulates the user typing "1 2" and hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
fmt.Println("Option 0 was chosen.")
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, os.Stdout, os.Stderr)
|
||||
menu.Option("Option 0", "0", false, optFunc)
|
||||
menu.Option("Option 1", "1", false, nil)
|
||||
menu.Option("Option 2", "2", false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
if IsTooManyErr(err) {
|
||||
fmt.Println("We caught the err: " + err.Error())
|
||||
} else {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
//Output:
|
||||
//0) Option 0
|
||||
//1) Option 1
|
||||
//2) Option 2
|
||||
//Choose an option.
|
||||
//We caught the err: Too many responses
|
||||
}
|
||||
|
||||
func Example_errorDuplicate() {
|
||||
reader := strings.NewReader("1 1\r\n") //Simulates the user typing "1 1" and hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
fmt.Println("Option 0 was chosen.")
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opts []Opt) error {
|
||||
for _, opt := range opts {
|
||||
fmt.Printf("%s has an id of %d.\n", opt.Text, opt.ID)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, os.Stdout, os.Stderr)
|
||||
menu.Action(actFunc)
|
||||
menu.AllowMultiple()
|
||||
menu.Option("Option 0", "0", false, optFunc)
|
||||
menu.Option("Option 1", "1", false, nil)
|
||||
menu.Option("Option 2", "2", false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
if IsDuplicateErr(err) {
|
||||
fmt.Println("We caught the err: " + err.Error())
|
||||
} else {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
//Output:
|
||||
//0) Option 0
|
||||
//1) Option 1
|
||||
//2) Option 2
|
||||
//Choose an option.
|
||||
//We caught the err: Duplicated response: 1
|
||||
|
||||
}
|
||||
|
||||
func TestNewMenu(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
for _, c := range newMenuCases {
|
||||
menu := NewMenu(c)
|
||||
assert.Equal(c, menu.question)
|
||||
assert.Nil(menu.function)
|
||||
assert.Nil(menu.options)
|
||||
assert.Equal(" ", menu.multiSeparator)
|
||||
assert.False(menu.allowMultiple)
|
||||
assert.False(menu.loopOnInvalid)
|
||||
assert.False(menu.clear)
|
||||
assert.NotNil(menu.ui)
|
||||
assert.Equal(3, menu.tries)
|
||||
}
|
||||
}
|
||||
|
||||
func TestClearOnMenuRun(t *testing.T) {
|
||||
menu := NewMenu("Testing")
|
||||
menu.ClearOnMenuRun()
|
||||
assert.True(t, menu.clear)
|
||||
}
|
||||
|
||||
func TestSetSeperator(t *testing.T) {
|
||||
menu := NewMenu("Testing")
|
||||
for _, c := range setSeperatorCases {
|
||||
menu.SetSeparator(c)
|
||||
assert.Equal(t, c, menu.multiSeparator)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSetTries(t *testing.T) {
|
||||
menu := NewMenu("Testing")
|
||||
for _, c := range setTriesCases {
|
||||
menu.SetTries(c)
|
||||
assert.Equal(t, c, menu.tries)
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoopOnInvalid(t *testing.T) {
|
||||
menu := NewMenu("Testing")
|
||||
menu.LoopOnInvalid()
|
||||
assert.True(t, menu.loopOnInvalid)
|
||||
}
|
||||
|
||||
func TestOption(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
menu := NewMenu("Testing")
|
||||
for i, c := range optionCases {
|
||||
menu.Option(c.name, c.value, c.def, c.function)
|
||||
require.Equal(t, i+1, len(menu.options))
|
||||
assert.Equal(i, menu.options[i].ID)
|
||||
assert.Equal(c.name, menu.options[i].Text)
|
||||
assert.Equal(c.def, menu.options[i].isDefault)
|
||||
assert.Equal(c.value, menu.options[i].Value)
|
||||
if c.function != nil {
|
||||
assert.NotNil(menu.options[i].function)
|
||||
} else {
|
||||
assert.Nil(menu.options[i].function)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestAction(t *testing.T) {
|
||||
menu := NewMenu("Testing")
|
||||
for _, c := range actionCases {
|
||||
menu.Action(c)
|
||||
if c != nil {
|
||||
assert.NotNil(t, menu.function)
|
||||
} else {
|
||||
assert.Nil(t, menu.function)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestAllowMultiple(t *testing.T) {
|
||||
menu := NewMenu("Testing")
|
||||
menu.AllowMultiple()
|
||||
assert.True(t, menu.allowMultiple)
|
||||
}
|
||||
|
||||
func TestAddColor(t *testing.T) {
|
||||
menu := NewMenu("Testing")
|
||||
for _, c := range addColorCases {
|
||||
menu.AddColor(c.opt, c.que, c.res, c.err)
|
||||
//Nothing to assert on just make sure the function does not fail
|
||||
}
|
||||
}
|
||||
|
||||
func TestSetDefaultIcon(t *testing.T) {
|
||||
menu := NewMenu("Testing")
|
||||
for _, c := range defaultIconCases {
|
||||
menu.SetDefaultIcon(c)
|
||||
assert.Equal(t, c, menu.defIcon)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestClearInAsk(t *testing.T) {
|
||||
stdOut := initTest()
|
||||
reader := strings.NewReader("1\r\n") //Simulates the user typing "1" and hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
assert.Fail(t, "Should not have called Option 0's function")
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opts []Opt) error {
|
||||
assert.Len(t, opts, 1)
|
||||
assert.Equal(t, 1, opts[0].ID)
|
||||
assert.Equal(t, "Option 1", opts[0].Text)
|
||||
assert.Nil(t, opts[0].function)
|
||||
assert.False(t, opts[0].isDefault)
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, stdOut, stdOut)
|
||||
menu.Action(actFunc)
|
||||
menu.ClearOnMenuRun()
|
||||
menu.Option("Option 0", "0", true, optFunc)
|
||||
menu.Option("Option 1", "1", false, nil)
|
||||
menu.Option("Option 2", "2", true, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
assert.Fail(t, err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func TestDefaultAction(t *testing.T) {
|
||||
stdOut := initTest()
|
||||
reader := strings.NewReader("\r\n") //Simulates the user hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
assert.Fail(t, "Should not have called option 0's function")
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opts []Opt) error {
|
||||
assert.Len(t, opts, 1)
|
||||
assert.Equal(t, -1, opts[0].ID)
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, stdOut, stdOut)
|
||||
menu.Action(actFunc)
|
||||
menu.Option("Option 0", "0", false, optFunc)
|
||||
menu.Option("Option 1", "1", false, nil)
|
||||
menu.Option("Option 2", "2", false, nil)
|
||||
// spew.Dump(menu)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
assert.Fail(t, err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func TestDefaultActionWithDefaultOption(t *testing.T) {
|
||||
stdOut := initTest()
|
||||
reader := strings.NewReader("\r\n") //Simulates the user hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
assert.Fail(t, "Should not have called option 0's function")
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opts []Opt) error {
|
||||
assert.Len(t, opts, 1)
|
||||
assert.Equal(t, 1, opts[0].ID)
|
||||
assert.Equal(t, "Option 1", opts[0].Text)
|
||||
assert.Nil(t, opts[0].function)
|
||||
assert.True(t, opts[0].isDefault)
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, stdOut, stdOut)
|
||||
menu.Action(actFunc)
|
||||
menu.Option("Option 0", "0", false, optFunc)
|
||||
menu.Option("Option 1", "1", true, nil)
|
||||
menu.Option("Option 2", "2", false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
assert.Fail(t, err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func TestOptionsFunction(t *testing.T) {
|
||||
stdOut := initTest()
|
||||
reader := strings.NewReader("0\r\n") //Simulates the user typing "0" and hitting the [enter] key
|
||||
optFunc := func(option Opt) error {
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opts []Opt) error {
|
||||
assert.Fail(t, "Should not have called the menu's default function")
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, stdOut, stdOut)
|
||||
menu.Action(actFunc)
|
||||
menu.Option("Option 0", "0", false, optFunc)
|
||||
menu.Option("Option 1", "1", true, nil)
|
||||
menu.Option("Option 2", "2", false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
assert.Fail(t, err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func TestWlogAskErr(t *testing.T) {
|
||||
stdOut := initTest()
|
||||
reader := strings.NewReader("1") //Simulates the user typing "1" without hitting [enter]. Can't happen when reader is os.Stdin
|
||||
optFunc := func(option Opt) error {
|
||||
assert.Fail(t, "Should not have called option 0's function")
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opts []Opt) error {
|
||||
assert.Fail(t, "Should not have called the menu's default function")
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, stdOut, stdOut)
|
||||
menu.Action(actFunc)
|
||||
menu.Option("Option 0", "0", false, optFunc)
|
||||
menu.Option("Option 1", "1", true, nil)
|
||||
menu.Option("Option 2", "2", false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
assert.Equal(t, "EOF", err.Error())
|
||||
} else {
|
||||
assert.Fail(t, "Expected to get an error of EOF but instead got no error")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLetterForResponse(t *testing.T) {
|
||||
stdOut := initTest()
|
||||
reader := strings.NewReader("a\r\n") //Simulates the user typing "a" and hitting [enter].
|
||||
optFunc := func(option Opt) error {
|
||||
assert.Fail(t, "Should not have called option 0's function")
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opts []Opt) error {
|
||||
assert.Fail(t, "Should not have called the menu's default function")
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, stdOut, stdOut)
|
||||
menu.Action(actFunc)
|
||||
menu.Option("Option 0", "0", false, optFunc)
|
||||
menu.Option("Option 1", "1", true, nil)
|
||||
menu.Option("Option 2", "2", false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
require.True(t, IsInvalidErr(err))
|
||||
e := err.(*MenuError)
|
||||
assert.Equal(t, "a", e.Res)
|
||||
} else {
|
||||
assert.Fail(t, "Expected to get an Invalid Response error but instead got no error")
|
||||
}
|
||||
}
|
||||
|
||||
func TestActionError(t *testing.T) {
|
||||
stdout := initTest()
|
||||
for _, c := range errorCases {
|
||||
reader := strings.NewReader(c.input)
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, stdout, stdout)
|
||||
menu.Action(c.defFunction)
|
||||
if c.multiDef {
|
||||
menu.AllowMultiple()
|
||||
}
|
||||
menu.Option("Option 0", "0", c.singDef, c.optFunction)
|
||||
menu.Option("Option 1", "1", false, nil)
|
||||
menu.Option("Option 2", "2", c.multiDef, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
assert.Equal(t, c.expected, err.Error())
|
||||
} else {
|
||||
assert.Fail(t, "Expected an error but did not get one")
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestLoopAndTries(t *testing.T) {
|
||||
stdOut := initTest()
|
||||
optFunc := func(option Opt) error {
|
||||
assert.Fail(t, "Should not have called option 0's function")
|
||||
return nil
|
||||
}
|
||||
for _, c := range setTriesCases {
|
||||
reader := strings.NewReader("a") //Simulates the user typing "a" and not hitting [enter].
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, stdOut, stdOut)
|
||||
menu.SetTries(c)
|
||||
menu.LoopOnInvalid()
|
||||
menu.ClearOnMenuRun()
|
||||
menu.Option("Option 0", "0", false, optFunc)
|
||||
menu.Option("Option 1", "1", false, nil)
|
||||
menu.Option("Option 2", "2", false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
require.True(t, IsMenuErr(err))
|
||||
e := err.(*MenuError)
|
||||
assert.Equal(t, 0, e.TriesLeft)
|
||||
} else {
|
||||
assert.Fail(t, "Expected to get a general Menu error but instead got no error")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestYesNo(t *testing.T) {
|
||||
for _, c := range ynCases {
|
||||
stdOut := initTest()
|
||||
reader := strings.NewReader(c.input)
|
||||
menu := NewMenu("Yes or No")
|
||||
menu.ChangeReaderWriter(reader, stdOut, stdOut)
|
||||
menu.IsYesNo(c.def)
|
||||
menu.Action(c.defFunc)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
assert.Equal(t, c.expected, err.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestTrim(t *testing.T) {
|
||||
for _, c := range trimCases {
|
||||
stdOut := initTest()
|
||||
actOut := initTest()
|
||||
reader := strings.NewReader(c.input)
|
||||
menu := NewMenu("Testing")
|
||||
menu.ChangeReaderWriter(reader, stdOut, stdOut)
|
||||
menu.SetSeparator(c.del)
|
||||
menu.Action(func(opts []Opt) error {
|
||||
var actual []string
|
||||
for _, opt := range opts {
|
||||
actual = append(actual, opt.Text)
|
||||
}
|
||||
fmt.Fprintf(actOut, "%s\r\n", strings.Join(actual, " "))
|
||||
return nil
|
||||
})
|
||||
menu.AllowMultiple()
|
||||
menu.Option("0", nil, false, nil)
|
||||
menu.Option("1", nil, false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
assert.Fail(t, err.Error())
|
||||
}
|
||||
act, err := actOut.ReadString((byte)('\n'))
|
||||
if err != nil {
|
||||
assert.Fail(t, err.Error())
|
||||
}
|
||||
assert.Equal(t, c.expected, act)
|
||||
}
|
||||
}
|
||||
|
||||
func initTest() *bytes.Buffer {
|
||||
var b []byte
|
||||
return bytes.NewBuffer(b)
|
||||
}
|
||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user