mirror of
https://github.com/beego/bee.git
synced 2024-06-02 05:53:28 +00:00
Compare commits
461 Commits
Author | SHA1 | Date | |
---|---|---|---|
|
a106dd6ba2 | ||
|
c8ccbb7e70 | ||
|
07a06e75cc | ||
|
2d3dae7fa0 | ||
|
e879c37436 | ||
|
0db82ee3bb | ||
|
0420c775cf | ||
|
969ce1d1eb | ||
|
95cab0c26f | ||
|
01569d265c | ||
|
0cf9bdde10 | ||
|
5aa9764e55 | ||
|
ebb8c0ad31 | ||
|
8c91192dae | ||
|
d78f259fdf | ||
|
22ea0fa345 | ||
|
ef570ffb7b | ||
|
651d604cca | ||
|
f9d0af2146 | ||
|
afe1678459 | ||
|
a75a0c8723 | ||
|
97df75a28c | ||
|
d04f11c57a | ||
|
d7fcaf0921 | ||
|
7ee363657f | ||
|
1b4f90114a | ||
|
47e6e7c145 | ||
|
50ffa7ed8c | ||
|
8a299fa755 | ||
|
e3d0b640df | ||
|
8f8166b3da | ||
|
1a47e0e945 | ||
|
9a9dbc0d78 | ||
|
3b15b101e2 | ||
|
0d739375d1 | ||
|
c36f132fd8 | ||
|
e22201eae3 | ||
|
bdeabd3b70 | ||
|
112769d885 | ||
|
97e66f3d18 | ||
|
ee7b335232 | ||
|
5bbc43e153 | ||
|
d0cfcd96ba | ||
|
d63e07087c | ||
|
6b1c370fd0 | ||
|
7b3d1b5f64 | ||
|
457c2f7dc8 | ||
|
551d7cb8d3 | ||
|
665150ec64 | ||
|
f73ad3252b | ||
|
844823a367 | ||
|
cbb523325d | ||
|
22a4b07b4a | ||
|
0a690decfb | ||
|
21f42e22a4 | ||
|
9428a53176 | ||
|
824fa639f0 | ||
|
637637efb2 | ||
|
7f80477761 | ||
|
d995fc5328 | ||
|
02a66c7bd7 | ||
|
9541046a29 | ||
|
d0bc4f76fc | ||
|
100f0af03a | ||
|
3a727232d7 | ||
|
29e2203ba1 | ||
|
a44ef93f99 | ||
|
88101ce7a5 | ||
|
6f2d624a32 | ||
|
734d77f60d | ||
|
e9edae7563 | ||
|
7db43f16d5 | ||
|
6c17de3436 | ||
|
ea846e1ec6 | ||
|
ee56934371 | ||
|
f150956981 | ||
|
adbc0af183 | ||
|
639eccfef9 | ||
|
486ac49f38 | ||
|
9f04de468a | ||
|
31930ecd29 | ||
|
037f319c11 | ||
|
4eb58f534b | ||
|
3b2304a480 | ||
|
06e09b0f87 | ||
|
8eb7a2dfaf | ||
|
ebb2fee3fc | ||
|
d1611db053 | ||
|
c93bda1d3c | ||
|
642d933c4a | ||
|
c693549cc6 | ||
|
cdb41fad3f | ||
|
c619ccf1b3 | ||
|
9d68c302bd | ||
|
958e4564d1 | ||
|
e3e1a9849f | ||
|
32230de2ea | ||
|
b8213eba24 | ||
|
62abed87e9 | ||
|
9c423bbc97 | ||
|
1653a01eaa | ||
|
bc846b254e | ||
|
62c20031a1 | ||
|
a5fb1b4690 | ||
|
0080f54a08 | ||
|
36e532fa4b | ||
|
f2a93c8c75 | ||
|
22b64e8157 | ||
|
fa7aee4d01 | ||
|
1ed5c71087 | ||
|
29e5fadf6a | ||
|
9f1d5fc5eb | ||
|
e0fd237002 | ||
|
f5665162f7 | ||
|
cef7185b35 | ||
|
1605bacc9e | ||
|
3ea61a8926 | ||
|
674b52a732 | ||
|
7a7357a983 | ||
|
81ed06ea01 | ||
|
8f0badc630 | ||
|
8b942b1f22 | ||
|
d496d26a1c | ||
|
c0d29b0d2a | ||
|
b45ded3355 | ||
|
2c673b7d1e | ||
|
53b0b68d00 | ||
|
2862ff48db | ||
|
b56b3794e4 | ||
|
413f4ef129 | ||
|
f0815cc9b2 | ||
|
732fdfa321 | ||
|
82e40f9010 | ||
|
9c63635169 | ||
|
e0ea0abf5b | ||
|
07fa523da9 | ||
|
40a90af5be | ||
|
e539c34ea0 | ||
|
a23c76305a | ||
|
c562cedf96 | ||
|
a68e8ae3e8 | ||
|
e0ada8860d | ||
|
6a2f44720e | ||
|
3e5f9cf213 | ||
|
19d0116825 | ||
|
0a2c7f0c57 | ||
|
59fa4ace1e | ||
|
a9d3de0872 | ||
|
174d2ab2e8 | ||
|
dbb41fa430 | ||
|
f601e441f3 | ||
|
e0ce20407c | ||
|
0e6ce7dea0 | ||
|
13a81b8140 | ||
|
d9633cd9af | ||
|
5005bd4408 | ||
|
b90f93409e | ||
|
cc2b2d1054 | ||
|
4637afa242 | ||
|
f3240109bf | ||
|
8ac965a433 | ||
|
adf3cc850f | ||
|
91e4ac31a4 | ||
|
8893ad2d13 | ||
|
fccdcbbeda | ||
|
c37857ed01 | ||
|
20b7aec77a | ||
|
2c329fd606 | ||
|
cf3c3bdc8b | ||
|
4158ab284e | ||
|
4110083cae | ||
|
9db1e8fb4c | ||
|
8758f6eaa1 | ||
|
8cceb76836 | ||
|
bb5e0435c9 | ||
|
9433a7b66f | ||
|
a780721c88 | ||
|
8e91f22aeb | ||
|
1334a99810 | ||
|
ad695cd8c6 | ||
|
fc12eabbb6 | ||
|
8ebb7977f9 | ||
|
3121f64b8f | ||
|
141938e899 | ||
|
dfde62c411 | ||
|
f7419c24fc | ||
|
f5cc7abe8d | ||
|
280d71a799 | ||
|
b1dbdd023c | ||
|
8357e26a9f | ||
|
5d223c6513 | ||
|
81ed8514be | ||
|
c687a67d52 | ||
|
f694778078 | ||
|
6a86284cec | ||
|
6f1ff54713 | ||
|
1febc2de16 | ||
|
521e9a3bc0 | ||
|
4562f47f4d | ||
|
0f9b9ea345 | ||
|
36a17c40b0 | ||
|
384d46897b | ||
|
de82e68f73 | ||
|
231e90d274 | ||
|
4b7edb7235 | ||
c5099ba2a0 | |||
|
9f600ccb45 | ||
|
d06f56566c | ||
|
12019e22af | ||
|
f7120a52bf | ||
|
df31f04b26 | ||
|
ab854cc874 | ||
|
4d1304b78c | ||
|
22af6cc712 | ||
|
f2696160ae | ||
|
d24c05d83e | ||
|
79d6746aa2 | ||
|
19be52dc8c | ||
|
87a287cac2 | ||
|
8252ea9f88 | ||
|
babd8d04b4 | ||
|
d3a825f92d | ||
|
5ba90258a4 | ||
10a2e56df0 | |||
3d977edbb4 | |||
|
10bb0454f6 | ||
|
ebc19047f2 | ||
|
2ef722e758 | ||
|
db6c162b03 | ||
|
bf5480b2df | ||
|
e3e401cadd | ||
|
d63a5eb53d | ||
|
6689e25d6f | ||
|
cb2fae3e68 | ||
|
393e315ba5 | ||
|
f728b23527 | ||
|
5ed819a025 | ||
|
b05f8ace9e | ||
|
815bdd737a | ||
|
50e2cfbcbd | ||
|
1fb0bebe30 | ||
|
25a2684240 | ||
|
62388db728 | ||
|
f496e0f36f | ||
|
ecd8896cf3 | ||
|
2b0ceaf411 | ||
|
a386880832 | ||
|
7b2e95e9bd | ||
|
6dd625232d | ||
|
cd82742af9 | ||
|
25063a62c4 | ||
|
bb68873f45 | ||
|
5c1ee91097 | ||
|
c1896dd4cf | ||
|
cb47a5a5fe | ||
|
b0432d7c04 | ||
|
c23138b457 | ||
|
611eecc204 | ||
|
be6186155f | ||
|
685c16d5eb | ||
|
3469f0ae4c | ||
|
d217d0b85c | ||
|
4ca7777c32 | ||
|
4f53288e6a | ||
|
b2087846ab | ||
|
d23d3e5bfb | ||
|
5ba8b15e3a | ||
|
d70cedc7de | ||
|
418230b131 | ||
|
fdf0e9c768 | ||
|
1937c6ddb7 | ||
|
4ab8b8e30e | ||
|
bfeaf6c8df | ||
|
363abeeae2 | ||
|
76a4feb17e | ||
|
2265bfb0ff | ||
|
e75824f797 | ||
|
4711873e25 | ||
|
d8cb6ff47f | ||
|
7d78224b99 | ||
|
f8e9f67ae6 | ||
|
b94b47d7d5 | ||
|
c34cb05355 | ||
|
9dc796dc1e | ||
|
a5cddac554 | ||
|
65e995ca17 | ||
|
52e3087bb4 | ||
|
e90da8f77b | ||
|
ceefd74c9b | ||
|
fbbb713465 | ||
|
ae1f1a9bf4 | ||
|
2936241fe6 | ||
|
cdec05b5e3 | ||
|
0405523d35 | ||
|
b8040300b6 | ||
|
0a42575490 | ||
|
52c8cb8b6a | ||
|
58bfa5c1ec | ||
|
d0b3ede4ec | ||
|
2b04261f9c | ||
|
254174eda6 | ||
|
aae0cc4587 | ||
|
3b3da1655d | ||
|
d29f8e8ec3 | ||
|
899aede629 | ||
|
b65934a764 | ||
|
3523212500 | ||
|
1b03d81aa2 | ||
|
552111517e | ||
|
3673659178 | ||
|
042a4cf244 | ||
|
ea63bf253e | ||
|
416fb2ec9d | ||
|
5c3807e29a | ||
|
34faa8fca6 | ||
|
0099ef2f92 | ||
|
681fc57e16 | ||
|
812b8c4e5b | ||
|
f5471680e4 | ||
|
46fc9d75ae | ||
|
b4bee5cceb | ||
|
f657d22fc7 | ||
|
8b01334faf | ||
|
9e4a43f08a | ||
|
0000e7d7c9 | ||
|
0d0e583346 | ||
|
bacdca6037 | ||
|
69760a96ec | ||
|
e8c0f528a6 | ||
|
fef25c2c5f | ||
|
a1e8589798 | ||
|
49d791502a | ||
|
4ef471170e | ||
|
a7ff7b8615 | ||
|
50ae47c61f | ||
|
3f63cd706f | ||
|
65aec741fb | ||
|
e1f3353511 | ||
|
823dca76d5 | ||
|
a117a48d7b | ||
|
c76d4d1451 | ||
|
1a79d6dcca | ||
|
1cc7abaac6 | ||
|
963e1c9d05 | ||
|
e84d6f9a9c | ||
|
ffbe29ae22 | ||
|
6afbafa250 | ||
|
3daa0d2639 | ||
|
94553cdca7 | ||
|
096cef1b6c | ||
|
de2656749f | ||
|
c5bc543c13 | ||
|
db57fa50a0 | ||
|
8f8ece57b0 | ||
|
593ef9c47c | ||
|
3b316ecd81 | ||
|
05a49939e1 | ||
|
07ca21401c | ||
|
3b0dd10302 | ||
|
a353b6d69a | ||
|
1336d6b411 | ||
|
52a37a86a9 | ||
|
d9306e9fcf | ||
|
a22baf1b65 | ||
|
1e478bb32e | ||
|
f80384a9f5 | ||
|
5600cb1c36 | ||
|
4dd6983d3c | ||
|
bef9d4d5e8 | ||
|
abf3c41032 | ||
|
0cffaf8ea6 | ||
|
701b2e2ea8 | ||
|
e6bb3f893c | ||
|
e3a67ef07f | ||
|
db5616465b | ||
|
48c931d635 | ||
|
8c0742c140 | ||
|
072cc85708 | ||
|
72f144baad | ||
|
ad641afb34 | ||
|
427541fcd9 | ||
|
36436c0f53 | ||
|
69023e9ae0 | ||
|
32e7b144f9 | ||
|
7811c335e9 | ||
|
f9939bc286 | ||
|
9760ce6623 | ||
|
8d646b4be1 | ||
|
aa19ecedb0 | ||
|
77a36964e5 | ||
|
855ac34dd3 | ||
|
e543958fe3 | ||
|
01703caefa | ||
|
c01a15197e | ||
|
dd3f690bf7 | ||
|
20c6a26952 | ||
|
609ed8fcd5 | ||
|
fd4772fe82 | ||
|
64a8570c70 | ||
|
09f53c0400 | ||
|
792cb3f311 | ||
|
e57cc7b94e | ||
|
3be35d9d81 | ||
|
6394f3f80a | ||
|
426237fefe | ||
|
b3ea5dda0e | ||
|
4ef1a0e80e | ||
|
d1330dfc0b | ||
|
513debab13 | ||
|
feea8877c0 | ||
|
3d5b13d84e | ||
|
380cce650e | ||
|
78836f0a4c | ||
|
0e79cb596e | ||
|
abf2b4a4cb | ||
|
c782720cb7 | ||
|
dacee6ba16 | ||
|
6dce4df2cb | ||
|
39cc0bc04e | ||
|
c1a60c0ec3 | ||
|
c538bfbc8f | ||
|
74baba4f63 | ||
|
9db35a8270 | ||
|
cc74358320 | ||
|
1f6e89a670 | ||
|
932f16ba4e | ||
|
8eed14a851 | ||
|
38f8d6fc56 | ||
|
3b5e381a48 | ||
|
7bd1e415dd | ||
|
ee3dd8f436 | ||
|
c2161cac3f | ||
|
cf1809a64b | ||
|
14eeb07402 | ||
|
b867a57d65 | ||
|
268a9d8346 | ||
|
c53a621c52 | ||
|
496f162f9b | ||
|
62260c1033 | ||
|
c5f8b8d28e | ||
|
3f15b69bcc | ||
|
de62ae6043 | ||
|
24eee99d5b | ||
|
4ad45b3cd7 | ||
|
848618b21e | ||
|
f9d0b2f2fc | ||
|
0dbf0c9693 | ||
|
09e6fb26e2 | ||
|
20fdbe6a1a | ||
|
3c676b602c | ||
|
bbe1a4efe3 | ||
|
1799601be1 | ||
|
3a51d4830c | ||
|
e215969963 | ||
|
d801cc2469 | ||
|
059df76c3e | ||
|
d8b9d96d84 | ||
|
040b160ecd | ||
|
4ea1715df6 | ||
|
699d76bc95 | ||
|
81c6de6cb3 |
25
.github/workflows/go.yml
vendored
Normal file
25
.github/workflows/go.yml
vendored
Normal file
|
@ -0,0 +1,25 @@
|
||||||
|
name: Go
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches: [ develop ]
|
||||||
|
pull_request:
|
||||||
|
branches: [ develop ]
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
|
||||||
|
build:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v3
|
||||||
|
|
||||||
|
- name: Set up Go
|
||||||
|
uses: actions/setup-go@v3
|
||||||
|
with:
|
||||||
|
go-version: 1.18
|
||||||
|
|
||||||
|
- name: Build
|
||||||
|
run: go build -v ./...
|
||||||
|
|
||||||
|
- name: Test
|
||||||
|
run: go test -v ./...
|
2
.gitignore
vendored
2
.gitignore
vendored
|
@ -7,6 +7,7 @@
|
||||||
_obj
|
_obj
|
||||||
_test
|
_test
|
||||||
.idea
|
.idea
|
||||||
|
.vscode
|
||||||
|
|
||||||
# Architecture specific extensions/prefixes
|
# Architecture specific extensions/prefixes
|
||||||
*.[568vq]
|
*.[568vq]
|
||||||
|
@ -29,3 +30,4 @@ _testmain.go
|
||||||
bee
|
bee
|
||||||
*.exe~
|
*.exe~
|
||||||
.goxc.local.json
|
.goxc.local.json
|
||||||
|
vendor
|
||||||
|
|
|
@ -11,4 +11,4 @@
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"ConfigVersion": "0.9"
|
"ConfigVersion": "0.9"
|
||||||
}
|
}
|
||||||
|
|
24
.travis.yml
24
.travis.yml
|
@ -1,6 +1,22 @@
|
||||||
language: go
|
language: go
|
||||||
|
|
||||||
go:
|
go:
|
||||||
- 1.5.3
|
- 1.14.6
|
||||||
- 1.4.3
|
install:
|
||||||
- 1.3.3
|
- export PATH=$PATH:$HOME/gopath/bin
|
||||||
|
- go get -u github.com/opennota/check/cmd/structcheck
|
||||||
|
- go get -u honnef.co/go/tools/cmd/staticcheck
|
||||||
|
- go get -u github.com/mdempsky/unconvert
|
||||||
|
- go get -u github.com/gordonklaus/ineffassign
|
||||||
|
script:
|
||||||
|
- pwd
|
||||||
|
- cd $(dirname `dirname $(pwd)`)/beego/bee
|
||||||
|
- export GO111MODULE="on"
|
||||||
|
- go mod download
|
||||||
|
- go test -coverprofile=coverage.txt -covermode=atomic ./...
|
||||||
|
- find . ! \( -path './vendor' -prune \) -type f -name '*.go' -print0 | xargs -0 gofmt -l -s
|
||||||
|
- go list ./... | grep -v /vendor/ | grep -v /pkg/mod/
|
||||||
|
- go vet $(go list ./... | grep -v /vendor/ | grep -v /pkg/mod/ )
|
||||||
|
- structcheck $(go list ./... | grep -v /vendor/ | grep -v /pkg/mod/ )
|
||||||
|
- staticcheck $(go list ./... | grep -v /vendor/ | grep -v /pkg/mod/ )
|
||||||
|
- unconvert $(go list ./... | grep -v /vendor/ | grep -v /pkg/mod/ )
|
||||||
|
- ineffassign .
|
||||||
|
|
7
Beefile
7
Beefile
|
@ -1,9 +1,7 @@
|
||||||
version: 0
|
version: 0
|
||||||
gopm:
|
|
||||||
enable: false
|
|
||||||
install: false
|
|
||||||
go_install: false
|
go_install: false
|
||||||
watch_ext: []
|
watch_ext: [".go"]
|
||||||
|
watch_ext_static: [".html", ".tpl", ".js", ".css"]
|
||||||
dir_structure:
|
dir_structure:
|
||||||
watch_all: false
|
watch_all: false
|
||||||
controllers: ""
|
controllers: ""
|
||||||
|
@ -13,3 +11,4 @@ cmd_args: []
|
||||||
envs: []
|
envs: []
|
||||||
database:
|
database:
|
||||||
driver: "mysql"
|
driver: "mysql"
|
||||||
|
enable_reload: false
|
||||||
|
|
2
LICENSE
2
LICENSE
|
@ -188,4 +188,4 @@ third-party archives.
|
||||||
distributed under the License is distributed on an "AS IS" BASIS,
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
See the License for the specific language governing permissions and
|
See the License for the specific language governing permissions and
|
||||||
limitations under the License.
|
limitations under the License.
|
||||||
|
|
9
Makefile
9
Makefile
|
@ -1,3 +1,5 @@
|
||||||
|
VERSION = $(shell grep 'const version' cmd/commands/version/version.go | sed -E 's/.*"(.+)"$$/v\1/')
|
||||||
|
|
||||||
.PHONY: all test clean build install
|
.PHONY: all test clean build install
|
||||||
|
|
||||||
GOFLAGS ?= $(GOFLAGS:)
|
GOFLAGS ?= $(GOFLAGS:)
|
||||||
|
@ -18,3 +20,10 @@ bench: install
|
||||||
|
|
||||||
clean:
|
clean:
|
||||||
go clean $(GOFLAGS) -i ./...
|
go clean $(GOFLAGS) -i ./...
|
||||||
|
|
||||||
|
publish:
|
||||||
|
mkdir -p bin/$(VERSION)
|
||||||
|
cd bin/$(VERSION)
|
||||||
|
xgo -v -x --targets="windows/*,darwin/*,linux/386,linux/amd64,linux/arm-5,linux/arm64" -out bee_$(VERSION) github.com/beego/bee/v2
|
||||||
|
cd ..
|
||||||
|
ghr -u beego -r bee $(VERSION) $(VERSION)
|
||||||
|
|
411
README.md
411
README.md
|
@ -1,49 +1,67 @@
|
||||||
|
|
||||||
bee
|
bee
|
||||||
===
|
===
|
||||||
|
|
||||||
Bee is a command-line tool facilitating development of Beego-based application.
|
Bee is a command-line tool facilitating development of Beego-based application.
|
||||||
|
|
||||||
[![Build Status](https://drone.io/github.com/beego/bee/status.png)](https://drone.io/github.com/beego/bee/latest)
|
[![Build Status](https://img.shields.io/travis/beego/bee.svg?branch=master&label=master)](https://travis-ci.org/beego/bee)
|
||||||
|
[![Build Status](https://img.shields.io/travis/beego/bee.svg?branch=develop&label=develop)](https://travis-ci.org/beego/bee)
|
||||||
|
|
||||||
## Requirements
|
## Requirements
|
||||||
|
|
||||||
- Go version >= 1.3.
|
- Go version >= 1.13
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
To install `bee` use the `go get` command:
|
To install or update `bee` use the `go install` command:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
go get github.com/beego/bee
|
go install github.com/beego/bee/v2@latest
|
||||||
```
|
```
|
||||||
|
|
||||||
Then you can add `bee` binary to PATH environment variable in your `~/.bashrc` or `~/.bash_profile` file:
|
## Then you can add `bee` binary to PATH environment variable in your `~/.bashrc` or `~/.bash_profile` file:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
export PATH=$PATH:<your_main_gopath>/bin
|
export PATH=$PATH:<your_main_gopath>/bin
|
||||||
```
|
```
|
||||||
|
|
||||||
|
## Installing and updating bee prior Go version 1.17
|
||||||
|
|
||||||
|
To install `bee` use the `go get` command:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
go get github.com/beego/bee/v2
|
||||||
|
```
|
||||||
|
|
||||||
> If you already have `bee` installed, updating `bee` is simple:
|
> If you already have `bee` installed, updating `bee` is simple:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
go get -u github.com/beego/bee
|
go get -u github.com/beego/bee/v2
|
||||||
```
|
```
|
||||||
|
|
||||||
## Basic commands
|
## Basic commands
|
||||||
|
|
||||||
Bee provides a variety of commands which can be helpful at various stages of development. The top level commands include:
|
Bee provides a variety of commands which can be helpful at various stages of development. The top level commands include:
|
||||||
|
|
||||||
```
|
```
|
||||||
new Create a Beego application
|
|
||||||
run Run the app and start a Web server for development
|
|
||||||
pack Compress a beego project into a single file
|
|
||||||
api Create an API beego application
|
|
||||||
hprose Create an rpc application use hprose base on beego framework
|
|
||||||
bale Packs non-Go files to Go source files
|
|
||||||
version Prints the current Bee version
|
version Prints the current Bee version
|
||||||
|
migrate Runs database migrations
|
||||||
|
api Creates a Beego API application
|
||||||
|
bale Transforms non-Go files to Go source files
|
||||||
|
fix Fixes your application by making it compatible with newer versions of Beego
|
||||||
|
pro Source code generator
|
||||||
|
dlv Start a debugging session using Delve
|
||||||
|
dockerize Generates a Dockerfile and docker-compose.yaml for your Beego application
|
||||||
generate Source code generator
|
generate Source code generator
|
||||||
migrate Run database migrations
|
hprose Creates an RPC application based on Hprose and Beego frameworks
|
||||||
fix Fix the Beego application to make it compatible with Beego 1.6
|
new Creates a Beego application
|
||||||
|
pack Compresses a Beego application into a single file
|
||||||
|
rs Run customized scripts
|
||||||
|
run Run the application by starting a local development server
|
||||||
|
server serving static content over HTTP on port
|
||||||
|
update Update Bee
|
||||||
```
|
```
|
||||||
|
|
||||||
### bee version
|
### bee version
|
||||||
|
|
||||||
To display the current version of `bee`, `beego` and `go` installed on your machine:
|
To display the current version of `bee`, `beego` and `go` installed on your machine:
|
||||||
|
@ -55,18 +73,38 @@ ______
|
||||||
| |_/ / ___ ___
|
| |_/ / ___ ___
|
||||||
| ___ \ / _ \ / _ \
|
| ___ \ / _ \ / _ \
|
||||||
| |_/ /| __/| __/
|
| |_/ /| __/| __/
|
||||||
\____/ \___| \___| v1.5.0
|
\____/ \___| \___| v2.0.4
|
||||||
|
|
||||||
├── Beego : 1.7.0
|
├── Beego : 2.0.4
|
||||||
├── GoVersion : go1.6.2
|
├── GoVersion : go1.14.1
|
||||||
├── GOOS : windows
|
├── GOOS : darwin
|
||||||
├── GOARCH : amd64
|
├── GOARCH : amd64
|
||||||
├── NumCPU : 4
|
├── NumCPU : 4
|
||||||
├── GOPATH : C:\Users\beeuser\go
|
├── GOPATH : /home/beeuser/.go
|
||||||
├── GOROOT : C:\go
|
├── GOROOT : /usr/local/Cellar/go/1.14.1/libexec
|
||||||
├── Compiler : gc
|
├── Compiler : gc
|
||||||
└── Date : Monday, 22 Aug 2016
|
└── Published : 2020-09-13
|
||||||
```
|
```
|
||||||
|
|
||||||
|
You can also change the output format using `-o` flag:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
$ bee version -o json
|
||||||
|
{
|
||||||
|
"GoVersion": "go1.14.1",
|
||||||
|
"GOOS": "darwin",
|
||||||
|
"GOARCH": "amd64",
|
||||||
|
"NumCPU": 4,
|
||||||
|
"GOPATH": "/home/beeuser/.go",
|
||||||
|
"GOROOT": "/usr/local/Cellar/go/1.14.1/libexec",
|
||||||
|
"Compiler": "gc",
|
||||||
|
"BeeVersion": "2.0.4",
|
||||||
|
"BeegoVersion": "2.0.4",
|
||||||
|
"Published": "2020-09-13"
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
For more information on the usage, run `bee help version`.
|
||||||
|
|
||||||
### bee new
|
### bee new
|
||||||
|
|
||||||
|
@ -74,31 +112,27 @@ To create a new Beego web application:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
$ bee new my-web-app
|
$ bee new my-web-app
|
||||||
______
|
2020/09/14 22:28:51 INFO ▶ 0001 generate new project support go modules.
|
||||||
| ___ \
|
2020/09/14 22:28:51 INFO ▶ 0002 Creating application...
|
||||||
| |_/ / ___ ___
|
create /Users/beeuser/learn/my-web-app/go.mod
|
||||||
| ___ \ / _ \ / _ \
|
create /Users/beeuser/learn/my-web-app/
|
||||||
| |_/ /| __/| __/
|
create /Users/beeuser/learn/my-web-app/conf/
|
||||||
\____/ \___| \___| v1.5.0
|
create /Users/beeuser/learn/my-web-app/controllers/
|
||||||
2016/08/22 14:53:45 [INFO] Creating application...
|
create /Users/beeuser/learn/my-web-app/models/
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\
|
create /Users/beeuser/learn/my-web-app/routers/
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\conf\
|
create /Users/beeuser/learn/my-web-app/tests/
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\controllers\
|
create /Users/beeuser/learn/my-web-app/static/
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\models\
|
create /Users/beeuser/learn/my-web-app/static/js/
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\routers\
|
create /Users/beeuser/learn/my-web-app/static/css/
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\tests\
|
create /Users/beeuser/learn/my-web-app/static/img/
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\static\
|
create /Users/beeuser/learn/my-web-app/views/
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\static\js\
|
create /Users/beeuser/learn/my-web-app/conf/app.conf
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\static\css\
|
create /Users/beeuser/learn/my-web-app/controllers/default.go
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\static\img\
|
create /Users/beeuser/learn/my-web-app/views/index.tpl
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\views\
|
create /Users/beeuser/learn/my-web-app/routers/router.go
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\conf\app.conf
|
create /Users/beeuser/learn/my-web-app/tests/default_test.go
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\controllers\default.go
|
create /Users/beeuser/learn/my-web-app/main.go
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\views\index.tpl
|
2020/09/14 22:28:51 SUCCESS ▶ 0003 New application successfully created!
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\routers\router.go
|
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\tests\default_test.go
|
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app\main.go
|
|
||||||
2016/08/22 14:53:45 [SUCC] New application successfully created!
|
|
||||||
```
|
```
|
||||||
|
|
||||||
For more information on the usage, run `bee help new`.
|
For more information on the usage, run `bee help new`.
|
||||||
|
@ -111,12 +145,6 @@ To run the application we just created, you can navigate to the application fold
|
||||||
$ cd my-web-app && bee run
|
$ cd my-web-app && bee run
|
||||||
```
|
```
|
||||||
|
|
||||||
Or from anywhere in your machine:
|
|
||||||
|
|
||||||
```
|
|
||||||
$ bee run github.com/user/my-web-app
|
|
||||||
```
|
|
||||||
|
|
||||||
For more information on the usage, run `bee help run`.
|
For more information on the usage, run `bee help run`.
|
||||||
|
|
||||||
### bee pack
|
### bee pack
|
||||||
|
@ -130,18 +158,51 @@ ______
|
||||||
| |_/ / ___ ___
|
| |_/ / ___ ___
|
||||||
| ___ \ / _ \ / _ \
|
| ___ \ / _ \ / _ \
|
||||||
| |_/ /| __/| __/
|
| |_/ /| __/| __/
|
||||||
\____/ \___| \___| v1.5.0
|
\____/ \___| \___| v2.0.0
|
||||||
2016/08/22 15:11:01 Packaging application: C:\Users\beeuser\go\src\github.com\user\my-web-app
|
2016/12/26 22:29:29 INFO ▶ 0001 Packaging application on '/home/beeuser/.go/src/github.com/user/my-web-app'...
|
||||||
2016/08/22 15:11:01 Building application...
|
2016/12/26 22:29:29 INFO ▶ 0002 Building application...
|
||||||
2016/08/22 15:11:01 Env: GOOS=windows GOARCH=amd64
|
2016/12/26 22:29:29 INFO ▶ 0003 Using: GOOS=linux GOARCH=amd64
|
||||||
2016/08/22 15:11:08 Build successful
|
2016/12/26 22:29:31 SUCCESS ▶ 0004 Build Successful!
|
||||||
2016/08/22 15:11:08 Excluding relpath prefix: .
|
2016/12/26 22:29:31 INFO ▶ 0005 Writing to output: /home/beeuser/.go/src/github.com/user/my-web-app/my-web-app.tar.gz
|
||||||
2016/08/22 15:11:08 Excluding relpath suffix: .go:.DS_Store:.tmp
|
2016/12/26 22:29:31 INFO ▶ 0006 Excluding relpath prefix: .
|
||||||
2016/08/22 15:11:10 Writing to output: `C:\Users\beeuser\go\src\github.com\user\my-web-app\my-web-app.tar.gz`
|
2016/12/26 22:29:31 INFO ▶ 0007 Excluding relpath suffix: .go:.DS_Store:.tmp
|
||||||
|
2016/12/26 22:29:32 SUCCESS ▶ 0008 Application packed!
|
||||||
```
|
```
|
||||||
|
|
||||||
For more information on the usage, run `bee help pack`.
|
For more information on the usage, run `bee help pack`.
|
||||||
|
|
||||||
|
### bee rs
|
||||||
|
Inspired by makefile / npm scripts.
|
||||||
|
Run script allows you to run arbitrary commands using Bee.
|
||||||
|
Custom commands are provided from the "scripts" object inside bee.json or Beefile.
|
||||||
|
|
||||||
|
To run a custom command, use: $ bee rs mycmd ARGS
|
||||||
|
|
||||||
|
```bash
|
||||||
|
$ bee help rs
|
||||||
|
|
||||||
|
USAGE
|
||||||
|
bee rs
|
||||||
|
|
||||||
|
DESCRIPTION
|
||||||
|
Run script allows you to run arbitrary commands using Bee.
|
||||||
|
Custom commands are provided from the "scripts" object inside bee.json or Beefile.
|
||||||
|
|
||||||
|
To run a custom command, use: $ bee rs mycmd ARGS
|
||||||
|
|
||||||
|
AVAILABLE SCRIPTS
|
||||||
|
gtest
|
||||||
|
APP_ENV=test APP_CONF_PATH=$(pwd)/conf go test -v -cover
|
||||||
|
gtestall
|
||||||
|
APP_ENV=test APP_CONF_PATH=$(pwd)/conf go test -v -cover $(go list ./... | grep -v /vendor/)
|
||||||
|
|
||||||
|
```
|
||||||
|
|
||||||
|
*Run your scripts with:*
|
||||||
|
```$ bee rs gtest tests/*.go```
|
||||||
|
```$ bee rs gtestall```
|
||||||
|
|
||||||
|
|
||||||
### bee api
|
### bee api
|
||||||
|
|
||||||
To create a Beego API application:
|
To create a Beego API application:
|
||||||
|
@ -153,23 +214,25 @@ ______
|
||||||
| |_/ / ___ ___
|
| |_/ / ___ ___
|
||||||
| ___ \ / _ \ / _ \
|
| ___ \ / _ \ / _ \
|
||||||
| |_/ /| __/| __/
|
| |_/ /| __/| __/
|
||||||
\____/ \___| \___| v1.5.0
|
\____/ \___| \___| v2.0.0
|
||||||
2016/08/22 15:14:10 [INFO] Creating API...
|
2020/09/14 22:35:11 INFO ▶ 0001 generate api project support go modules.
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api
|
2020/09/14 22:35:11 INFO ▶ 0002 Creating API...
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\conf
|
create /Users/beeuser/code/learn/my-api/go.mod
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\controllers
|
create /Users/beeuser/code/learn/my-api
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\tests
|
create /Users/beeuser/code/learn/my-api/conf
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\conf\app.conf
|
create /Users/beeuser/code/learn/my-api/controllers
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\models
|
create /Users/beeuser/code/learn/my-api/tests
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\routers\
|
create /Users/beeuser/code/learn/my-api/conf/app.conf
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\controllers\object.go
|
create /Users/beeuser/code/learn/my-api/models
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\controllers\user.go
|
create /Users/beeuser/code/learn/my-api/routers/
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\tests\default_test.go
|
create /Users/beeuser/code/learn/my-api/controllers/object.go
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\routers\router.go
|
create /Users/beeuser/code/learn/my-api/controllers/user.go
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\models\object.go
|
create /Users/beeuser/code/learn/my-api/tests/default_test.go
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\models\user.go
|
create /Users/beeuser/code/learn/my-api/routers/router.go
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-api\main.go
|
create /Users/beeuser/code/learn/my-api/models/object.go
|
||||||
2016/08/22 15:14:10 [SUCC] New API successfully created!
|
create /Users/beeuser/code/learn/my-api/models/user.go
|
||||||
|
create /Users/beeuser/code/learn/my-api/main.go
|
||||||
|
2020/09/14 22:35:11 SUCCESS ▶ 0003 New API successfully created!
|
||||||
```
|
```
|
||||||
|
|
||||||
For more information on the usage, run `bee help api`.
|
For more information on the usage, run `bee help api`.
|
||||||
|
@ -185,16 +248,18 @@ ______
|
||||||
| |_/ / ___ ___
|
| |_/ / ___ ___
|
||||||
| ___ \ / _ \ / _ \
|
| ___ \ / _ \ / _ \
|
||||||
| |_/ /| __/| __/
|
| |_/ /| __/| __/
|
||||||
\____/ \___| \___| v1.5.0
|
\____/ \___| \___| v2.0.0
|
||||||
2016/08/22 16:09:13 [INFO] Creating Hprose application...
|
2020/09/14 22:36:39 INFO ▶ 0001 generate api project support go modules.
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-rpc-app
|
2020/09/14 22:36:39 INFO ▶ 0002 Creating Hprose application...
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-rpc-app\conf
|
create /Users/beeuser/code/learn/my-rpc-app/go.mod
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-rpc-app\conf\app.conf
|
create /Users/beeuser/code/learn/my-rpc-app
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-rpc-app\models
|
create /Users/beeuser/code/learn/my-rpc-app/conf
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-rpc-app\models\object.go
|
create /Users/beeuser/code/learn/my-rpc-app/conf/app.conf
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-rpc-app\models\user.go
|
create /Users/beeuser/code/learn/my-rpc-app/models
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-rpc-app\main.go
|
create /Users/beeuser/code/learn/my-rpc-app/models/object.go
|
||||||
2016/08/22 16:09:13 [SUCC] New Hprose application successfully created!
|
create /Users/beeuser/code/learn/my-rpc-app/models/user.go
|
||||||
|
create /Users/beeuser/code/learn/my-rpc-app/main.go
|
||||||
|
2020/09/14 22:36:39 SUCCESS ▶ 0003 New Hprose application successfully created!
|
||||||
```
|
```
|
||||||
|
|
||||||
For more information on the usage, run `bee help hprose`.
|
For more information on the usage, run `bee help hprose`.
|
||||||
|
@ -210,11 +275,8 @@ ______
|
||||||
| |_/ / ___ ___
|
| |_/ / ___ ___
|
||||||
| ___ \ / _ \ / _ \
|
| ___ \ / _ \ / _ \
|
||||||
| |_/ /| __/| __/
|
| |_/ /| __/| __/
|
||||||
\____/ \___| \___| v1.5.0
|
\____/ \___| \___| v2.0.0
|
||||||
2016/08/22 16:37:24 [INFO] Detected bee.json
|
2020/09/14 22:37:56 SUCCESS ▶ 0001 Baled resources successfully!
|
||||||
2016/08/22 16:37:24 [INFO] Packaging directory(static/js)
|
|
||||||
2016/08/22 16:37:24 [INFO] Packaging directory(static/css)
|
|
||||||
2016/08/22 16:37:24 [SUCC] Baled resources successfully!
|
|
||||||
```
|
```
|
||||||
|
|
||||||
For more information on the usage, run `bee help bale`.
|
For more information on the usage, run `bee help bale`.
|
||||||
|
@ -238,15 +300,129 @@ ______
|
||||||
| |_/ / ___ ___
|
| |_/ / ___ ___
|
||||||
| ___ \ / _ \ / _ \
|
| ___ \ / _ \ / _ \
|
||||||
| |_/ /| __/| __/
|
| |_/ /| __/| __/
|
||||||
\____/ \___| \___| v1.5.0
|
\____/ \___| \___| v2.0.0
|
||||||
2016/08/22 16:55:30 [INFO] Using 'Hello' as controller name
|
2020/09/14 22:38:44 INFO ▶ 0001 Using 'Hello' as controller name
|
||||||
2016/08/22 16:55:30 [INFO] Using 'controllers' as package name
|
2020/09/14 22:38:44 INFO ▶ 0002 Using 'controllers' as package name
|
||||||
create C:\Users\beeuser\go\src\github.com\user\my-web-app/controllers/hello.go
|
create /Users/beeuser/code/learn/my-api/controllers/hello.go
|
||||||
2016/08/22 16:55:30 [SUCC] Controller successfully generated!
|
2020/09/14 22:38:44 SUCCESS ▶ 0003 Controller successfully generated!
|
||||||
```
|
```
|
||||||
|
|
||||||
For more information on the usage, run `bee help generate`.
|
For more information on the usage, run `bee help generate`.
|
||||||
|
|
||||||
|
### bee dockerize
|
||||||
|
|
||||||
|
Bee also helps you dockerize your Beego application by generating a Dockerfile and a docker-compose.yaml file.
|
||||||
|
|
||||||
|
For example, to generate a Dockerfile with `golang:1.20.1` baseimage and exposing port `9000`:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
$ bee dockerize -baseimage=golang:1.20.1 -expose=9000
|
||||||
|
______
|
||||||
|
| ___ \
|
||||||
|
| |_/ / ___ ___
|
||||||
|
| ___ \ / _ \ / _ \
|
||||||
|
| |_/ /| __/| __/
|
||||||
|
\____/ \___| \___| v2.0.4
|
||||||
|
2023/05/02 21:03:05 INFO ▶ 0001 Generating Dockerfile and docker-compose.yaml...
|
||||||
|
2023/05/02 21:03:05 SUCCESS ▶ 0002 Dockerfile generated.
|
||||||
|
2023/05/02 21:03:05 SUCCESS ▶ 0003 docker-compose.yaml generated.
|
||||||
|
```
|
||||||
|
|
||||||
|
For more information on the usage, run `bee help dockerize`.
|
||||||
|
|
||||||
|
### bee dlv
|
||||||
|
|
||||||
|
Bee can also help with debugging your application. To start a debugging session:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
______
|
||||||
|
| ___ \
|
||||||
|
| |_/ / ___ ___
|
||||||
|
| ___ \ / _ \ / _ \
|
||||||
|
| |_/ /| __/| __/
|
||||||
|
\____/ \___| \___| v2.0.0
|
||||||
|
2020/09/14 22:40:12 INFO ▶ 0001 Starting Delve Debugger...
|
||||||
|
Type 'help' for list of commands.
|
||||||
|
(dlv) break main.main
|
||||||
|
Breakpoint 1 set at 0x40100f for main.main() ./main.go:8
|
||||||
|
|
||||||
|
(dlv) continue
|
||||||
|
> main.main() ./main.go:8 (hits goroutine(1):1 total:1) (PC: 0x40100f)
|
||||||
|
3: import (
|
||||||
|
4: _ "github.com/user/myapp/routers"
|
||||||
|
5: beego "github.com/beego/beego/v2/server/web"
|
||||||
|
6: )
|
||||||
|
7:
|
||||||
|
=> 8: func main() {
|
||||||
|
9: beego.Run()
|
||||||
|
10: }
|
||||||
|
11:
|
||||||
|
```
|
||||||
|
|
||||||
|
For more information on the usage, run `bee help dlv`.
|
||||||
|
|
||||||
|
### bee pro
|
||||||
|
|
||||||
|
#### bee pro toml
|
||||||
|
|
||||||
|
To create a beegopro.toml file
|
||||||
|
|
||||||
|
```bash
|
||||||
|
$ bee pro toml
|
||||||
|
2020/09/14 22:51:18 SUCCESS ▶ 0001 Successfully created file beegopro.toml
|
||||||
|
2020/09/14 22:51:18 SUCCESS ▶ 0002 Toml successfully generated!
|
||||||
|
```
|
||||||
|
|
||||||
|
#### bee pro gen
|
||||||
|
|
||||||
|
Source code generator by beegopro.toml
|
||||||
|
|
||||||
|
```bash
|
||||||
|
$ bee pro gen
|
||||||
|
2020/09/14 23:01:13 INFO ▶ 0001 Create /Users/beeuser/.beego/beego-pro Success!
|
||||||
|
2020/09/14 23:01:13 INFO ▶ 0002 git pull /Users/beeuser/.beego/beego-pro
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0003 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0004 Using 'example' as package name from controllers
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0005 create file '/Users/beeuser/code/learn/my-web-app/controllers/bee_default_controller.go' from controllers
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0006 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0007 Using 'example' as package name from controllers
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0008 create file '/Users/beeuser/code/learn/my-web-app/controllers/example.go' from controllers
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0009 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0010 Using 'example' as package name from models
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0011 create file '/Users/beeuser/code/learn/my-web-app/models/bee_default_model.go' from models
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0012 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0013 Using 'example' as package name from models
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0014 create file '/Users/beeuser/code/learn/my-web-app/models/example.go' from models
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0015 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0016 Using 'example' as package name from routers
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0017 create file '/Users/beeuser/code/learn/my-web-app/routers/example.go' from routers
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0018 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0019 Using 'example' as package name from example
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0020 create file '/Users/beeuser/code/learn/my-web-app/ant/src/pages/example/list.tsx' from example
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0021 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0022 Using 'example' as package name from example
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0023 create file '/Users/beeuser/code/learn/my-web-app/ant/src/pages/example/formconfig.tsx' from example
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0024 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0025 Using 'example' as package name from example
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0026 create file '/Users/beeuser/code/learn/my-web-app/ant/src/pages/example/create.tsx' from example
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0027 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0028 Using 'example' as package name from example
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0029 create file '/Users/beeuser/code/learn/my-web-app/ant/src/pages/example/update.tsx' from example
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0030 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0031 Using 'example' as package name from example
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0032 create file '/Users/beeuser/code/learn/my-web-app/ant/src/pages/example/info.tsx' from example
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0033 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0034 Using 'example' as package name from sql
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0035 create file '/Users/beeuser/code/learn/my-web-app/sql/example_up.sql' from sql
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0036 2020/09/14 23:01:15 INFO ▶ 0001 db exec info ./sql/example_up.sql
|
||||||
|
2020/09/14 23:01:15 SUCCESS ▶ 0002 Migration successfully generated!
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0037 Using 'example' as name
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0038 Using 'example' as package name from sql
|
||||||
|
2020/09/14 23:01:15 INFO ▶ 0039 create file '/Users/beeuser/code/learn/my-web-app/sql/example_down.sql' from sql
|
||||||
|
2020/09/14 23:01:15 SUCCESS ▶ 0040 Gen successfully generated!
|
||||||
|
```
|
||||||
|
|
||||||
|
####
|
||||||
## Shortcuts
|
## Shortcuts
|
||||||
|
|
||||||
Because you'll likely type these generator commands over and over, it makes sense to create aliases:
|
Because you'll likely type these generator commands over and over, it makes sense to create aliases:
|
||||||
|
@ -270,10 +446,33 @@ For instance, to get more information about the `run` command:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
$ bee help run
|
$ bee help run
|
||||||
usage: bee run [appname] [watchall] [-main=*.go] [-downdoc=true] [-gendoc=true] [-vendor=true] [-e=folderToExclude] [-tags=goBuildTags]
|
USAGE
|
||||||
|
bee run [appname] [watchall] [-main=*.go] [-downdoc=true] [-gendoc=true] [-vendor=true] [-e=folderToExclude] [-tags=goBuildTags] [-runmode=BEEGO_RUNMODE]
|
||||||
|
|
||||||
Run command will supervise the file system of the beego project using inotify,
|
OPTIONS
|
||||||
it will recompile and restart the app after any modifications.
|
-downdoc
|
||||||
|
Enable auto-download of the swagger file if it does not exist.
|
||||||
|
|
||||||
|
-e=[]
|
||||||
|
List of paths to exclude.
|
||||||
|
|
||||||
|
-gendoc
|
||||||
|
Enable auto-generate the docs.
|
||||||
|
|
||||||
|
-main=[]
|
||||||
|
Specify main go files.
|
||||||
|
|
||||||
|
-runmode
|
||||||
|
Set the Beego run mode.
|
||||||
|
|
||||||
|
-tags
|
||||||
|
Set the build tags. See: https://golang.org/pkg/go/build/
|
||||||
|
|
||||||
|
-vendor=false
|
||||||
|
Enable watch vendor folder.
|
||||||
|
|
||||||
|
DESCRIPTION
|
||||||
|
Run command will supervise the filesystem of the application for any changes, and recompile/restart it.
|
||||||
```
|
```
|
||||||
|
|
||||||
## Contributing
|
## Contributing
|
||||||
|
@ -333,14 +532,14 @@ A feature should be made of commits splitted by **logical chunks** (no half-done
|
||||||
how many commits your changes require.
|
how many commits your changes require.
|
||||||
|
|
||||||
- Write insightful and descriptive commit messages. It lets us and future contributors quickly understand your changes
|
- Write insightful and descriptive commit messages. It lets us and future contributors quickly understand your changes
|
||||||
without having to read your changes. Please provide a summary in the first line (50-72 characters) and eventually,
|
without having to read your changes. Please provide a summary in the first line (50-72 characters) and eventually,
|
||||||
go to greater lengths in your message's body. A good example can be found in [Angular commit message format](https://github.com/angular/angular.js/blob/master/CONTRIBUTING.md#commit-message-format).
|
go to greater lengths in your message's body. A good example can be found in [Angular commit message format](https://github.com/angular/angular.js/blob/master/CONTRIBUTING.md#commit-message-format).
|
||||||
|
|
||||||
- Please **include the appropriate test cases** for your patch.
|
- Please **include the appropriate test cases** for your patch.
|
||||||
|
|
||||||
- Make sure all tests pass before submitting your changes.
|
- Make sure all tests pass before submitting your changes.
|
||||||
|
|
||||||
- Rebase your commits. It may be that new commits have been introduced on `develop`.
|
- Rebase your commits. It may be that new commits have been introduced on `develop`.
|
||||||
Rebasing will update your branch with the most recent code and make your changes easier to review:
|
Rebasing will update your branch with the most recent code and make your changes easier to review:
|
||||||
|
|
||||||
```
|
```
|
||||||
|
@ -378,7 +577,7 @@ Rebasing will update your branch with the most recent code and make your changes
|
||||||
## Licence
|
## Licence
|
||||||
|
|
||||||
```text
|
```text
|
||||||
Copyright 2016 bee authors
|
Copyright 2020 bee authors
|
||||||
|
|
||||||
Licensed under the Apache License, Version 2.0 (the "License");
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
you may not use this file except in compliance with the License.
|
you may not use this file except in compliance with the License.
|
||||||
|
@ -391,4 +590,4 @@ distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
See the License for the specific language governing permissions and
|
See the License for the specific language governing permissions and
|
||||||
limitations under the License.
|
limitations under the License.
|
||||||
```
|
```
|
||||||
|
|
280
autorouter.go
280
autorouter.go
|
@ -1,280 +0,0 @@
|
||||||
// Copyright 2013 bee authors
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"go/ast"
|
|
||||||
gobuild "go/build"
|
|
||||||
"go/doc"
|
|
||||||
"go/parser"
|
|
||||||
"go/token"
|
|
||||||
"io"
|
|
||||||
"io/ioutil"
|
|
||||||
"os"
|
|
||||||
"path"
|
|
||||||
"runtime"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
var cmdRouter = &Command{
|
|
||||||
UsageLine: "router",
|
|
||||||
Short: "auto-generate routers for the app controllers",
|
|
||||||
Long: `
|
|
||||||
|
|
||||||
`,
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
cmdRouter.Run = autoRouter
|
|
||||||
}
|
|
||||||
|
|
||||||
func autoRouter(cmd *Command, args []string) int {
|
|
||||||
fmt.Println("[INFO] Starting auto-generating routers...")
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
// getControllerInfo returns controllers that embeded "beego.controller"
|
|
||||||
// and their methods of package in given path.
|
|
||||||
func getControllerInfo(path string) (map[string][]string, error) {
|
|
||||||
now := time.Now()
|
|
||||||
path = strings.TrimSuffix(path, "/")
|
|
||||||
dir, err := os.Open(path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
fis, err := dir.Readdir(0)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
files := make([]*source, 0, len(fis))
|
|
||||||
for _, fi := range fis {
|
|
||||||
// Only load Go files
|
|
||||||
if strings.HasSuffix(fi.Name(), ".go") {
|
|
||||||
f, err := os.Open(path + "/" + fi.Name())
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
p := make([]byte, fi.Size())
|
|
||||||
_, err = f.Read(p)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
f.Close()
|
|
||||||
files = append(files, &source{
|
|
||||||
name: path + "/" + fi.Name(),
|
|
||||||
data: p,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
rw := &routerWalker{
|
|
||||||
pdoc: &Package{
|
|
||||||
ImportPath: path,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
cm := make(map[string][]string)
|
|
||||||
pdoc, err := rw.build(files)
|
|
||||||
for _, t := range pdoc.Types {
|
|
||||||
// Check if embeded "beego.Controller".
|
|
||||||
if strings.Index(t.Decl, "beego.Controller") > -1 {
|
|
||||||
for _, f := range t.Methods {
|
|
||||||
cm[t.Name] = append(cm[t.Name], f.Name)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
fmt.Println(time.Since(now))
|
|
||||||
return cm, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// source represents a source code file.
|
|
||||||
type source struct {
|
|
||||||
name string
|
|
||||||
data []byte
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *source) Name() string { return s.name }
|
|
||||||
func (s *source) Size() int64 { return int64(len(s.data)) }
|
|
||||||
func (s *source) Mode() os.FileMode { return 0 }
|
|
||||||
func (s *source) ModTime() time.Time { return time.Time{} }
|
|
||||||
func (s *source) IsDir() bool { return false }
|
|
||||||
func (s *source) Sys() interface{} { return nil }
|
|
||||||
|
|
||||||
// routerWalker holds the state used when building the documentation.
|
|
||||||
type routerWalker struct {
|
|
||||||
pdoc *Package
|
|
||||||
srcs map[string]*source // Source files
|
|
||||||
fset *token.FileSet
|
|
||||||
buf []byte // scratch space for printNode method
|
|
||||||
}
|
|
||||||
|
|
||||||
// Package represents full information and documentation for a package.
|
|
||||||
type Package struct {
|
|
||||||
ImportPath string
|
|
||||||
Types []*Type // Top-level declarations
|
|
||||||
}
|
|
||||||
|
|
||||||
// Type represents structs and interfaces.
|
|
||||||
type Type struct {
|
|
||||||
Name string // Type name
|
|
||||||
Decl string
|
|
||||||
Methods []*Func // Exported methods
|
|
||||||
}
|
|
||||||
|
|
||||||
// Func represents functions
|
|
||||||
type Func struct {
|
|
||||||
Name string
|
|
||||||
}
|
|
||||||
|
|
||||||
// build generates data from source files.
|
|
||||||
func (w *routerWalker) build(srcs []*source) (*Package, error) {
|
|
||||||
// Add source files to walker, I skipped references here
|
|
||||||
w.srcs = make(map[string]*source)
|
|
||||||
for _, src := range srcs {
|
|
||||||
w.srcs[src.name] = src
|
|
||||||
}
|
|
||||||
|
|
||||||
w.fset = token.NewFileSet()
|
|
||||||
|
|
||||||
// Find the package and associated files
|
|
||||||
ctxt := gobuild.Context{
|
|
||||||
GOOS: runtime.GOOS,
|
|
||||||
GOARCH: runtime.GOARCH,
|
|
||||||
CgoEnabled: true,
|
|
||||||
JoinPath: path.Join,
|
|
||||||
IsAbsPath: path.IsAbs,
|
|
||||||
SplitPathList: func(list string) []string { return strings.Split(list, ":") },
|
|
||||||
IsDir: func(path string) bool { panic("unexpected") },
|
|
||||||
HasSubdir: func(root, dir string) (rel string, ok bool) { panic("unexpected") },
|
|
||||||
ReadDir: func(dir string) (fi []os.FileInfo, err error) { return w.readDir(dir) },
|
|
||||||
OpenFile: func(path string) (r io.ReadCloser, err error) { return w.openFile(path) },
|
|
||||||
Compiler: "gc",
|
|
||||||
}
|
|
||||||
|
|
||||||
bpkg, err := ctxt.ImportDir(w.pdoc.ImportPath, 0)
|
|
||||||
// Continue if there are no Go source files; we still want the directory info
|
|
||||||
_, nogo := err.(*gobuild.NoGoError)
|
|
||||||
if err != nil {
|
|
||||||
if nogo {
|
|
||||||
err = nil
|
|
||||||
} else {
|
|
||||||
return nil, errors.New("routerWalker.build -> " + err.Error())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Parse the Go files
|
|
||||||
files := make(map[string]*ast.File)
|
|
||||||
for _, name := range append(bpkg.GoFiles, bpkg.CgoFiles...) {
|
|
||||||
file, err := parser.ParseFile(w.fset, name, w.srcs[name].data, parser.ParseComments)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.New("routerWalker.build -> parse go files: " + err.Error())
|
|
||||||
}
|
|
||||||
files[name] = file
|
|
||||||
}
|
|
||||||
|
|
||||||
apkg, _ := ast.NewPackage(w.fset, files, simpleImporter, nil)
|
|
||||||
|
|
||||||
mode := doc.Mode(0)
|
|
||||||
if w.pdoc.ImportPath == "builtin" {
|
|
||||||
mode |= doc.AllDecls
|
|
||||||
}
|
|
||||||
|
|
||||||
pdoc := doc.New(apkg, w.pdoc.ImportPath, mode)
|
|
||||||
|
|
||||||
w.pdoc.Types = w.types(pdoc.Types)
|
|
||||||
|
|
||||||
return w.pdoc, err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *routerWalker) funcs(fdocs []*doc.Func) []*Func {
|
|
||||||
var result []*Func
|
|
||||||
for _, d := range fdocs {
|
|
||||||
result = append(result, &Func{
|
|
||||||
Name: d.Name,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
return result
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *routerWalker) types(tdocs []*doc.Type) []*Type {
|
|
||||||
var result []*Type
|
|
||||||
for _, d := range tdocs {
|
|
||||||
result = append(result, &Type{
|
|
||||||
Decl: w.printDecl(d.Decl),
|
|
||||||
Name: d.Name,
|
|
||||||
Methods: w.funcs(d.Methods),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
return result
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *routerWalker) printDecl(decl ast.Node) string {
|
|
||||||
var d Code
|
|
||||||
d, w.buf = printDecl(decl, w.fset, w.buf)
|
|
||||||
return d.Text
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *routerWalker) readDir(dir string) ([]os.FileInfo, error) {
|
|
||||||
if dir != w.pdoc.ImportPath {
|
|
||||||
panic("unexpected")
|
|
||||||
}
|
|
||||||
fis := make([]os.FileInfo, 0, len(w.srcs))
|
|
||||||
for _, src := range w.srcs {
|
|
||||||
fis = append(fis, src)
|
|
||||||
}
|
|
||||||
return fis, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *routerWalker) openFile(path string) (io.ReadCloser, error) {
|
|
||||||
if strings.HasPrefix(path, w.pdoc.ImportPath+"/") {
|
|
||||||
if src, ok := w.srcs[path[len(w.pdoc.ImportPath)+1:]]; ok {
|
|
||||||
return ioutil.NopCloser(bytes.NewReader(src.data)), nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
panic("unexpected")
|
|
||||||
}
|
|
||||||
|
|
||||||
func simpleImporter(imports map[string]*ast.Object, path string) (*ast.Object, error) {
|
|
||||||
pkg := imports[path]
|
|
||||||
if pkg == nil {
|
|
||||||
// Guess the package name without importing it. Start with the last
|
|
||||||
// element of the path.
|
|
||||||
name := path[strings.LastIndex(path, "/")+1:]
|
|
||||||
|
|
||||||
// Trim commonly used prefixes and suffixes containing illegal name
|
|
||||||
// runes.
|
|
||||||
name = strings.TrimSuffix(name, ".go")
|
|
||||||
name = strings.TrimSuffix(name, "-go")
|
|
||||||
name = strings.TrimPrefix(name, "go.")
|
|
||||||
name = strings.TrimPrefix(name, "go-")
|
|
||||||
name = strings.TrimPrefix(name, "biogo.")
|
|
||||||
|
|
||||||
// It's also common for the last element of the path to contain an
|
|
||||||
// extra "go" prefix, but not always.
|
|
||||||
// TODO: examine unresolved ids to detect when trimming the "go" prefix is appropriate.
|
|
||||||
pkg = ast.NewObj(ast.Pkg, name)
|
|
||||||
pkg.Data = ast.NewScope(nil)
|
|
||||||
imports[path] = pkg
|
|
||||||
}
|
|
||||||
return pkg, nil
|
|
||||||
}
|
|
|
@ -1,9 +0,0 @@
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestGetControllerInfo(t *testing.T) {
|
|
||||||
getControllerInfo("testdata/router/")
|
|
||||||
}
|
|
60
banner.go
60
banner.go
|
@ -1,60 +0,0 @@
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io"
|
|
||||||
"io/ioutil"
|
|
||||||
"os"
|
|
||||||
"runtime"
|
|
||||||
"text/template"
|
|
||||||
)
|
|
||||||
|
|
||||||
type vars struct {
|
|
||||||
GoVersion string
|
|
||||||
GOOS string
|
|
||||||
GOARCH string
|
|
||||||
NumCPU int
|
|
||||||
GOPATH string
|
|
||||||
GOROOT string
|
|
||||||
Compiler string
|
|
||||||
BeeVersion string
|
|
||||||
BeegoVersion string
|
|
||||||
}
|
|
||||||
|
|
||||||
// InitBanner loads the banner and prints it to output
|
|
||||||
// All errors are ignored, the application will not
|
|
||||||
// print the banner in case of error.
|
|
||||||
func InitBanner(out io.Writer, in io.Reader) {
|
|
||||||
if in == nil {
|
|
||||||
logger.Fatal("The input is nil")
|
|
||||||
}
|
|
||||||
|
|
||||||
banner, err := ioutil.ReadAll(in)
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("Error while trying to read the banner: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
show(out, string(banner))
|
|
||||||
}
|
|
||||||
|
|
||||||
func show(out io.Writer, content string) {
|
|
||||||
t, err := template.New("banner").
|
|
||||||
Funcs(template.FuncMap{"Now": Now}).
|
|
||||||
Parse(content)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("Cannot parse the banner template: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = t.Execute(out, vars{
|
|
||||||
getGoVersion(),
|
|
||||||
runtime.GOOS,
|
|
||||||
runtime.GOARCH,
|
|
||||||
runtime.NumCPU(),
|
|
||||||
os.Getenv("GOPATH"),
|
|
||||||
runtime.GOROOT(),
|
|
||||||
runtime.Compiler,
|
|
||||||
version,
|
|
||||||
getBeegoVersion(),
|
|
||||||
})
|
|
||||||
MustCheck(err)
|
|
||||||
}
|
|
260
bee.go
260
bee.go
|
@ -1,260 +0,0 @@
|
||||||
// Copyright 2013 bee authors
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
|
|
||||||
// Bee is a tool for developing applications based on beego framework.
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"log"
|
|
||||||
"os"
|
|
||||||
"strings"
|
|
||||||
"text/template"
|
|
||||||
)
|
|
||||||
|
|
||||||
const version = "1.6.2"
|
|
||||||
|
|
||||||
// Command is the unit of execution
|
|
||||||
type Command struct {
|
|
||||||
// Run runs the command.
|
|
||||||
// The args are the arguments after the command name.
|
|
||||||
Run func(cmd *Command, args []string) int
|
|
||||||
|
|
||||||
// PreRun performs an operation before running the command
|
|
||||||
PreRun func(cmd *Command, args []string)
|
|
||||||
|
|
||||||
// UsageLine is the one-line usage message.
|
|
||||||
// The first word in the line is taken to be the command name.
|
|
||||||
UsageLine string
|
|
||||||
|
|
||||||
// Short is the short description shown in the 'go help' output.
|
|
||||||
Short string
|
|
||||||
|
|
||||||
// Long is the long message shown in the 'go help <this-command>' output.
|
|
||||||
Long string
|
|
||||||
|
|
||||||
// Flag is a set of flags specific to this command.
|
|
||||||
Flag flag.FlagSet
|
|
||||||
|
|
||||||
// CustomFlags indicates that the command will do its own
|
|
||||||
// flag parsing.
|
|
||||||
CustomFlags bool
|
|
||||||
|
|
||||||
// output out writer if set in SetOutput(w)
|
|
||||||
output *io.Writer
|
|
||||||
}
|
|
||||||
|
|
||||||
// Name returns the command's name: the first word in the usage line.
|
|
||||||
func (c *Command) Name() string {
|
|
||||||
name := c.UsageLine
|
|
||||||
i := strings.Index(name, " ")
|
|
||||||
if i >= 0 {
|
|
||||||
name = name[:i]
|
|
||||||
}
|
|
||||||
return name
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetOutput sets the destination for usage and error messages.
|
|
||||||
// If output is nil, os.Stderr is used.
|
|
||||||
func (c *Command) SetOutput(output io.Writer) {
|
|
||||||
c.output = &output
|
|
||||||
}
|
|
||||||
|
|
||||||
// Out returns the out writer of the current command.
|
|
||||||
// If cmd.output is nil, os.Stderr is used.
|
|
||||||
func (c *Command) Out() io.Writer {
|
|
||||||
if c.output != nil {
|
|
||||||
return *c.output
|
|
||||||
}
|
|
||||||
return NewColorWriter(os.Stderr)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Usage puts out the usage for the command.
|
|
||||||
func (c *Command) Usage() {
|
|
||||||
tmpl(cmdUsage, c)
|
|
||||||
os.Exit(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Runnable reports whether the command can be run; otherwise
|
|
||||||
// it is a documentation pseudo-command such as import path.
|
|
||||||
func (c *Command) Runnable() bool {
|
|
||||||
return c.Run != nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Command) Options() map[string]string {
|
|
||||||
options := make(map[string]string)
|
|
||||||
c.Flag.VisitAll(func(f *flag.Flag) {
|
|
||||||
defaultVal := f.DefValue
|
|
||||||
if len(defaultVal) > 0 {
|
|
||||||
if strings.Contains(defaultVal, ":") {
|
|
||||||
// Truncate the flag's default value by appending '...' at the end
|
|
||||||
options[f.Name+"="+strings.Split(defaultVal, ":")[0]+":..."] = f.Usage
|
|
||||||
} else {
|
|
||||||
options[f.Name+"="+defaultVal] = f.Usage
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
options[f.Name] = f.Usage
|
|
||||||
}
|
|
||||||
})
|
|
||||||
return options
|
|
||||||
}
|
|
||||||
|
|
||||||
var availableCommands = []*Command{
|
|
||||||
cmdNew,
|
|
||||||
cmdRun,
|
|
||||||
cmdPack,
|
|
||||||
cmdApiapp,
|
|
||||||
cmdHproseapp,
|
|
||||||
//cmdRouter,
|
|
||||||
//cmdTest,
|
|
||||||
cmdBale,
|
|
||||||
cmdVersion,
|
|
||||||
cmdGenerate,
|
|
||||||
//cmdRundocs,
|
|
||||||
cmdMigrate,
|
|
||||||
cmdFix,
|
|
||||||
}
|
|
||||||
|
|
||||||
var logger = GetBeeLogger(os.Stdout)
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
currentpath, _ := os.Getwd()
|
|
||||||
|
|
||||||
flag.Usage = usage
|
|
||||||
flag.Parse()
|
|
||||||
log.SetFlags(0)
|
|
||||||
|
|
||||||
args := flag.Args()
|
|
||||||
|
|
||||||
if len(args) < 1 {
|
|
||||||
usage()
|
|
||||||
}
|
|
||||||
|
|
||||||
if args[0] == "help" {
|
|
||||||
help(args[1:])
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, cmd := range availableCommands {
|
|
||||||
if cmd.Name() == args[0] && cmd.Run != nil {
|
|
||||||
cmd.Flag.Usage = func() { cmd.Usage() }
|
|
||||||
if cmd.CustomFlags {
|
|
||||||
args = args[1:]
|
|
||||||
} else {
|
|
||||||
cmd.Flag.Parse(args[1:])
|
|
||||||
args = cmd.Flag.Args()
|
|
||||||
}
|
|
||||||
|
|
||||||
if cmd.PreRun != nil {
|
|
||||||
cmd.PreRun(cmd, args)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if current directory is inside the GOPATH,
|
|
||||||
// if so parse the packages inside it.
|
|
||||||
if strings.Contains(currentpath, GetGOPATHs()[0]+"/src") && isGenerateDocs(cmd.Name(), args) {
|
|
||||||
parsePackagesFromDir(currentpath)
|
|
||||||
}
|
|
||||||
|
|
||||||
os.Exit(cmd.Run(cmd, args))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
printErrorAndExit("Unknown subcommand")
|
|
||||||
}
|
|
||||||
|
|
||||||
func isGenerateDocs(name string, args []string) bool {
|
|
||||||
if name != "generate" {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
for _, a := range args {
|
|
||||||
if a == "docs" {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
var usageTemplate = `Bee is a Fast and Flexible tool for managing your Beego Web Application.
|
|
||||||
|
|
||||||
{{"USAGE" | headline}}
|
|
||||||
{{"bee command [arguments]" | bold}}
|
|
||||||
|
|
||||||
{{"AVAILABLE COMMANDS" | headline}}
|
|
||||||
{{range .}}{{if .Runnable}}
|
|
||||||
{{.Name | printf "%-11s" | bold}} {{.Short}}{{end}}{{end}}
|
|
||||||
|
|
||||||
Use {{"bee help [command]" | bold}} for more information about a command.
|
|
||||||
|
|
||||||
{{"ADDITIONAL HELP TOPICS" | headline}}
|
|
||||||
{{range .}}{{if not .Runnable}}
|
|
||||||
{{.Name | printf "%-11s"}} {{.Short}}{{end}}{{end}}
|
|
||||||
|
|
||||||
Use {{"bee help [topic]" | bold}} for more information about that topic.
|
|
||||||
`
|
|
||||||
|
|
||||||
var helpTemplate = `{{"USAGE" | headline}}
|
|
||||||
{{.UsageLine | printf "bee %s" | bold}}
|
|
||||||
{{if .Options}}{{endline}}{{"OPTIONS" | headline}}{{range $k,$v := .Options}}
|
|
||||||
{{$k | printf "-%-12s" | bold}} {{$v}}{{end}}{{endline}}{{end}}
|
|
||||||
{{"DESCRIPTION" | headline}}
|
|
||||||
{{tmpltostr .Long . | trim}}
|
|
||||||
`
|
|
||||||
|
|
||||||
var errorTemplate = `bee: %s.
|
|
||||||
Use {{"bee help" | bold}} for more information.
|
|
||||||
`
|
|
||||||
|
|
||||||
var cmdUsage = `Use {{printf "bee help %s" .Name | bold}} for more information.{{endline}}`
|
|
||||||
|
|
||||||
func usage() {
|
|
||||||
tmpl(usageTemplate, availableCommands)
|
|
||||||
os.Exit(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func tmpl(text string, data interface{}) {
|
|
||||||
output := NewColorWriter(os.Stderr)
|
|
||||||
|
|
||||||
t := template.New("usage").Funcs(BeeFuncMap())
|
|
||||||
template.Must(t.Parse(text))
|
|
||||||
|
|
||||||
err := t.Execute(output, data)
|
|
||||||
MustCheck(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func help(args []string) {
|
|
||||||
if len(args) == 0 {
|
|
||||||
usage()
|
|
||||||
}
|
|
||||||
if len(args) != 1 {
|
|
||||||
printErrorAndExit("Too many arguments")
|
|
||||||
}
|
|
||||||
|
|
||||||
arg := args[0]
|
|
||||||
|
|
||||||
for _, cmd := range availableCommands {
|
|
||||||
if cmd.Name() == arg {
|
|
||||||
tmpl(helpTemplate, cmd)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
printErrorAndExit("Unknown help topic")
|
|
||||||
}
|
|
||||||
|
|
||||||
func printErrorAndExit(message string) {
|
|
||||||
tmpl(fmt.Sprintf(errorTemplate, message), nil)
|
|
||||||
os.Exit(2)
|
|
||||||
}
|
|
12
bee.json
12
bee.json
|
@ -1,11 +1,8 @@
|
||||||
{
|
{
|
||||||
"version": 0,
|
"version": 0,
|
||||||
"gopm": {
|
|
||||||
"enable": false,
|
|
||||||
"install": false
|
|
||||||
},
|
|
||||||
"go_install": false,
|
"go_install": false,
|
||||||
"watch_ext": [],
|
"watch_ext": [".go"],
|
||||||
|
"watch_ext_static": [".html", ".tpl", ".js", ".css"],
|
||||||
"dir_structure": {
|
"dir_structure": {
|
||||||
"watch_all": false,
|
"watch_all": false,
|
||||||
"controllers": "",
|
"controllers": "",
|
||||||
|
@ -16,5 +13,6 @@
|
||||||
"envs": [],
|
"envs": [],
|
||||||
"database": {
|
"database": {
|
||||||
"driver": "mysql"
|
"driver": "mysql"
|
||||||
}
|
},
|
||||||
}
|
"enable_reload": false
|
||||||
|
}
|
||||||
|
|
108
cmd/bee.go
Normal file
108
cmd/bee.go
Normal file
|
@ -0,0 +1,108 @@
|
||||||
|
// Copyright 2013 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
|
||||||
|
// Package cmd ...
|
||||||
|
package cmd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/api"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/bale"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/beefix"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/beegopro"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/dev"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/dlv"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/dockerize"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/generate"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/hprose"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/migrate"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/new"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/pack"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/rs"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/run"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/server"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/update"
|
||||||
|
_ "github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
func IfGenerateDocs(name string, args []string) bool {
|
||||||
|
if name != "generate" {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
for _, a := range args {
|
||||||
|
if a == "docs" {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
var usageTemplate = `Bee is a Fast and Flexible tool for managing your Beego Web Application.
|
||||||
|
|
||||||
|
You are using bee for beego v2.x. If you are working on beego v1.x, please downgrade version to bee v1.12.0
|
||||||
|
|
||||||
|
{{"USAGE" | headline}}
|
||||||
|
{{"bee command [arguments]" | bold}}
|
||||||
|
|
||||||
|
{{"AVAILABLE COMMANDS" | headline}}
|
||||||
|
{{range .}}{{if .Runnable}}
|
||||||
|
{{.Name | printf "%-11s" | bold}} {{.Short}}{{end}}{{end}}
|
||||||
|
|
||||||
|
Use {{"bee help [command]" | bold}} for more information about a command.
|
||||||
|
|
||||||
|
{{"ADDITIONAL HELP TOPICS" | headline}}
|
||||||
|
{{range .}}{{if not .Runnable}}
|
||||||
|
{{.Name | printf "%-11s"}} {{.Short}}{{end}}{{end}}
|
||||||
|
|
||||||
|
Use {{"bee help [topic]" | bold}} for more information about that topic.
|
||||||
|
`
|
||||||
|
|
||||||
|
var helpTemplate = `{{"USAGE" | headline}}
|
||||||
|
{{.UsageLine | printf "bee %s" | bold}}
|
||||||
|
{{if .Options}}{{endline}}{{"OPTIONS" | headline}}{{range $k,$v := .Options}}
|
||||||
|
{{$k | printf "-%s" | bold}}
|
||||||
|
{{$v}}
|
||||||
|
{{end}}{{end}}
|
||||||
|
{{"DESCRIPTION" | headline}}
|
||||||
|
{{tmpltostr .Long . | trim}}
|
||||||
|
`
|
||||||
|
|
||||||
|
var ErrorTemplate = `bee: %s.
|
||||||
|
Use {{"bee help" | bold}} for more information.
|
||||||
|
`
|
||||||
|
|
||||||
|
func Usage() {
|
||||||
|
utils.Tmpl(usageTemplate, commands.AvailableCommands)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Help(args []string) {
|
||||||
|
if len(args) == 0 {
|
||||||
|
Usage()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if len(args) != 1 {
|
||||||
|
utils.PrintErrorAndExit("Too many arguments", ErrorTemplate)
|
||||||
|
}
|
||||||
|
|
||||||
|
arg := args[0]
|
||||||
|
|
||||||
|
for _, cmd := range commands.AvailableCommands {
|
||||||
|
if cmd.Name() == arg {
|
||||||
|
utils.Tmpl(helpTemplate, cmd)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
utils.PrintErrorAndExit("Unknown help topic", ErrorTemplate)
|
||||||
|
}
|
|
@ -12,24 +12,33 @@
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
|
|
||||||
package main
|
package apiapp
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
path "path/filepath"
|
path "path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
"github.com/beego/bee/v2/generate"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
var cmdApiapp = &Command{
|
var CmdApiapp = &commands.Command{
|
||||||
// CustomFlags: true,
|
// CustomFlags: true,
|
||||||
UsageLine: "api [appname]",
|
UsageLine: "api [appname]",
|
||||||
Short: "Creates a Beego API application",
|
Short: "Creates a Beego API application",
|
||||||
Long: `
|
Long: `
|
||||||
The command 'api' creates a Beego API application.
|
The command 'api' creates a Beego API application.
|
||||||
|
now default supoort generate a go modules project.
|
||||||
|
|
||||||
{{"Example:"|bold}}
|
{{"Example:"|bold}}
|
||||||
$ bee api [appname] [-tables=""] [-driver=mysql] [-conn=root:@tcp(127.0.0.1:3306)/test]
|
$ bee api [appname] [-tables=""] [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"] [-gopath=false] [-beego=v1.12.3]
|
||||||
|
|
||||||
If 'conn' argument is empty, the command will generate an example API application. Otherwise the command
|
If 'conn' argument is empty, the command will generate an example API application. Otherwise the command
|
||||||
will connect to your database and generate models based on the existing tables.
|
will connect to your database and generate models based on the existing tables.
|
||||||
|
@ -37,6 +46,7 @@ var cmdApiapp = &Command{
|
||||||
The command 'api' creates a folder named [appname] with the following structure:
|
The command 'api' creates a folder named [appname] with the following structure:
|
||||||
|
|
||||||
├── main.go
|
├── main.go
|
||||||
|
├── go.mod
|
||||||
├── {{"conf"|foldername}}
|
├── {{"conf"|foldername}}
|
||||||
│ └── app.conf
|
│ └── app.conf
|
||||||
├── {{"controllers"|foldername}}
|
├── {{"controllers"|foldername}}
|
||||||
|
@ -50,23 +60,23 @@ var cmdApiapp = &Command{
|
||||||
└── object.go
|
└── object.go
|
||||||
└── user.go
|
└── user.go
|
||||||
`,
|
`,
|
||||||
PreRun: func(cmd *Command, args []string) { ShowShortVersionBanner() },
|
PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() },
|
||||||
Run: createapi,
|
Run: createAPI,
|
||||||
}
|
}
|
||||||
|
|
||||||
var apiconf = `appname = {{.Appname}}
|
var apiconf = `appname = {{.Appname}}
|
||||||
httpport = 8080
|
httpport = 8080
|
||||||
runmode = dev
|
runmode = dev
|
||||||
autorender = false
|
autorender = false
|
||||||
copyrequestbody = true
|
copyrequestbody = true
|
||||||
EnableDocs = true
|
EnableDocs = true
|
||||||
|
sqlconn = {{.SQLConnStr}}
|
||||||
`
|
`
|
||||||
var apiMaingo = `package main
|
var apiMaingo = `package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
_ "{{.Appname}}/routers"
|
_ "{{.Appname}}/routers"
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
@ -83,16 +93,18 @@ var apiMainconngo = `package main
|
||||||
import (
|
import (
|
||||||
_ "{{.Appname}}/routers"
|
_ "{{.Appname}}/routers"
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
"github.com/astaxie/beego/orm"
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/beego/v2/client/orm"
|
||||||
{{.DriverPkg}}
|
{{.DriverPkg}}
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
|
||||||
orm.RegisterDataBase("default", "{{.DriverName}}", "{{.conn}}")
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
sqlConn,err := beego.AppConfig.String("sqlconn")
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatal("%s", err)
|
||||||
|
}
|
||||||
|
orm.RegisterDataBase("default", "{{.DriverName}}", sqlConn)
|
||||||
if beego.BConfig.RunMode == "dev" {
|
if beego.BConfig.RunMode == "dev" {
|
||||||
beego.BConfig.WebConfig.DirectoryIndex = true
|
beego.BConfig.WebConfig.DirectoryIndex = true
|
||||||
beego.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
|
beego.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
|
||||||
|
@ -100,6 +112,14 @@ func main() {
|
||||||
beego.Run()
|
beego.Run()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
`
|
||||||
|
var goMod = `
|
||||||
|
module %s
|
||||||
|
|
||||||
|
go %s
|
||||||
|
|
||||||
|
require github.com/beego/beego/v2 %s
|
||||||
|
require github.com/smartystreets/goconvey v1.6.4
|
||||||
`
|
`
|
||||||
|
|
||||||
var apirouter = `// @APIVersion 1.0.0
|
var apirouter = `// @APIVersion 1.0.0
|
||||||
|
@ -114,7 +134,7 @@ package routers
|
||||||
import (
|
import (
|
||||||
"{{.Appname}}/controllers"
|
"{{.Appname}}/controllers"
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
|
@ -134,7 +154,7 @@ func init() {
|
||||||
}
|
}
|
||||||
`
|
`
|
||||||
|
|
||||||
var apiModels = `package models
|
var APIModels = `package models
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
@ -189,7 +209,7 @@ func Delete(ObjectId string) {
|
||||||
|
|
||||||
`
|
`
|
||||||
|
|
||||||
var apiModels2 = `package models
|
var APIModels2 = `package models
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
@ -283,7 +303,7 @@ import (
|
||||||
"{{.Appname}}/models"
|
"{{.Appname}}/models"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Operations about object
|
// Operations about object
|
||||||
|
@ -376,7 +396,7 @@ import (
|
||||||
"{{.Appname}}/models"
|
"{{.Appname}}/models"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Operations about Users
|
// Operations about Users
|
||||||
|
@ -501,12 +521,13 @@ import (
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
_ "{{.Appname}}/routers"
|
_ "{{.Appname}}/routers"
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
|
"github.com/beego/beego/v2/core/logs"
|
||||||
. "github.com/smartystreets/goconvey/convey"
|
. "github.com/smartystreets/goconvey/convey"
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
_, file, _, _ := runtime.Caller(1)
|
_, file, _, _ := runtime.Caller(0)
|
||||||
apppath, _ := filepath.Abs(filepath.Dir(filepath.Join(file, ".." + string(filepath.Separator))))
|
apppath, _ := filepath.Abs(filepath.Dir(filepath.Join(file, ".." + string(filepath.Separator))))
|
||||||
beego.TestBeegoInit(apppath)
|
beego.TestBeegoInit(apppath)
|
||||||
}
|
}
|
||||||
|
@ -517,7 +538,7 @@ func TestGet(t *testing.T) {
|
||||||
w := httptest.NewRecorder()
|
w := httptest.NewRecorder()
|
||||||
beego.BeeApp.Handlers.ServeHTTP(w, r)
|
beego.BeeApp.Handlers.ServeHTTP(w, r)
|
||||||
|
|
||||||
beego.Trace("testing", "TestGet", "Code[%d]\n%s", w.Code, w.Body.String())
|
logs.Info("testing", "TestGet", "Code[%d]\n%s", w.Code, w.Body.String())
|
||||||
|
|
||||||
Convey("Subject: Test Station Endpoint\n", t, func() {
|
Convey("Subject: Test Station Endpoint\n", t, func() {
|
||||||
Convey("Status Code Should Be 200", func() {
|
Convey("Status Code Should Be 200", func() {
|
||||||
|
@ -530,135 +551,144 @@ func TestGet(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
`
|
`
|
||||||
|
var gopath utils.DocValue
|
||||||
|
var beegoVersion utils.DocValue
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
cmdApiapp.Flag.Var(&tables, "tables", "List of table names separated by a comma.")
|
CmdApiapp.Flag.Var(&generate.Tables, "tables", "List of table names separated by a comma.")
|
||||||
cmdApiapp.Flag.Var(&driver, "driver", "Database driver. Either mysql, postgres or sqlite.")
|
CmdApiapp.Flag.Var(&generate.SQLDriver, "driver", "Database driver. Either mysql, postgres or sqlite.")
|
||||||
cmdApiapp.Flag.Var(&conn, "conn", "Connection string used by the driver to connect to a database instance.")
|
CmdApiapp.Flag.Var(&generate.SQLConn, "conn", "Connection string used by the driver to connect to a database instance.")
|
||||||
|
CmdApiapp.Flag.Var(&gopath, "gopath", "Support go path,default false")
|
||||||
|
CmdApiapp.Flag.Var(&beegoVersion, "beego", "set beego version,only take effect by go mod")
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdApiapp)
|
||||||
}
|
}
|
||||||
|
|
||||||
func createapi(cmd *Command, args []string) int {
|
func createAPI(cmd *commands.Command, args []string) int {
|
||||||
output := cmd.Out()
|
output := cmd.Out()
|
||||||
|
|
||||||
if len(args) < 1 {
|
if len(args) < 1 {
|
||||||
logger.Fatal("Argument [appname] is missing")
|
beeLogger.Log.Fatal("Argument [appname] is missing")
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(args) > 1 {
|
if len(args) >= 2 {
|
||||||
cmd.Flag.Parse(args[1:])
|
err := cmd.Flag.Parse(args[1:])
|
||||||
}
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatal("Parse args err " + err.Error())
|
||||||
apppath, packpath, err := checkEnv(args[0])
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("%s", err)
|
|
||||||
}
|
|
||||||
if driver == "" {
|
|
||||||
driver = "mysql"
|
|
||||||
}
|
|
||||||
if conn == "" {
|
|
||||||
}
|
|
||||||
|
|
||||||
logger.Info("Creating API...")
|
|
||||||
|
|
||||||
os.MkdirAll(apppath, 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", apppath, "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "conf"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "conf"), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "controllers"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "controllers"), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "tests"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "tests"), "\x1b[0m")
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "conf", "app.conf"), "\x1b[0m")
|
|
||||||
WriteToFile(path.Join(apppath, "conf", "app.conf"),
|
|
||||||
strings.Replace(apiconf, "{{.Appname}}", path.Base(args[0]), -1))
|
|
||||||
|
|
||||||
if conn != "" {
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "main.go"), "\x1b[0m")
|
|
||||||
maingoContent := strings.Replace(apiMainconngo, "{{.Appname}}", packpath, -1)
|
|
||||||
maingoContent = strings.Replace(maingoContent, "{{.DriverName}}", string(driver), -1)
|
|
||||||
if driver == "mysql" {
|
|
||||||
maingoContent = strings.Replace(maingoContent, "{{.DriverPkg}}", `_ "github.com/go-sql-driver/mysql"`, -1)
|
|
||||||
} else if driver == "postgres" {
|
|
||||||
maingoContent = strings.Replace(maingoContent, "{{.DriverPkg}}", `_ "github.com/lib/pq"`, -1)
|
|
||||||
}
|
}
|
||||||
WriteToFile(path.Join(apppath, "main.go"),
|
}
|
||||||
|
var appPath string
|
||||||
|
var packPath string
|
||||||
|
var err error
|
||||||
|
if gopath == `true` {
|
||||||
|
beeLogger.Log.Info("Generate api project support GOPATH")
|
||||||
|
version.ShowShortVersionBanner()
|
||||||
|
appPath, packPath, err = utils.CheckEnv(args[0])
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("%s", err)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Info("Generate api project support go modules.")
|
||||||
|
appPath = path.Join(utils.GetBeeWorkPath(), args[0])
|
||||||
|
packPath = args[0]
|
||||||
|
if beegoVersion.String() == `` {
|
||||||
|
beegoVersion.Set(utils.BEEGO_VERSION)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if utils.IsExist(appPath) {
|
||||||
|
beeLogger.Log.Errorf(colors.Bold("Application '%s' already exists"), appPath)
|
||||||
|
beeLogger.Log.Warn(colors.Bold("Do you want to overwrite it? [Yes|No] "))
|
||||||
|
if !utils.AskForConfirmation() {
|
||||||
|
os.Exit(2)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
appName := path.Base(args[0])
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("%s", err)
|
||||||
|
}
|
||||||
|
if generate.SQLDriver == "" {
|
||||||
|
generate.SQLDriver = "mysql"
|
||||||
|
}
|
||||||
|
|
||||||
|
beeLogger.Log.Info("Creating API...")
|
||||||
|
|
||||||
|
os.MkdirAll(appPath, 0755)
|
||||||
|
if gopath != `true` { //generate first for calc model name
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "go.mod"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(appPath, "go.mod"), fmt.Sprintf(goMod, packPath, utils.GetGoVersionSkipMinor(), beegoVersion.String()))
|
||||||
|
}
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", appPath, "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "conf"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "conf"), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "controllers"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "controllers"), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "tests"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "tests"), "\x1b[0m")
|
||||||
|
|
||||||
|
if generate.SQLConn != "" {
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "conf", "app.conf"), "\x1b[0m")
|
||||||
|
confContent := strings.Replace(apiconf, "{{.Appname}}", appName, -1)
|
||||||
|
confContent = strings.Replace(confContent, "{{.SQLConnStr}}", generate.SQLConn.String(), -1)
|
||||||
|
utils.WriteToFile(path.Join(appPath, "conf", "app.conf"), confContent)
|
||||||
|
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "main.go"), "\x1b[0m")
|
||||||
|
mainGoContent := strings.Replace(apiMainconngo, "{{.Appname}}", packPath, -1)
|
||||||
|
mainGoContent = strings.Replace(mainGoContent, "{{.DriverName}}", string(generate.SQLDriver), -1)
|
||||||
|
if generate.SQLDriver == "mysql" {
|
||||||
|
mainGoContent = strings.Replace(mainGoContent, "{{.DriverPkg}}", `_ "github.com/go-sql-driver/mysql"`, -1)
|
||||||
|
} else if generate.SQLDriver == "postgres" {
|
||||||
|
mainGoContent = strings.Replace(mainGoContent, "{{.DriverPkg}}", `_ "github.com/lib/pq"`, -1)
|
||||||
|
}
|
||||||
|
utils.WriteToFile(path.Join(appPath, "main.go"),
|
||||||
strings.Replace(
|
strings.Replace(
|
||||||
maingoContent,
|
mainGoContent,
|
||||||
"{{.conn}}",
|
"{{.conn}}",
|
||||||
conn.String(),
|
generate.SQLConn.String(),
|
||||||
-1,
|
-1,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
logger.Infof("Using '%s' as 'driver'", driver)
|
beeLogger.Log.Infof("Using '%s' as 'driver'", generate.SQLDriver)
|
||||||
logger.Infof("Using '%s' as 'conn'", conn)
|
beeLogger.Log.Infof("Using '%s' as 'conn'", generate.SQLConn)
|
||||||
logger.Infof("Using '%s' as 'tables'", tables)
|
beeLogger.Log.Infof("Using '%s' as 'tables'", generate.Tables)
|
||||||
generateAppcode(string(driver), string(conn), "3", string(tables), apppath)
|
generate.GenerateAppcode(string(generate.SQLDriver), string(generate.SQLConn), "3", string(generate.Tables), appPath)
|
||||||
} else {
|
} else {
|
||||||
os.Mkdir(path.Join(apppath, "models"), 0755)
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "conf", "app.conf"), "\x1b[0m")
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models"), "\x1b[0m")
|
confContent := strings.Replace(apiconf, "{{.Appname}}", appName, -1)
|
||||||
os.Mkdir(path.Join(apppath, "routers"), 0755)
|
confContent = strings.Replace(confContent, "{{.SQLConnStr}}", "", -1)
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "routers")+string(path.Separator), "\x1b[0m")
|
utils.WriteToFile(path.Join(appPath, "conf", "app.conf"), confContent)
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "controllers", "object.go"), "\x1b[0m")
|
os.Mkdir(path.Join(appPath, "models"), 0755)
|
||||||
WriteToFile(path.Join(apppath, "controllers", "object.go"),
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "models"), "\x1b[0m")
|
||||||
strings.Replace(apiControllers, "{{.Appname}}", packpath, -1))
|
os.Mkdir(path.Join(appPath, "routers"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "routers")+string(path.Separator), "\x1b[0m")
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "controllers", "user.go"), "\x1b[0m")
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "controllers", "object.go"), "\x1b[0m")
|
||||||
WriteToFile(path.Join(apppath, "controllers", "user.go"),
|
utils.WriteToFile(path.Join(appPath, "controllers", "object.go"),
|
||||||
strings.Replace(apiControllers2, "{{.Appname}}", packpath, -1))
|
strings.Replace(apiControllers, "{{.Appname}}", packPath, -1))
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "tests", "default_test.go"), "\x1b[0m")
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "controllers", "user.go"), "\x1b[0m")
|
||||||
WriteToFile(path.Join(apppath, "tests", "default_test.go"),
|
utils.WriteToFile(path.Join(appPath, "controllers", "user.go"),
|
||||||
strings.Replace(apiTests, "{{.Appname}}", packpath, -1))
|
strings.Replace(apiControllers2, "{{.Appname}}", packPath, -1))
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "routers", "router.go"), "\x1b[0m")
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "tests", "default_test.go"), "\x1b[0m")
|
||||||
WriteToFile(path.Join(apppath, "routers", "router.go"),
|
utils.WriteToFile(path.Join(appPath, "tests", "default_test.go"),
|
||||||
strings.Replace(apirouter, "{{.Appname}}", packpath, -1))
|
strings.Replace(apiTests, "{{.Appname}}", packPath, -1))
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models", "object.go"), "\x1b[0m")
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "routers", "router.go"), "\x1b[0m")
|
||||||
WriteToFile(path.Join(apppath, "models", "object.go"), apiModels)
|
utils.WriteToFile(path.Join(appPath, "routers", "router.go"),
|
||||||
|
strings.Replace(apirouter, "{{.Appname}}", packPath, -1))
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models", "user.go"), "\x1b[0m")
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "models", "object.go"), "\x1b[0m")
|
||||||
WriteToFile(path.Join(apppath, "models", "user.go"), apiModels2)
|
utils.WriteToFile(path.Join(appPath, "models", "object.go"), APIModels)
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "main.go"), "\x1b[0m")
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "models", "user.go"), "\x1b[0m")
|
||||||
WriteToFile(path.Join(apppath, "main.go"),
|
utils.WriteToFile(path.Join(appPath, "models", "user.go"), APIModels2)
|
||||||
strings.Replace(apiMaingo, "{{.Appname}}", packpath, -1))
|
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "main.go"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(appPath, "main.go"),
|
||||||
|
strings.Replace(apiMaingo, "{{.Appname}}", packPath, -1))
|
||||||
}
|
}
|
||||||
logger.Success("New API successfully created!")
|
beeLogger.Log.Success("New API successfully created!")
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
func checkEnv(appname string) (apppath, packpath string, err error) {
|
|
||||||
gps := GetGOPATHs()
|
|
||||||
if len(gps) == 0 {
|
|
||||||
logger.Fatal("GOPATH environment variable is not set or empty")
|
|
||||||
}
|
|
||||||
currpath, _ := os.Getwd()
|
|
||||||
currpath = path.Join(currpath, appname)
|
|
||||||
for _, gpath := range gps {
|
|
||||||
gsrcpath := path.Join(gpath, "src")
|
|
||||||
if strings.HasPrefix(currpath, gsrcpath) {
|
|
||||||
packpath = strings.Replace(currpath[len(gsrcpath)+1:], string(path.Separator), "/", -1)
|
|
||||||
return currpath, packpath, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// In case of multiple paths in the GOPATH, by default
|
|
||||||
// we use the first path
|
|
||||||
gopath := gps[0]
|
|
||||||
|
|
||||||
logger.Warn("You current workdir is not inside $GOPATH/src.")
|
|
||||||
logger.Debugf("GOPATH: %s", __FILE__(), __LINE__(), gopath)
|
|
||||||
|
|
||||||
gosrcpath := path.Join(gopath, "src")
|
|
||||||
apppath = path.Join(gosrcpath, appname)
|
|
||||||
|
|
||||||
if _, e := os.Stat(apppath); os.IsNotExist(e) == false {
|
|
||||||
err = fmt.Errorf("Cannot create application without removing '%s' first.", apppath)
|
|
||||||
logger.Errorf("Path '%s' already exists", apppath)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
packpath = strings.Join(strings.Split(apppath[len(gosrcpath)+1:], string(path.Separator)), "/")
|
|
||||||
return
|
|
||||||
}
|
|
|
@ -11,8 +11,7 @@
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
|
package bale
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
@ -24,9 +23,15 @@ import (
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"runtime"
|
"runtime"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
"github.com/beego/bee/v2/config"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
var cmdBale = &Command{
|
var CmdBale = &commands.Command{
|
||||||
UsageLine: "bale",
|
UsageLine: "bale",
|
||||||
Short: "Transforms non-Go files to Go source files",
|
Short: "Transforms non-Go files to Go source files",
|
||||||
Long: `Bale command compress all the static files in to a single binary file.
|
Long: `Bale command compress all the static files in to a single binary file.
|
||||||
|
@ -37,47 +42,46 @@ var cmdBale = &Command{
|
||||||
It will auto-generate an unpack function to the main package then run it during the runtime.
|
It will auto-generate an unpack function to the main package then run it during the runtime.
|
||||||
This is mainly used for zealots who are requiring 100% Go code.
|
This is mainly used for zealots who are requiring 100% Go code.
|
||||||
`,
|
`,
|
||||||
PreRun: func(cmd *Command, args []string) { ShowShortVersionBanner() },
|
PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() },
|
||||||
Run: runBale,
|
Run: runBale,
|
||||||
}
|
}
|
||||||
|
|
||||||
func runBale(cmd *Command, args []string) int {
|
func init() {
|
||||||
err := loadConfig()
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdBale)
|
||||||
if err != nil {
|
}
|
||||||
logger.Fatalf("Failed to load configuration: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
|
func runBale(cmd *commands.Command, args []string) int {
|
||||||
os.RemoveAll("bale")
|
os.RemoveAll("bale")
|
||||||
os.Mkdir("bale", os.ModePerm)
|
os.Mkdir("bale", os.ModePerm)
|
||||||
|
|
||||||
// Pack and compress data
|
// Pack and compress data
|
||||||
for _, p := range conf.Bale.Dirs {
|
for _, p := range config.Conf.Bale.Dirs {
|
||||||
if !isExist(p) {
|
if !utils.IsExist(p) {
|
||||||
logger.Warnf("Skipped directory: %s", p)
|
beeLogger.Log.Warnf("Skipped directory: %s", p)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
logger.Infof("Packaging directory: %s", p)
|
beeLogger.Log.Infof("Packaging directory: %s", p)
|
||||||
filepath.Walk(p, walkFn)
|
filepath.Walk(p, walkFn)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Generate auto-uncompress function.
|
// Generate auto-uncompress function.
|
||||||
buf := new(bytes.Buffer)
|
buf := new(bytes.Buffer)
|
||||||
buf.WriteString(fmt.Sprintf(BaleHeader, conf.Bale.Import,
|
buf.WriteString(fmt.Sprintf(BaleHeader, config.Conf.Bale.Import,
|
||||||
strings.Join(resFiles, "\",\n\t\t\""),
|
strings.Join(resFiles, "\",\n\t\t\""),
|
||||||
strings.Join(resFiles, ",\n\t\tbale.R")))
|
strings.Join(resFiles, ",\n\t\tbale.R")))
|
||||||
|
|
||||||
fw, err := os.Create("bale.go")
|
fw, err := os.Create("bale.go")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Failed to create file: %s", err)
|
beeLogger.Log.Fatalf("Failed to create file: %s", err)
|
||||||
}
|
}
|
||||||
defer fw.Close()
|
defer fw.Close()
|
||||||
|
|
||||||
_, err = fw.Write(buf.Bytes())
|
_, err = fw.Write(buf.Bytes())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Failed to write data: %s", err)
|
beeLogger.Log.Fatalf("Failed to write data: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Success("Baled resources successfully!")
|
beeLogger.Log.Success("Baled resources successfully!")
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -138,7 +142,7 @@ func saveFile(filePath string, b []byte) (int, error) {
|
||||||
|
|
||||||
var resFiles = make([]string, 0, 10)
|
var resFiles = make([]string, 0, 10)
|
||||||
|
|
||||||
func walkFn(resPath string, info os.FileInfo, err error) error {
|
func walkFn(resPath string, info os.FileInfo, _ error) error {
|
||||||
if info.IsDir() || filterSuffix(resPath) {
|
if info.IsDir() || filterSuffix(resPath) {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
@ -146,7 +150,7 @@ func walkFn(resPath string, info os.FileInfo, err error) error {
|
||||||
// Open resource files
|
// Open resource files
|
||||||
fr, err := os.Open(resPath)
|
fr, err := os.Open(resPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Failed to read file: %s", err)
|
beeLogger.Log.Fatalf("Failed to read file: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Convert path
|
// Convert path
|
||||||
|
@ -164,7 +168,7 @@ func walkFn(resPath string, info os.FileInfo, err error) error {
|
||||||
os.MkdirAll(path.Dir(resPath), os.ModePerm)
|
os.MkdirAll(path.Dir(resPath), os.ModePerm)
|
||||||
fw, err := os.Create("bale/" + resPath + ".go")
|
fw, err := os.Create("bale/" + resPath + ".go")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Failed to create file: %s", err)
|
beeLogger.Log.Fatalf("Failed to create file: %s", err)
|
||||||
}
|
}
|
||||||
defer fw.Close()
|
defer fw.Close()
|
||||||
|
|
||||||
|
@ -184,7 +188,7 @@ func walkFn(resPath string, info os.FileInfo, err error) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
func filterSuffix(name string) bool {
|
func filterSuffix(name string) bool {
|
||||||
for _, s := range conf.Bale.IngExt {
|
for _, s := range config.Conf.Bale.IngExt {
|
||||||
if strings.HasSuffix(name, s) {
|
if strings.HasSuffix(name, s) {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
48
cmd/commands/beefix/fix.go
Normal file
48
cmd/commands/beefix/fix.go
Normal file
|
@ -0,0 +1,48 @@
|
||||||
|
package beefix
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
var CmdFix = &commands.Command{
|
||||||
|
UsageLine: "fix",
|
||||||
|
Short: "Fixes your application by making it compatible with newer versions of Beego",
|
||||||
|
Long: `
|
||||||
|
The command 'fix' will try to solve those issues by upgrading your code base
|
||||||
|
to be compatible with Beego old version
|
||||||
|
-s source version
|
||||||
|
-t target version
|
||||||
|
|
||||||
|
example: bee fix -s 1 -t 2 means that upgrade Beego version from v1.x to v2.x
|
||||||
|
`,
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
source, target utils.DocValue
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
CmdFix.Run = runFix
|
||||||
|
CmdFix.PreRun = func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() }
|
||||||
|
CmdFix.Flag.Var(&source, "s", "source version")
|
||||||
|
CmdFix.Flag.Var(&target, "t", "target version")
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdFix)
|
||||||
|
}
|
||||||
|
|
||||||
|
func runFix(cmd *commands.Command, args []string) int {
|
||||||
|
t := target.String()
|
||||||
|
if t == "" || t == "1.6" {
|
||||||
|
return fixTo16(cmd, args)
|
||||||
|
} else if strings.HasPrefix(t, "2") {
|
||||||
|
// upgrade to v2
|
||||||
|
return fix1To2()
|
||||||
|
}
|
||||||
|
|
||||||
|
beeLogger.Log.Info("The target is compatible version, do nothing")
|
||||||
|
return 0
|
||||||
|
}
|
61
cmd/commands/beefix/fix1To2.go
Normal file
61
cmd/commands/beefix/fix1To2.go
Normal file
|
@ -0,0 +1,61 @@
|
||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package beefix
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
func fix1To2() int {
|
||||||
|
beeLogger.Log.Info("Upgrading the application...")
|
||||||
|
|
||||||
|
cmdStr := `go get -u github.com/beego/beego/v2@master`
|
||||||
|
err := runShell(cmdStr)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
beeLogger.Log.Error(`fetch v2.0.1 failed. Please try to run: export GO111MODULE=on
|
||||||
|
and if your network is not stable, please try to use proxy, for example: export GOPROXY=https://goproxy.cn;'
|
||||||
|
`)
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
|
||||||
|
cmdStr = `find ./ -name '*.go' -type f -exec sed -i '' -e 's/github.com\/astaxie\/beego/github.com\/beego\/beego\/v2\/adapter/g' {} \;`
|
||||||
|
err = runShell(cmdStr)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
cmdStr = `find ./ -name '*.go' -type f -exec sed -i '' -e 's/"github.com\/beego\/beego\/v2\/adapter"/beego "github.com\/beego\/beego\/v2\/adapter"/g' {} \;`
|
||||||
|
err = runShell(cmdStr)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func runShell(cmdStr string) error {
|
||||||
|
c := exec.Command("sh", "-c", cmdStr)
|
||||||
|
c.Stdout = os.Stdout
|
||||||
|
err := c.Run()
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("execute command [%s] failed: %s", cmdStr, err.Error())
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
|
@ -1,4 +1,18 @@
|
||||||
package main
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package beefix
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
@ -9,31 +23,21 @@ import (
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
)
|
)
|
||||||
|
|
||||||
var cmdFix = &Command{
|
// fixTo16 upgrade beego to 1.6
|
||||||
UsageLine: "fix",
|
func fixTo16(cmd *commands.Command, args []string) int {
|
||||||
Short: "Fixes your application by making it compatible with newer versions of Beego",
|
|
||||||
Long: `As of {{"Beego 1.6"|bold}}, there are some backward compatibility issues.
|
|
||||||
|
|
||||||
The command 'fix' will try to solve those issues by upgrading your code base
|
|
||||||
to be compatible with Beego version 1.6+.
|
|
||||||
`,
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
cmdFix.Run = runFix
|
|
||||||
cmdFix.PreRun = func(cmd *Command, args []string) { ShowShortVersionBanner() }
|
|
||||||
}
|
|
||||||
|
|
||||||
func runFix(cmd *Command, args []string) int {
|
|
||||||
output := cmd.Out()
|
output := cmd.Out()
|
||||||
|
|
||||||
logger.Info("Upgrading the application...")
|
beeLogger.Log.Info("Upgrading the application...")
|
||||||
|
|
||||||
dir, err := os.Getwd()
|
dir, err := os.Getwd()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Error while getting the current working directory: %s", err)
|
beeLogger.Log.Fatalf("Error while getting the current working directory: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
|
filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
|
||||||
|
@ -50,13 +54,13 @@ func runFix(cmd *Command, args []string) int {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
err = fixFile(path)
|
err = fixFile(path)
|
||||||
fmt.Fprintf(output, GreenBold("\tfix\t")+"%s\n", path)
|
fmt.Fprintf(output, colors.GreenBold("\tfix\t")+"%s\n", path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Could not fix file: %s", err)
|
beeLogger.Log.Errorf("Could not fix file: %s", err)
|
||||||
}
|
}
|
||||||
return err
|
return err
|
||||||
})
|
})
|
||||||
logger.Success("Upgrade Done!")
|
beeLogger.Log.Success("Upgrade Done!")
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
63
cmd/commands/beegopro/beegopro.go
Normal file
63
cmd/commands/beegopro/beegopro.go
Normal file
|
@ -0,0 +1,63 @@
|
||||||
|
// Copyright 2013 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/internal/app/module/beegopro"
|
||||||
|
"github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
var CmdBeegoPro = &commands.Command{
|
||||||
|
UsageLine: "pro [command]",
|
||||||
|
Short: "Source code generator",
|
||||||
|
Long: ``,
|
||||||
|
Run: BeegoPro,
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
CmdBeegoPro.Flag.Var(&beegopro.SQL, "sql", "sql file path")
|
||||||
|
CmdBeegoPro.Flag.Var(&beegopro.SQLMode, "sqlmode", "sql mode")
|
||||||
|
CmdBeegoPro.Flag.Var(&beegopro.SQLModePath, "sqlpath", "sql mode path")
|
||||||
|
CmdBeegoPro.Flag.Var(&beegopro.GitRemotePath, "url", "git remote path")
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdBeegoPro)
|
||||||
|
}
|
||||||
|
|
||||||
|
func BeegoPro(cmd *commands.Command, args []string) int {
|
||||||
|
if len(args) < 1 {
|
||||||
|
beeLogger.Log.Fatal("Command is missing")
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(args) >= 2 {
|
||||||
|
cmd.Flag.Parse(args[1:])
|
||||||
|
}
|
||||||
|
|
||||||
|
gcmd := args[0]
|
||||||
|
switch gcmd {
|
||||||
|
case "gen":
|
||||||
|
beegopro.DefaultBeegoPro.Run()
|
||||||
|
case "toml":
|
||||||
|
beegopro.DefaultBeegoPro.InitToml()
|
||||||
|
case "config":
|
||||||
|
beegopro.DefaultBeegoPro.GenConfig()
|
||||||
|
case "migration":
|
||||||
|
beegopro.DefaultBeegoPro.Migration(args)
|
||||||
|
default:
|
||||||
|
beeLogger.Log.Fatal("Command is missing")
|
||||||
|
}
|
||||||
|
beeLogger.Log.Successf("%s successfully generated!", strings.Title(gcmd))
|
||||||
|
return 0
|
||||||
|
}
|
95
cmd/commands/command.go
Normal file
95
cmd/commands/command.go
Normal file
|
@ -0,0 +1,95 @@
|
||||||
|
package commands
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Command is the unit of execution
|
||||||
|
type Command struct {
|
||||||
|
// Run runs the command.
|
||||||
|
// The args are the arguments after the command name.
|
||||||
|
Run func(cmd *Command, args []string) int
|
||||||
|
|
||||||
|
// PreRun performs an operation before running the command
|
||||||
|
PreRun func(cmd *Command, args []string)
|
||||||
|
|
||||||
|
// UsageLine is the one-line Usage message.
|
||||||
|
// The first word in the line is taken to be the command name.
|
||||||
|
UsageLine string
|
||||||
|
|
||||||
|
// Short is the short description shown in the 'go help' output.
|
||||||
|
Short string
|
||||||
|
|
||||||
|
// Long is the long message shown in the 'go help <this-command>' output.
|
||||||
|
Long string
|
||||||
|
|
||||||
|
// Flag is a set of flags specific to this command.
|
||||||
|
Flag flag.FlagSet
|
||||||
|
|
||||||
|
// CustomFlags indicates that the command will do its own
|
||||||
|
// flag parsing.
|
||||||
|
CustomFlags bool
|
||||||
|
|
||||||
|
// output out writer if set in SetOutput(w)
|
||||||
|
output *io.Writer
|
||||||
|
}
|
||||||
|
|
||||||
|
var AvailableCommands = []*Command{}
|
||||||
|
var cmdUsage = `Use {{printf "bee help %s" .Name | bold}} for more information.{{endline}}`
|
||||||
|
|
||||||
|
// Name returns the command's name: the first word in the Usage line.
|
||||||
|
func (c *Command) Name() string {
|
||||||
|
name := c.UsageLine
|
||||||
|
i := strings.Index(name, " ")
|
||||||
|
if i >= 0 {
|
||||||
|
name = name[:i]
|
||||||
|
}
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetOutput sets the destination for Usage and error messages.
|
||||||
|
// If output is nil, os.Stderr is used.
|
||||||
|
func (c *Command) SetOutput(output io.Writer) {
|
||||||
|
c.output = &output
|
||||||
|
}
|
||||||
|
|
||||||
|
// Out returns the out writer of the current command.
|
||||||
|
// If cmd.output is nil, os.Stderr is used.
|
||||||
|
func (c *Command) Out() io.Writer {
|
||||||
|
if c.output != nil {
|
||||||
|
return *c.output
|
||||||
|
}
|
||||||
|
|
||||||
|
return colors.NewColorWriter(os.Stderr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Usage puts out the Usage for the command.
|
||||||
|
func (c *Command) Usage() {
|
||||||
|
utils.Tmpl(cmdUsage, c)
|
||||||
|
os.Exit(2)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Runnable reports whether the command can be run; otherwise
|
||||||
|
// it is a documentation pseudo-command such as import path.
|
||||||
|
func (c *Command) Runnable() bool {
|
||||||
|
return c.Run != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Command) Options() map[string]string {
|
||||||
|
options := make(map[string]string)
|
||||||
|
c.Flag.VisitAll(func(f *flag.Flag) {
|
||||||
|
defaultVal := f.DefValue
|
||||||
|
if len(defaultVal) > 0 {
|
||||||
|
options[f.Name+"="+defaultVal] = f.Usage
|
||||||
|
} else {
|
||||||
|
options[f.Name] = f.Usage
|
||||||
|
}
|
||||||
|
})
|
||||||
|
return options
|
||||||
|
}
|
56
cmd/commands/dev/cmd.go
Normal file
56
cmd/commands/dev/cmd.go
Normal file
|
@ -0,0 +1,56 @@
|
||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package dev
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
var CmdDev = &commands.Command{
|
||||||
|
CustomFlags: true,
|
||||||
|
UsageLine: "dev [command]",
|
||||||
|
Short: "Commands which used to help to develop beego and bee",
|
||||||
|
Long: `
|
||||||
|
Commands that help developer develop, build and test beego.
|
||||||
|
- githook Prepare githooks
|
||||||
|
`,
|
||||||
|
Run: Run,
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdDev)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Run(cmd *commands.Command, args []string) int {
|
||||||
|
if len(args) < 1 {
|
||||||
|
beeLogger.Log.Fatal("Command is missing")
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(args) >= 2 {
|
||||||
|
cmd.Flag.Parse(args[1:])
|
||||||
|
}
|
||||||
|
|
||||||
|
gcmd := args[0]
|
||||||
|
|
||||||
|
switch gcmd {
|
||||||
|
|
||||||
|
case "githook":
|
||||||
|
initGitHook()
|
||||||
|
default:
|
||||||
|
beeLogger.Log.Fatal("Unknown command")
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
47
cmd/commands/dev/githook.go
Normal file
47
cmd/commands/dev/githook.go
Normal file
|
@ -0,0 +1,47 @@
|
||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package dev
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
var preCommit = `
|
||||||
|
goimports -w -format-only ./ \
|
||||||
|
ineffassign . \
|
||||||
|
staticcheck -show-ignored -checks "-ST1017,-U1000,-ST1005,-S1034,-S1012,-SA4006,-SA6005,-SA1019,-SA1024" ./ \
|
||||||
|
`
|
||||||
|
|
||||||
|
// for now, we simply override pre-commit file
|
||||||
|
func initGitHook() {
|
||||||
|
// pcf => pre-commit file
|
||||||
|
pcfPath := "./.git/hooks/pre-commit"
|
||||||
|
pcf, err := os.OpenFile(pcfPath, os.O_RDWR|os.O_CREATE, 0777)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("try to create or open file failed: %s, cause: %s", pcfPath, err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
defer pcf.Close()
|
||||||
|
_, err = pcf.Write(([]byte)(preCommit))
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("could not init githooks: %s", err.Error())
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Successf("The githooks has been added, the content is:\n %s ", preCommit)
|
||||||
|
}
|
||||||
|
}
|
16
cmd/commands/dlv/dlv.go
Normal file
16
cmd/commands/dlv/dlv.go
Normal file
|
@ -0,0 +1,16 @@
|
||||||
|
// Copyright 2017 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
|
||||||
|
// Package dlv ...
|
||||||
|
package dlv
|
249
cmd/commands/dlv/dlv_amd64.go
Normal file
249
cmd/commands/dlv/dlv_amd64.go
Normal file
|
@ -0,0 +1,249 @@
|
||||||
|
// Copyright 2017 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
|
||||||
|
// Package dlv ...
|
||||||
|
package dlv
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
"github.com/fsnotify/fsnotify"
|
||||||
|
"github.com/go-delve/delve/pkg/terminal"
|
||||||
|
"github.com/go-delve/delve/service"
|
||||||
|
"github.com/go-delve/delve/service/debugger"
|
||||||
|
"github.com/go-delve/delve/service/rpc2"
|
||||||
|
"github.com/go-delve/delve/service/rpccommon"
|
||||||
|
)
|
||||||
|
|
||||||
|
var cmdDlv = &commands.Command{
|
||||||
|
CustomFlags: true,
|
||||||
|
UsageLine: "dlv [-package=\"\"] [-port=8181] [-verbose=false]",
|
||||||
|
Short: "Start a debugging session using Delve",
|
||||||
|
Long: `dlv command start a debugging session using debugging tool Delve.
|
||||||
|
|
||||||
|
To debug your application using Delve, use: {{"$ bee dlv" | bold}}
|
||||||
|
|
||||||
|
For more information on Delve: https://github.com/go-delve/delve
|
||||||
|
`,
|
||||||
|
PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() },
|
||||||
|
Run: runDlv,
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
packageName string
|
||||||
|
verbose bool
|
||||||
|
port int
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
fs := flag.NewFlagSet("dlv", flag.ContinueOnError)
|
||||||
|
fs.StringVar(&packageName, "package", "", "The package to debug (Must have a main package)")
|
||||||
|
fs.BoolVar(&verbose, "verbose", false, "Enable verbose mode")
|
||||||
|
fs.IntVar(&port, "port", 8181, "Port to listen to for clients")
|
||||||
|
cmdDlv.Flag = *fs
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, cmdDlv)
|
||||||
|
}
|
||||||
|
|
||||||
|
func runDlv(cmd *commands.Command, args []string) int {
|
||||||
|
if err := cmd.Flag.Parse(args); err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Error while parsing flags: %v", err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
addr = fmt.Sprintf("127.0.0.1:%d", port)
|
||||||
|
paths = make([]string, 0)
|
||||||
|
notifyChan = make(chan int)
|
||||||
|
)
|
||||||
|
|
||||||
|
if err := loadPathsToWatch(&paths); err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Error while loading paths to watch: %v", err.Error())
|
||||||
|
}
|
||||||
|
go startWatcher(paths, notifyChan)
|
||||||
|
return startDelveDebugger(addr, notifyChan)
|
||||||
|
}
|
||||||
|
|
||||||
|
// buildDebug builds a debug binary in the current working directory
|
||||||
|
func buildDebug() (string, error) {
|
||||||
|
args := []string{"-gcflags", "-N -l", "-o", "debug"}
|
||||||
|
args = append(args, utils.SplitQuotedFields("-ldflags='-linkmode internal'")...)
|
||||||
|
args = append(args, packageName)
|
||||||
|
if err := utils.GoCommand("build", args...); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
fp, err := filepath.Abs("./debug")
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return fp, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// loadPathsToWatch loads the paths that needs to be watched for changes
|
||||||
|
func loadPathsToWatch(paths *[]string) error {
|
||||||
|
directory, err := os.Getwd()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
filepath.Walk(directory, func(path string, info os.FileInfo, _ error) error {
|
||||||
|
if strings.HasSuffix(info.Name(), "docs") {
|
||||||
|
return filepath.SkipDir
|
||||||
|
}
|
||||||
|
if strings.HasSuffix(info.Name(), "swagger") {
|
||||||
|
return filepath.SkipDir
|
||||||
|
}
|
||||||
|
if strings.HasSuffix(info.Name(), "vendor") {
|
||||||
|
return filepath.SkipDir
|
||||||
|
}
|
||||||
|
|
||||||
|
if filepath.Ext(info.Name()) == ".go" {
|
||||||
|
*paths = append(*paths, path)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// startDelveDebugger starts the Delve debugger server
|
||||||
|
func startDelveDebugger(addr string, ch chan int) int {
|
||||||
|
beeLogger.Log.Info("Starting Delve Debugger...")
|
||||||
|
|
||||||
|
fp, err := buildDebug()
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Error while building debug binary: %v", err)
|
||||||
|
}
|
||||||
|
defer os.Remove(fp)
|
||||||
|
|
||||||
|
abs, err := filepath.Abs("./debug")
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("%v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create and start the debugger server
|
||||||
|
listener, err := net.Listen("tcp", addr)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not start listener: %s", err)
|
||||||
|
}
|
||||||
|
defer listener.Close()
|
||||||
|
|
||||||
|
server := rpccommon.NewServer(&service.Config{
|
||||||
|
Listener: listener,
|
||||||
|
AcceptMulti: true,
|
||||||
|
APIVersion: 2,
|
||||||
|
ProcessArgs: []string{abs},
|
||||||
|
Debugger: debugger.Config{
|
||||||
|
AttachPid: 0,
|
||||||
|
WorkingDir: ".",
|
||||||
|
Backend: "default",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
if err := server.Run(); err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not start debugger server: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start the Delve client REPL
|
||||||
|
client := rpc2.NewClient(addr)
|
||||||
|
// Make sure the client is restarted when new changes are introduced
|
||||||
|
go func() {
|
||||||
|
for {
|
||||||
|
if val := <-ch; val == 0 {
|
||||||
|
if _, err := client.Restart(true); err != nil {
|
||||||
|
utils.Notify("Error while restarting the client: "+err.Error(), "bee")
|
||||||
|
} else {
|
||||||
|
if verbose {
|
||||||
|
utils.Notify("Delve Debugger Restarted", "bee")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
// Create the terminal and connect it to the client debugger
|
||||||
|
term := terminal.New(client, nil)
|
||||||
|
status, err := term.Run()
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not start Delve REPL: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Stop and kill the debugger server once user quits the REPL
|
||||||
|
if err := server.Stop(); err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not stop Delve server: %v", err)
|
||||||
|
}
|
||||||
|
return status
|
||||||
|
}
|
||||||
|
|
||||||
|
var eventsModTime = make(map[string]int64)
|
||||||
|
|
||||||
|
// startWatcher starts the fsnotify watcher on the passed paths
|
||||||
|
func startWatcher(paths []string, ch chan int) {
|
||||||
|
watcher, err := fsnotify.NewWatcher()
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not start the watcher: %v", err)
|
||||||
|
}
|
||||||
|
defer watcher.Close()
|
||||||
|
|
||||||
|
// Feed the paths to the watcher
|
||||||
|
for _, path := range paths {
|
||||||
|
if err := watcher.Add(path); err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not set a watch on path: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case evt := <-watcher.Events:
|
||||||
|
build := true
|
||||||
|
if filepath.Ext(evt.Name) != ".go" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
mt := utils.GetFileModTime(evt.Name)
|
||||||
|
if t := eventsModTime[evt.Name]; mt == t {
|
||||||
|
build = false
|
||||||
|
}
|
||||||
|
eventsModTime[evt.Name] = mt
|
||||||
|
|
||||||
|
if build {
|
||||||
|
go func() {
|
||||||
|
if verbose {
|
||||||
|
utils.Notify("Rebuilding application with the new changes", "bee")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait 1s before re-build until there is no file change
|
||||||
|
scheduleTime := time.Now().Add(1 * time.Second)
|
||||||
|
time.Sleep(time.Until(scheduleTime))
|
||||||
|
_, err := buildDebug()
|
||||||
|
if err != nil {
|
||||||
|
utils.Notify("Build Failed: "+err.Error(), "bee")
|
||||||
|
} else {
|
||||||
|
ch <- 0 // Notify listeners
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
case err := <-watcher.Errors:
|
||||||
|
if err != nil {
|
||||||
|
ch <- -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
170
cmd/commands/dockerize/dockerize.go
Normal file
170
cmd/commands/dockerize/dockerize.go
Normal file
|
@ -0,0 +1,170 @@
|
||||||
|
// Copyright 2016 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
|
||||||
|
package dockerize
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"text/template"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
const dockerBuildTemplate = `# Build Golang binary
|
||||||
|
FROM {{.BaseImage}} AS build-golang
|
||||||
|
|
||||||
|
WORKDIR {{.Appdir}}
|
||||||
|
|
||||||
|
COPY . .
|
||||||
|
RUN go get -v && go build -v -o /usr/local/bin/{{.Entrypoint}}
|
||||||
|
|
||||||
|
EXPOSE {{.Expose}}
|
||||||
|
CMD ["{{.Entrypoint}}"]
|
||||||
|
`
|
||||||
|
|
||||||
|
const composeBuildTemplate = `version: '3'
|
||||||
|
networks:
|
||||||
|
{{.Appname}}_network_compose:
|
||||||
|
driver: bridge
|
||||||
|
services:
|
||||||
|
{{.Appname}}:
|
||||||
|
container_name: {{.Appname}}
|
||||||
|
build: .
|
||||||
|
restart: unless-stopped
|
||||||
|
networks:
|
||||||
|
{{.Appname}}_network_compose:
|
||||||
|
ports:{{.Expose}}
|
||||||
|
`
|
||||||
|
|
||||||
|
// Dockerfile holds the information about the Docker container.
|
||||||
|
type Dockerfile struct {
|
||||||
|
BaseImage string
|
||||||
|
Appdir string
|
||||||
|
Entrypoint string
|
||||||
|
Expose string
|
||||||
|
}
|
||||||
|
|
||||||
|
// docker-compose.yaml
|
||||||
|
type Composefile struct {
|
||||||
|
Appname string
|
||||||
|
Expose string
|
||||||
|
}
|
||||||
|
|
||||||
|
var CmdDockerize = &commands.Command{
|
||||||
|
CustomFlags: true,
|
||||||
|
UsageLine: "dockerize",
|
||||||
|
Short: "Generates a Dockerfile and docker-compose.yaml for your Beego application",
|
||||||
|
Long: `Dockerize generates a Dockerfile and docker-compose.yaml for your Beego Web Application.
|
||||||
|
The Dockerfile will compile and run the application.
|
||||||
|
The docker-compose.yaml can be used to build and deploy the generated Dockerfile.
|
||||||
|
{{"Example:"|bold}}
|
||||||
|
$ bee dockerize -expose="3000,80,25"
|
||||||
|
`,
|
||||||
|
PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() },
|
||||||
|
Run: dockerizeApp,
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
expose string
|
||||||
|
baseImage string
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
fs := flag.NewFlagSet("dockerize", flag.ContinueOnError)
|
||||||
|
fs.StringVar(&baseImage, "baseimage", "golang:1.20.2", "Set the base image of the Docker container.")
|
||||||
|
fs.StringVar(&expose, "expose", "8080", "Port(s) to expose for the Docker container.")
|
||||||
|
CmdDockerize.Flag = *fs
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdDockerize)
|
||||||
|
}
|
||||||
|
|
||||||
|
func dockerizeApp(cmd *commands.Command, args []string) int {
|
||||||
|
if err := cmd.Flag.Parse(args); err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Error parsing flags: %v", err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
beeLogger.Log.Info("Generating Dockerfile and docker-compose.yaml...")
|
||||||
|
|
||||||
|
gopath := os.Getenv("GOPATH")
|
||||||
|
dir, err := filepath.Abs(".")
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
appdir := strings.Replace(dir, gopath, "", 1)
|
||||||
|
|
||||||
|
// In case of multiple ports to expose inside the container,
|
||||||
|
// replace all the commas with whitespaces.
|
||||||
|
// See the verb EXPOSE in the Docker documentation.
|
||||||
|
exposedockerfile := strings.Replace(expose, ",", " ", -1)
|
||||||
|
|
||||||
|
// Multiple ports expose for docker-compose.yaml
|
||||||
|
ports := strings.Fields(strings.Replace(expose, ",", " ", -1))
|
||||||
|
exposecompose := ""
|
||||||
|
for _, port := range ports {
|
||||||
|
composeport := ("\n - " + "\"" + port + ":" + port + "\"")
|
||||||
|
exposecompose += composeport
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
_, entrypoint := path.Split(appdir)
|
||||||
|
dockerfile := Dockerfile{
|
||||||
|
BaseImage: baseImage,
|
||||||
|
Appdir: appdir,
|
||||||
|
Entrypoint: entrypoint,
|
||||||
|
Expose: exposedockerfile,
|
||||||
|
}
|
||||||
|
composefile := Composefile{
|
||||||
|
Appname: entrypoint,
|
||||||
|
Expose: exposecompose,
|
||||||
|
}
|
||||||
|
|
||||||
|
generateDockerfile(dockerfile)
|
||||||
|
generatecomposefile(composefile)
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func generateDockerfile(df Dockerfile) {
|
||||||
|
t := template.Must(template.New("dockerBuildTemplate").Parse(dockerBuildTemplate)).Funcs(utils.BeeFuncMap())
|
||||||
|
|
||||||
|
f, err := os.Create("Dockerfile")
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Error writing Dockerfile: %v", err.Error())
|
||||||
|
}
|
||||||
|
defer utils.CloseFile(f)
|
||||||
|
|
||||||
|
t.Execute(f, df)
|
||||||
|
|
||||||
|
beeLogger.Log.Success("Dockerfile generated.")
|
||||||
|
}
|
||||||
|
|
||||||
|
func generatecomposefile(df Composefile) {
|
||||||
|
t := template.Must(template.New("composeBuildTemplate").Parse(composeBuildTemplate)).Funcs(utils.BeeFuncMap())
|
||||||
|
|
||||||
|
f, err := os.Create("docker-compose.yaml")
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Error writing docker-compose.yaml: %v", err.Error())
|
||||||
|
}
|
||||||
|
defer utils.CloseFile(f)
|
||||||
|
|
||||||
|
t.Execute(f, df)
|
||||||
|
|
||||||
|
beeLogger.Log.Success("docker-compose.yaml generated.")
|
||||||
|
}
|
234
cmd/commands/generate/generate.go
Normal file
234
cmd/commands/generate/generate.go
Normal file
|
@ -0,0 +1,234 @@
|
||||||
|
// Copyright 2013 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
package generate
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
"github.com/beego/bee/v2/config"
|
||||||
|
"github.com/beego/bee/v2/generate"
|
||||||
|
"github.com/beego/bee/v2/generate/swaggergen"
|
||||||
|
"github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
var CmdGenerate = &commands.Command{
|
||||||
|
UsageLine: "generate [command]",
|
||||||
|
Short: "Source code generator",
|
||||||
|
Long: `▶ {{"To scaffold out your entire application:"|bold}}
|
||||||
|
|
||||||
|
$ bee generate scaffold [scaffoldname] [-fields="title:string,body:text"] [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"]
|
||||||
|
|
||||||
|
▶ {{"To generate a Model based on fields:"|bold}}
|
||||||
|
|
||||||
|
$ bee generate model [modelname] [-fields="name:type"]
|
||||||
|
|
||||||
|
▶ {{"To generate a controller:"|bold}}
|
||||||
|
|
||||||
|
$ bee generate controller [controllerfile]
|
||||||
|
|
||||||
|
▶ {{"To generate a CRUD view:"|bold}}
|
||||||
|
|
||||||
|
$ bee generate view [viewpath]
|
||||||
|
|
||||||
|
▶ {{"To generate a migration file for making database schema updates:"|bold}}
|
||||||
|
|
||||||
|
$ bee generate migration [migrationfile] [-fields="name:type"]
|
||||||
|
|
||||||
|
▶ {{"To generate swagger doc file:"|bold}}
|
||||||
|
|
||||||
|
$ bee generate docs
|
||||||
|
|
||||||
|
▶ {{"To generate swagger doc file:"|bold}}
|
||||||
|
|
||||||
|
$ bee generate routers [-ctrlDir=/path/to/controller/directory] [-routersFile=/path/to/routers/file.go] [-routersPkg=myPackage]
|
||||||
|
|
||||||
|
▶ {{"To generate a test case:"|bold}}
|
||||||
|
|
||||||
|
$ bee generate test [routerfile]
|
||||||
|
|
||||||
|
▶ {{"To generate appcode based on an existing database:"|bold}}
|
||||||
|
|
||||||
|
$ bee generate appcode [-tables=""] [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"] [-level=3]
|
||||||
|
`,
|
||||||
|
PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() },
|
||||||
|
Run: GenerateCode,
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
CmdGenerate.Flag.Var(&generate.Tables, "tables", "List of table names separated by a comma.")
|
||||||
|
CmdGenerate.Flag.Var(&generate.SQLDriver, "driver", "Database SQLDriver. Either mysql, postgres or sqlite.")
|
||||||
|
CmdGenerate.Flag.Var(&generate.SQLConn, "conn", "Connection string used by the SQLDriver to connect to a database instance.")
|
||||||
|
CmdGenerate.Flag.Var(&generate.Level, "level", "Either 1, 2 or 3. i.e. 1=models; 2=models and controllers; 3=models, controllers and routers.")
|
||||||
|
CmdGenerate.Flag.Var(&generate.Fields, "fields", "List of table Fields.")
|
||||||
|
CmdGenerate.Flag.Var(&generate.DDL, "ddl", "Generate DDL Migration")
|
||||||
|
|
||||||
|
// bee generate routers
|
||||||
|
CmdGenerate.Flag.Var(&generate.ControllerDirectory, "ctrlDir",
|
||||||
|
"Controller directory. Bee scans this directory and its sub directory to generate routers")
|
||||||
|
CmdGenerate.Flag.Var(&generate.RoutersFile, "routersFile",
|
||||||
|
"Routers file. If not found, Bee create a new one. Bee will truncates this file and output routers info into this file")
|
||||||
|
CmdGenerate.Flag.Var(&generate.RouterPkg, "routersPkg",
|
||||||
|
`router's package. Default is routers, it means that "package routers" in the generated file`)
|
||||||
|
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdGenerate)
|
||||||
|
}
|
||||||
|
|
||||||
|
func GenerateCode(cmd *commands.Command, args []string) int {
|
||||||
|
currpath, _ := os.Getwd()
|
||||||
|
if len(args) < 1 {
|
||||||
|
beeLogger.Log.Fatal("Command is missing")
|
||||||
|
}
|
||||||
|
|
||||||
|
gcmd := args[0]
|
||||||
|
switch gcmd {
|
||||||
|
case "scaffold":
|
||||||
|
scaffold(cmd, args, currpath)
|
||||||
|
case "docs":
|
||||||
|
swaggergen.GenerateDocs(currpath)
|
||||||
|
case "appcode":
|
||||||
|
appCode(cmd, args, currpath)
|
||||||
|
case "migration":
|
||||||
|
migration(cmd, args, currpath)
|
||||||
|
case "controller":
|
||||||
|
controller(args, currpath)
|
||||||
|
case "model":
|
||||||
|
model(cmd, args, currpath)
|
||||||
|
case "view":
|
||||||
|
view(args, currpath)
|
||||||
|
case "routers":
|
||||||
|
genRouters(cmd, args)
|
||||||
|
default:
|
||||||
|
beeLogger.Log.Fatal("Command is missing")
|
||||||
|
}
|
||||||
|
beeLogger.Log.Successf("%s successfully generated!", strings.Title(gcmd))
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func genRouters(cmd *commands.Command, args []string) {
|
||||||
|
err := cmd.Flag.Parse(args[1:])
|
||||||
|
beeLogger.Log.Infof("input parameter: %v", args)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("could not parse input parameter: %+v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
generate.GenRouters()
|
||||||
|
}
|
||||||
|
|
||||||
|
func scaffold(cmd *commands.Command, args []string, currpath string) {
|
||||||
|
if len(args) < 2 {
|
||||||
|
beeLogger.Log.Fatal("Wrong number of arguments. Run: bee help generate")
|
||||||
|
}
|
||||||
|
|
||||||
|
cmd.Flag.Parse(args[2:])
|
||||||
|
if generate.SQLDriver == "" {
|
||||||
|
generate.SQLDriver = utils.DocValue(config.Conf.Database.Driver)
|
||||||
|
if generate.SQLDriver == "" {
|
||||||
|
generate.SQLDriver = "mysql"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if generate.SQLConn == "" {
|
||||||
|
generate.SQLConn = utils.DocValue(config.Conf.Database.Conn)
|
||||||
|
if generate.SQLConn == "" {
|
||||||
|
generate.SQLConn = "root:@tcp(127.0.0.1:3306)/test"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if generate.Fields == "" {
|
||||||
|
beeLogger.Log.Hint("Fields option should not be empty, i.e. -Fields=\"title:string,body:text\"")
|
||||||
|
beeLogger.Log.Fatal("Wrong number of arguments. Run: bee help generate")
|
||||||
|
}
|
||||||
|
sname := args[1]
|
||||||
|
generate.GenerateScaffold(sname, generate.Fields.String(), currpath, generate.SQLDriver.String(), generate.SQLConn.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
func appCode(cmd *commands.Command, args []string, currpath string) {
|
||||||
|
cmd.Flag.Parse(args[1:])
|
||||||
|
if generate.SQLDriver == "" {
|
||||||
|
generate.SQLDriver = utils.DocValue(config.Conf.Database.Driver)
|
||||||
|
if generate.SQLDriver == "" {
|
||||||
|
generate.SQLDriver = "mysql"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if generate.SQLConn == "" {
|
||||||
|
generate.SQLConn = utils.DocValue(config.Conf.Database.Conn)
|
||||||
|
if generate.SQLConn == "" {
|
||||||
|
if generate.SQLDriver == "mysql" {
|
||||||
|
generate.SQLConn = "root:@tcp(127.0.0.1:3306)/test"
|
||||||
|
} else if generate.SQLDriver == "postgres" {
|
||||||
|
generate.SQLConn = "postgres://postgres:postgres@127.0.0.1:5432/postgres"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if generate.Level == "" {
|
||||||
|
generate.Level = "3"
|
||||||
|
}
|
||||||
|
beeLogger.Log.Infof("Using '%s' as 'SQLDriver'", generate.SQLDriver)
|
||||||
|
beeLogger.Log.Infof("Using '%s' as 'SQLConn'", generate.SQLConn)
|
||||||
|
beeLogger.Log.Infof("Using '%s' as 'Tables'", generate.Tables)
|
||||||
|
beeLogger.Log.Infof("Using '%s' as 'Level'", generate.Level)
|
||||||
|
generate.GenerateAppcode(generate.SQLDriver.String(), generate.SQLConn.String(), generate.Level.String(), generate.Tables.String(), currpath)
|
||||||
|
}
|
||||||
|
|
||||||
|
func migration(cmd *commands.Command, args []string, currpath string) {
|
||||||
|
if len(args) < 2 {
|
||||||
|
beeLogger.Log.Fatal("Wrong number of arguments. Run: bee help generate")
|
||||||
|
}
|
||||||
|
cmd.Flag.Parse(args[2:])
|
||||||
|
mname := args[1]
|
||||||
|
|
||||||
|
beeLogger.Log.Infof("Using '%s' as migration name", mname)
|
||||||
|
|
||||||
|
upsql := ""
|
||||||
|
downsql := ""
|
||||||
|
if generate.Fields != "" {
|
||||||
|
dbMigrator := generate.NewDBDriver()
|
||||||
|
upsql = dbMigrator.GenerateCreateUp(mname)
|
||||||
|
downsql = dbMigrator.GenerateCreateDown(mname)
|
||||||
|
}
|
||||||
|
generate.GenerateMigration(mname, upsql, downsql, currpath)
|
||||||
|
}
|
||||||
|
|
||||||
|
func controller(args []string, currpath string) {
|
||||||
|
if len(args) == 2 {
|
||||||
|
cname := args[1]
|
||||||
|
generate.GenerateController(cname, currpath)
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Fatal("Wrong number of arguments. Run: bee help generate")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func model(cmd *commands.Command, args []string, currpath string) {
|
||||||
|
if len(args) < 2 {
|
||||||
|
beeLogger.Log.Fatal("Wrong number of arguments. Run: bee help generate")
|
||||||
|
}
|
||||||
|
cmd.Flag.Parse(args[2:])
|
||||||
|
if generate.Fields == "" {
|
||||||
|
beeLogger.Log.Hint("Fields option should not be empty, i.e. -Fields=\"title:string,body:text\"")
|
||||||
|
beeLogger.Log.Fatal("Wrong number of arguments. Run: bee help generate")
|
||||||
|
}
|
||||||
|
sname := args[1]
|
||||||
|
generate.GenerateModel(sname, generate.Fields.String(), currpath)
|
||||||
|
}
|
||||||
|
|
||||||
|
func view(args []string, currpath string) {
|
||||||
|
if len(args) == 2 {
|
||||||
|
cname := args[1]
|
||||||
|
generate.GenerateView(cname, currpath)
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Fatal("Wrong number of arguments. Run: bee help generate")
|
||||||
|
}
|
||||||
|
}
|
164
cmd/commands/hprose/hprose.go
Normal file
164
cmd/commands/hprose/hprose.go
Normal file
|
@ -0,0 +1,164 @@
|
||||||
|
package hprose
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/api"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
"github.com/beego/bee/v2/generate"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
var CmdHproseapp = &commands.Command{
|
||||||
|
// CustomFlags: true,
|
||||||
|
UsageLine: "hprose [appname]",
|
||||||
|
Short: "Creates an RPC application based on Hprose and Beego frameworks",
|
||||||
|
Long: `
|
||||||
|
The command 'hprose' creates an RPC application based on both Beego and Hprose (http://hprose.com/).
|
||||||
|
|
||||||
|
{{"To scaffold out your application, use:"|bold}}
|
||||||
|
|
||||||
|
$ bee hprose [appname] [-tables=""] [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"] [-gopath=false] [-beego=v1.12.3]
|
||||||
|
|
||||||
|
If 'conn' is empty, the command will generate a sample application. Otherwise the command
|
||||||
|
will connect to your database and generate models based on the existing tables.
|
||||||
|
|
||||||
|
The command 'hprose' creates a folder named [appname] with the following structure:
|
||||||
|
|
||||||
|
├── main.go
|
||||||
|
├── go.mod
|
||||||
|
├── {{"conf"|foldername}}
|
||||||
|
│ └── app.conf
|
||||||
|
└── {{"models"|foldername}}
|
||||||
|
└── object.go
|
||||||
|
└── user.go
|
||||||
|
`,
|
||||||
|
PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() },
|
||||||
|
Run: createhprose,
|
||||||
|
}
|
||||||
|
|
||||||
|
var goMod = `
|
||||||
|
module %s
|
||||||
|
|
||||||
|
go %s
|
||||||
|
|
||||||
|
require github.com/beego/beego/v2 %s
|
||||||
|
require github.com/smartystreets/goconvey v1.6.4
|
||||||
|
`
|
||||||
|
|
||||||
|
var gopath utils.DocValue
|
||||||
|
var beegoVersion utils.DocValue
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
CmdHproseapp.Flag.Var(&generate.Tables, "tables", "List of table names separated by a comma.")
|
||||||
|
CmdHproseapp.Flag.Var(&generate.SQLDriver, "driver", "Database driver. Either mysql, postgres or sqlite.")
|
||||||
|
CmdHproseapp.Flag.Var(&generate.SQLConn, "conn", "Connection string used by the driver to connect to a database instance.")
|
||||||
|
CmdHproseapp.Flag.Var(&gopath, "gopath", "Support go path,default false")
|
||||||
|
CmdHproseapp.Flag.Var(&beegoVersion, "beego", "set beego version,only take effect by go mod")
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdHproseapp)
|
||||||
|
}
|
||||||
|
|
||||||
|
func createhprose(cmd *commands.Command, args []string) int {
|
||||||
|
output := cmd.Out()
|
||||||
|
if len(args) == 0 {
|
||||||
|
beeLogger.Log.Fatal("Argument [appname] is missing")
|
||||||
|
}
|
||||||
|
|
||||||
|
curpath, _ := os.Getwd()
|
||||||
|
if len(args) >= 2 {
|
||||||
|
err := cmd.Flag.Parse(args[1:])
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatal("Parse args err " + err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
var apppath string
|
||||||
|
var packpath string
|
||||||
|
var err error
|
||||||
|
if gopath == `true` {
|
||||||
|
beeLogger.Log.Info("Generate api project support GOPATH")
|
||||||
|
version.ShowShortVersionBanner()
|
||||||
|
apppath, packpath, err = utils.CheckEnv(args[0])
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("%s", err)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Info("Generate api project support go modules.")
|
||||||
|
apppath = path.Join(utils.GetBeeWorkPath(), args[0])
|
||||||
|
packpath = args[0]
|
||||||
|
if beegoVersion.String() == `` {
|
||||||
|
beegoVersion.Set(utils.BEEGO_VERSION)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if utils.IsExist(apppath) {
|
||||||
|
beeLogger.Log.Errorf(colors.Bold("Application '%s' already exists"), apppath)
|
||||||
|
beeLogger.Log.Warn(colors.Bold("Do you want to overwrite it? [Yes|No] "))
|
||||||
|
if !utils.AskForConfirmation() {
|
||||||
|
os.Exit(2)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if generate.SQLDriver == "" {
|
||||||
|
generate.SQLDriver = "mysql"
|
||||||
|
}
|
||||||
|
beeLogger.Log.Info("Creating Hprose application...")
|
||||||
|
|
||||||
|
os.MkdirAll(apppath, 0755)
|
||||||
|
if gopath != `true` { //generate first for calc model name
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "go.mod"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(apppath, "go.mod"), fmt.Sprintf(goMod, packpath, utils.GetGoVersionSkipMinor(), beegoVersion.String()))
|
||||||
|
}
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", apppath, "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(apppath, "conf"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "conf"), "\x1b[0m")
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "conf", "app.conf"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(apppath, "conf", "app.conf"),
|
||||||
|
strings.Replace(generate.Hproseconf, "{{.Appname}}", args[0], -1))
|
||||||
|
|
||||||
|
if generate.SQLConn != "" {
|
||||||
|
beeLogger.Log.Infof("Using '%s' as 'driver'", generate.SQLDriver)
|
||||||
|
beeLogger.Log.Infof("Using '%s' as 'conn'", generate.SQLConn)
|
||||||
|
beeLogger.Log.Infof("Using '%s' as 'tables'", generate.Tables)
|
||||||
|
generate.GenerateHproseAppcode(string(generate.SQLDriver), string(generate.SQLConn), "1", string(generate.Tables), path.Join(curpath, args[0]))
|
||||||
|
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "main.go"), "\x1b[0m")
|
||||||
|
maingoContent := strings.Replace(generate.HproseMainconngo, "{{.Appname}}", packpath, -1)
|
||||||
|
maingoContent = strings.Replace(maingoContent, "{{.DriverName}}", string(generate.SQLDriver), -1)
|
||||||
|
maingoContent = strings.Replace(maingoContent, "{{HproseFunctionList}}", strings.Join(generate.HproseAddFunctions, ""), -1)
|
||||||
|
if generate.SQLDriver == "mysql" {
|
||||||
|
maingoContent = strings.Replace(maingoContent, "{{.DriverPkg}}", `_ "github.com/go-sql-driver/mysql"`, -1)
|
||||||
|
} else if generate.SQLDriver == "postgres" {
|
||||||
|
maingoContent = strings.Replace(maingoContent, "{{.DriverPkg}}", `_ "github.com/lib/pq"`, -1)
|
||||||
|
}
|
||||||
|
utils.WriteToFile(path.Join(apppath, "main.go"),
|
||||||
|
strings.Replace(
|
||||||
|
maingoContent,
|
||||||
|
"{{.conn}}",
|
||||||
|
generate.SQLConn.String(),
|
||||||
|
-1,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
os.Mkdir(path.Join(apppath, "models"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models"), "\x1b[0m")
|
||||||
|
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models", "object.go"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(apppath, "models", "object.go"), apiapp.APIModels)
|
||||||
|
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models", "user.go"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(apppath, "models", "user.go"), apiapp.APIModels2)
|
||||||
|
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "main.go"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(apppath, "main.go"),
|
||||||
|
strings.Replace(generate.HproseMaingo, "{{.Appname}}", packpath, -1))
|
||||||
|
}
|
||||||
|
beeLogger.Log.Success("New Hprose application successfully created!")
|
||||||
|
return 0
|
||||||
|
}
|
|
@ -11,8 +11,7 @@
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
|
package migrate
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
|
@ -23,126 +22,119 @@ import (
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
"github.com/beego/bee/v2/config"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
)
|
)
|
||||||
|
|
||||||
var cmdMigrate = &Command{
|
var CmdMigrate = &commands.Command{
|
||||||
UsageLine: "migrate [Command]",
|
UsageLine: "migrate [Command]",
|
||||||
Short: "Runs database migrations",
|
Short: "Runs database migrations",
|
||||||
Long: `The command 'migrate' allows you to run database migrations to keep it up-to-date.
|
Long: `The command 'migrate' allows you to run database migrations to keep it up-to-date.
|
||||||
|
|
||||||
▶ {{"To run all the migrations:"|bold}}
|
▶ {{"To run all the migrations:"|bold}}
|
||||||
|
|
||||||
$ bee migrate [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"]
|
$ bee migrate [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"] [-dir="path/to/migration"]
|
||||||
|
|
||||||
▶ {{"To rollback the last migration:"|bold}}
|
▶ {{"To rollback the last migration:"|bold}}
|
||||||
|
|
||||||
$ bee migrate rollback [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"]
|
$ bee migrate rollback [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"] [-dir="path/to/migration"]
|
||||||
|
|
||||||
▶ {{"To do a reset, which will rollback all the migrations:"|bold}}
|
▶ {{"To do a reset, which will rollback all the migrations:"|bold}}
|
||||||
|
|
||||||
$ bee migrate reset [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"]
|
$ bee migrate reset [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"] [-dir="path/to/migration"]
|
||||||
|
|
||||||
▶ {{"To update your schema:"|bold}}
|
▶ {{"To update your schema:"|bold}}
|
||||||
|
|
||||||
$ bee migrate refresh [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"]
|
$ bee migrate refresh [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"] [-dir="path/to/migration"]
|
||||||
`,
|
`,
|
||||||
PreRun: func(cmd *Command, args []string) { ShowShortVersionBanner() },
|
PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() },
|
||||||
Run: runMigration,
|
Run: RunMigration,
|
||||||
}
|
}
|
||||||
|
|
||||||
var mDriver docValue
|
var mDriver utils.DocValue
|
||||||
var mConn docValue
|
var mConn utils.DocValue
|
||||||
|
var mDir utils.DocValue
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
cmdMigrate.Flag.Var(&mDriver, "driver", "Database driver. Either mysql, postgres or sqlite.")
|
CmdMigrate.Flag.Var(&mDriver, "driver", "Database driver. Either mysql, postgres or sqlite.")
|
||||||
cmdMigrate.Flag.Var(&mConn, "conn", "Connection string used by the driver to connect to a database instance.")
|
CmdMigrate.Flag.Var(&mConn, "conn", "Connection string used by the driver to connect to a database instance.")
|
||||||
|
CmdMigrate.Flag.Var(&mDir, "dir", "The directory where the migration files are stored")
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdMigrate)
|
||||||
}
|
}
|
||||||
|
|
||||||
// runMigration is the entry point for starting a migration
|
// runMigration is the entry point for starting a migration
|
||||||
func runMigration(cmd *Command, args []string) int {
|
func RunMigration(cmd *commands.Command, args []string) int {
|
||||||
currpath, _ := os.Getwd()
|
currpath, _ := os.Getwd()
|
||||||
|
|
||||||
gps := GetGOPATHs()
|
|
||||||
if len(gps) == 0 {
|
|
||||||
logger.Fatal("GOPATH environment variable is not set or empty")
|
|
||||||
}
|
|
||||||
|
|
||||||
gopath := gps[0]
|
|
||||||
|
|
||||||
logger.Debugf("GOPATH: %s", __FILE__(), __LINE__(), gopath)
|
|
||||||
|
|
||||||
// Load the configuration
|
|
||||||
err := loadConfig()
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Failed to load configuration: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Getting command line arguments
|
// Getting command line arguments
|
||||||
if len(args) != 0 {
|
if len(args) != 0 {
|
||||||
cmd.Flag.Parse(args[1:])
|
cmd.Flag.Parse(args[1:])
|
||||||
}
|
}
|
||||||
if mDriver == "" {
|
if mDriver == "" {
|
||||||
mDriver = docValue(conf.Database.Driver)
|
mDriver = utils.DocValue(config.Conf.Database.Driver)
|
||||||
if mDriver == "" {
|
if mDriver == "" {
|
||||||
mDriver = "mysql"
|
mDriver = "mysql"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if mConn == "" {
|
if mConn == "" {
|
||||||
mConn = docValue(conf.Database.Conn)
|
mConn = utils.DocValue(config.Conf.Database.Conn)
|
||||||
if mConn == "" {
|
if mConn == "" {
|
||||||
mConn = "root:@tcp(127.0.0.1:3306)/test"
|
mConn = "root:@tcp(127.0.0.1:3306)/test"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
logger.Infof("Using '%s' as 'driver'", mDriver)
|
if mDir == "" {
|
||||||
logger.Infof("Using '%s' as 'conn'", mConn)
|
mDir = utils.DocValue(config.Conf.Database.Dir)
|
||||||
driverStr, connStr := string(mDriver), string(mConn)
|
if mDir == "" {
|
||||||
|
mDir = utils.DocValue(path.Join(currpath, "database", "migrations"))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
beeLogger.Log.Infof("Using '%s' as 'driver'", mDriver)
|
||||||
|
//Log sensitive connection information only when DEBUG is set to true.
|
||||||
|
beeLogger.Log.Debugf("Conn: %s", utils.FILE(), utils.LINE(), mConn)
|
||||||
|
beeLogger.Log.Infof("Using '%s' as 'dir'", mDir)
|
||||||
|
driverStr, connStr, dirStr := string(mDriver), string(mConn), string(mDir)
|
||||||
|
|
||||||
|
dirRune := []rune(dirStr)
|
||||||
|
|
||||||
|
if dirRune[0] != '/' && dirRune[1] != ':' {
|
||||||
|
dirStr = path.Join(currpath, dirStr)
|
||||||
|
}
|
||||||
|
|
||||||
if len(args) == 0 {
|
if len(args) == 0 {
|
||||||
// run all outstanding migrations
|
// run all outstanding migrations
|
||||||
logger.Info("Running all outstanding migrations")
|
beeLogger.Log.Info("Running all outstanding migrations")
|
||||||
migrateUpdate(currpath, driverStr, connStr)
|
MigrateUpdate(currpath, driverStr, connStr, dirStr)
|
||||||
} else {
|
} else {
|
||||||
mcmd := args[0]
|
mcmd := args[0]
|
||||||
switch mcmd {
|
switch mcmd {
|
||||||
case "rollback":
|
case "rollback":
|
||||||
logger.Info("Rolling back the last migration operation")
|
beeLogger.Log.Info("Rolling back the last migration operation")
|
||||||
migrateRollback(currpath, driverStr, connStr)
|
MigrateRollback(currpath, driverStr, connStr, dirStr)
|
||||||
case "reset":
|
case "reset":
|
||||||
logger.Info("Reseting all migrations")
|
beeLogger.Log.Info("Reseting all migrations")
|
||||||
migrateReset(currpath, driverStr, connStr)
|
MigrateReset(currpath, driverStr, connStr, dirStr)
|
||||||
case "refresh":
|
case "refresh":
|
||||||
logger.Info("Refreshing all migrations")
|
beeLogger.Log.Info("Refreshing all migrations")
|
||||||
migrateRefresh(currpath, driverStr, connStr)
|
MigrateRefresh(currpath, driverStr, connStr, dirStr)
|
||||||
default:
|
default:
|
||||||
logger.Fatal("Command is missing")
|
beeLogger.Log.Fatal("Command is missing")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
logger.Success("Migration successful!")
|
beeLogger.Log.Success("Migration successful!")
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
// migrateUpdate does the schema update
|
|
||||||
func migrateUpdate(currpath, driver, connStr string) {
|
|
||||||
migrate("upgrade", currpath, driver, connStr)
|
|
||||||
}
|
|
||||||
|
|
||||||
// migrateRollback rolls back the latest migration
|
|
||||||
func migrateRollback(currpath, driver, connStr string) {
|
|
||||||
migrate("rollback", currpath, driver, connStr)
|
|
||||||
}
|
|
||||||
|
|
||||||
// migrateReset rolls back all migrations
|
|
||||||
func migrateReset(currpath, driver, connStr string) {
|
|
||||||
migrate("reset", currpath, driver, connStr)
|
|
||||||
}
|
|
||||||
|
|
||||||
// migrationRefresh rolls back all migrations and start over again
|
|
||||||
func migrateRefresh(currpath, driver, connStr string) {
|
|
||||||
migrate("refresh", currpath, driver, connStr)
|
|
||||||
}
|
|
||||||
|
|
||||||
// migrate generates source code, build it, and invoke the binary who does the actual migration
|
// migrate generates source code, build it, and invoke the binary who does the actual migration
|
||||||
func migrate(goal, currpath, driver, connStr string) {
|
func migrate(goal, currpath, driver, connStr, dir string) {
|
||||||
dir := path.Join(currpath, "database", "migrations")
|
if dir == "" {
|
||||||
|
dir = path.Join(currpath, "database", "migrations")
|
||||||
|
}
|
||||||
postfix := ""
|
postfix := ""
|
||||||
if runtime.GOOS == "windows" {
|
if runtime.GOOS == "windows" {
|
||||||
postfix = ".exe"
|
postfix = ".exe"
|
||||||
|
@ -153,7 +145,7 @@ func migrate(goal, currpath, driver, connStr string) {
|
||||||
// Connect to database
|
// Connect to database
|
||||||
db, err := sql.Open(driver, connStr)
|
db, err := sql.Open(driver, connStr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Could not connect to database using '%s': %s", connStr, err)
|
beeLogger.Log.Fatalf("Could not connect to database using '%s': %s", connStr, err)
|
||||||
}
|
}
|
||||||
defer db.Close()
|
defer db.Close()
|
||||||
|
|
||||||
|
@ -171,50 +163,61 @@ func migrate(goal, currpath, driver, connStr string) {
|
||||||
func checkForSchemaUpdateTable(db *sql.DB, driver string) {
|
func checkForSchemaUpdateTable(db *sql.DB, driver string) {
|
||||||
showTableSQL := showMigrationsTableSQL(driver)
|
showTableSQL := showMigrationsTableSQL(driver)
|
||||||
if rows, err := db.Query(showTableSQL); err != nil {
|
if rows, err := db.Query(showTableSQL); err != nil {
|
||||||
logger.Fatalf("Could not show migrations table: %s", err)
|
beeLogger.Log.Fatalf("Could not show migrations table: %s", err)
|
||||||
} else if !rows.Next() {
|
} else if !rows.Next() {
|
||||||
// No migrations table, create new ones
|
// No migrations table, create new ones
|
||||||
createTableSQL := createMigrationsTableSQL(driver)
|
createTableSQL := createMigrationsTableSQL(driver)
|
||||||
|
|
||||||
logger.Infof("Creating 'migrations' table...")
|
beeLogger.Log.Infof("Creating 'migrations' table...")
|
||||||
|
|
||||||
if _, err := db.Query(createTableSQL); err != nil {
|
if _, err := db.Query(createTableSQL); err != nil {
|
||||||
logger.Fatalf("Could not create migrations table: %s", err)
|
beeLogger.Log.Fatalf("Could not create migrations table: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Checking that migrations table schema are expected
|
// Checking that migrations table schema are expected
|
||||||
selectTableSQL := selectMigrationsTableSQL(driver)
|
selectTableSQL := selectMigrationsTableSQL(driver)
|
||||||
if rows, err := db.Query(selectTableSQL); err != nil {
|
if rows, err := db.Query(selectTableSQL); err != nil {
|
||||||
logger.Fatalf("Could not show columns of migrations table: %s", err)
|
beeLogger.Log.Fatalf("Could not show columns of migrations table: %s", err)
|
||||||
} else {
|
} else {
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
var fieldBytes, typeBytes, nullBytes, keyBytes, defaultBytes, extraBytes []byte
|
var fieldBytes, typeBytes, nullBytes, keyBytes, defaultBytes, extraBytes []byte
|
||||||
if err := rows.Scan(&fieldBytes, &typeBytes, &nullBytes, &keyBytes, &defaultBytes, &extraBytes); err != nil {
|
if err := rows.Scan(&fieldBytes, &typeBytes, &nullBytes, &keyBytes, &defaultBytes, &extraBytes); err != nil {
|
||||||
logger.Fatalf("Could not read column information: %s", err)
|
beeLogger.Log.Fatalf("Could not read column information: %s", err)
|
||||||
}
|
}
|
||||||
fieldStr, typeStr, nullStr, keyStr, defaultStr, extraStr :=
|
fieldStr, typeStr, nullStr, keyStr, defaultStr, extraStr :=
|
||||||
string(fieldBytes), string(typeBytes), string(nullBytes), string(keyBytes), string(defaultBytes), string(extraBytes)
|
string(fieldBytes), string(typeBytes), string(nullBytes), string(keyBytes), string(defaultBytes), string(extraBytes)
|
||||||
if fieldStr == "id_migration" {
|
if fieldStr == "id_migration" {
|
||||||
if keyStr != "PRI" || extraStr != "auto_increment" {
|
if keyStr != "PRI" || extraStr != "auto_increment" {
|
||||||
logger.Hint("Expecting KEY: PRI, EXTRA: auto_increment")
|
beeLogger.Log.Hint("Expecting KEY: PRI, EXTRA: auto_increment")
|
||||||
logger.Fatalf("Column migration.id_migration type mismatch: KEY: %s, EXTRA: %s", keyStr, extraStr)
|
beeLogger.Log.Fatalf("Column migration.id_migration type mismatch: KEY: %s, EXTRA: %s", keyStr, extraStr)
|
||||||
}
|
}
|
||||||
} else if fieldStr == "name" {
|
} else if fieldStr == "name" {
|
||||||
if !strings.HasPrefix(typeStr, "varchar") || nullStr != "YES" {
|
if !strings.HasPrefix(typeStr, "varchar") || nullStr != "YES" {
|
||||||
logger.Hint("Expecting TYPE: varchar, NULL: YES")
|
beeLogger.Log.Hint("Expecting TYPE: varchar, NULL: YES")
|
||||||
logger.Fatalf("Column migration.name type mismatch: TYPE: %s, NULL: %s", typeStr, nullStr)
|
beeLogger.Log.Fatalf("Column migration.name type mismatch: TYPE: %s, NULL: %s", typeStr, nullStr)
|
||||||
}
|
}
|
||||||
} else if fieldStr == "created_at" {
|
} else if fieldStr == "created_at" {
|
||||||
if typeStr != "timestamp" || defaultStr != "CURRENT_TIMESTAMP" {
|
if typeStr != "timestamp" || (!strings.EqualFold(defaultStr, "CURRENT_TIMESTAMP") && !strings.EqualFold(defaultStr, "CURRENT_TIMESTAMP()")) {
|
||||||
logger.Hint("Expecting TYPE: timestamp, DEFAULT: CURRENT_TIMESTAMP")
|
beeLogger.Log.Hint("Expecting TYPE: timestamp, DEFAULT: CURRENT_TIMESTAMP || CURRENT_TIMESTAMP()")
|
||||||
logger.Fatalf("Column migration.timestamp type mismatch: TYPE: %s, DEFAULT: %s", typeStr, defaultStr)
|
beeLogger.Log.Fatalf("Column migration.timestamp type mismatch: TYPE: %s, DEFAULT: %s", typeStr, defaultStr)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func driverImportStatement(driver string) string {
|
||||||
|
switch driver {
|
||||||
|
case "mysql":
|
||||||
|
return "github.com/go-sql-driver/mysql"
|
||||||
|
case "postgres":
|
||||||
|
return "github.com/lib/pq"
|
||||||
|
default:
|
||||||
|
return "github.com/go-sql-driver/mysql"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func showMigrationsTableSQL(driver string) string {
|
func showMigrationsTableSQL(driver string) string {
|
||||||
switch driver {
|
switch driver {
|
||||||
case "mysql":
|
case "mysql":
|
||||||
|
@ -252,22 +255,22 @@ func selectMigrationsTableSQL(driver string) string {
|
||||||
func getLatestMigration(db *sql.DB, goal string) (file string, createdAt int64) {
|
func getLatestMigration(db *sql.DB, goal string) (file string, createdAt int64) {
|
||||||
sql := "SELECT name FROM migrations where status = 'update' ORDER BY id_migration DESC LIMIT 1"
|
sql := "SELECT name FROM migrations where status = 'update' ORDER BY id_migration DESC LIMIT 1"
|
||||||
if rows, err := db.Query(sql); err != nil {
|
if rows, err := db.Query(sql); err != nil {
|
||||||
logger.Fatalf("Could not retrieve migrations: %s", err)
|
beeLogger.Log.Fatalf("Could not retrieve migrations: %s", err)
|
||||||
} else {
|
} else {
|
||||||
if rows.Next() {
|
if rows.Next() {
|
||||||
if err := rows.Scan(&file); err != nil {
|
if err := rows.Scan(&file); err != nil {
|
||||||
logger.Fatalf("Could not read migrations in database: %s", err)
|
beeLogger.Log.Fatalf("Could not read migrations in database: %s", err)
|
||||||
}
|
}
|
||||||
createdAtStr := file[len(file)-15:]
|
createdAtStr := file[len(file)-15:]
|
||||||
if t, err := time.Parse("20060102_150405", createdAtStr); err != nil {
|
if t, err := time.Parse("20060102_150405", createdAtStr); err != nil {
|
||||||
logger.Fatalf("Could not parse time: %s", err)
|
beeLogger.Log.Fatalf("Could not parse time: %s", err)
|
||||||
} else {
|
} else {
|
||||||
createdAt = t.Unix()
|
createdAt = t.Unix()
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// migration table has no 'update' record, no point rolling back
|
// migration table has no 'update' record, no point rolling back
|
||||||
if goal == "rollback" {
|
if goal == "rollback" {
|
||||||
logger.Fatal("There is nothing to rollback")
|
beeLogger.Log.Fatal("There is nothing to rollback")
|
||||||
}
|
}
|
||||||
file, createdAt = "", 0
|
file, createdAt = "", 0
|
||||||
}
|
}
|
||||||
|
@ -279,17 +282,18 @@ func getLatestMigration(db *sql.DB, goal string) (file string, createdAt int64)
|
||||||
func writeMigrationSourceFile(dir, source, driver, connStr string, latestTime int64, latestName string, task string) {
|
func writeMigrationSourceFile(dir, source, driver, connStr string, latestTime int64, latestName string, task string) {
|
||||||
changeDir(dir)
|
changeDir(dir)
|
||||||
if f, err := os.OpenFile(source, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err != nil {
|
if f, err := os.OpenFile(source, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err != nil {
|
||||||
logger.Fatalf("Could not create file: %s", err)
|
beeLogger.Log.Fatalf("Could not create file: %s", err)
|
||||||
} else {
|
} else {
|
||||||
content := strings.Replace(MigrationMainTPL, "{{DBDriver}}", driver, -1)
|
content := strings.Replace(MigrationMainTPL, "{{DBDriver}}", driver, -1)
|
||||||
|
content = strings.Replace(content, "{{DriverRepo}}", driverImportStatement(driver), -1)
|
||||||
content = strings.Replace(content, "{{ConnStr}}", connStr, -1)
|
content = strings.Replace(content, "{{ConnStr}}", connStr, -1)
|
||||||
content = strings.Replace(content, "{{LatestTime}}", strconv.FormatInt(latestTime, 10), -1)
|
content = strings.Replace(content, "{{LatestTime}}", strconv.FormatInt(latestTime, 10), -1)
|
||||||
content = strings.Replace(content, "{{LatestName}}", latestName, -1)
|
content = strings.Replace(content, "{{LatestName}}", latestName, -1)
|
||||||
content = strings.Replace(content, "{{Task}}", task, -1)
|
content = strings.Replace(content, "{{Task}}", task, -1)
|
||||||
if _, err := f.WriteString(content); err != nil {
|
if _, err := f.WriteString(content); err != nil {
|
||||||
logger.Fatalf("Could not write to file: %s", err)
|
beeLogger.Log.Fatalf("Could not write to file: %s", err)
|
||||||
}
|
}
|
||||||
CloseFile(f)
|
utils.CloseFile(f)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -298,7 +302,7 @@ func buildMigrationBinary(dir, binary string) {
|
||||||
changeDir(dir)
|
changeDir(dir)
|
||||||
cmd := exec.Command("go", "build", "-o", binary)
|
cmd := exec.Command("go", "build", "-o", binary)
|
||||||
if out, err := cmd.CombinedOutput(); err != nil {
|
if out, err := cmd.CombinedOutput(); err != nil {
|
||||||
logger.Errorf("Could not build migration binary: %s", err)
|
beeLogger.Log.Errorf("Could not build migration binary: %s", err)
|
||||||
formatShellErrOutput(string(out))
|
formatShellErrOutput(string(out))
|
||||||
removeTempFile(dir, binary)
|
removeTempFile(dir, binary)
|
||||||
removeTempFile(dir, binary+".go")
|
removeTempFile(dir, binary+".go")
|
||||||
|
@ -312,7 +316,7 @@ func runMigrationBinary(dir, binary string) {
|
||||||
cmd := exec.Command("./" + binary)
|
cmd := exec.Command("./" + binary)
|
||||||
if out, err := cmd.CombinedOutput(); err != nil {
|
if out, err := cmd.CombinedOutput(); err != nil {
|
||||||
formatShellOutput(string(out))
|
formatShellOutput(string(out))
|
||||||
logger.Errorf("Could not run migration binary: %s", err)
|
beeLogger.Log.Errorf("Could not run migration binary: %s", err)
|
||||||
removeTempFile(dir, binary)
|
removeTempFile(dir, binary)
|
||||||
removeTempFile(dir, binary+".go")
|
removeTempFile(dir, binary+".go")
|
||||||
os.Exit(2)
|
os.Exit(2)
|
||||||
|
@ -325,7 +329,7 @@ func runMigrationBinary(dir, binary string) {
|
||||||
// It exits the system when encouter an error
|
// It exits the system when encouter an error
|
||||||
func changeDir(dir string) {
|
func changeDir(dir string) {
|
||||||
if err := os.Chdir(dir); err != nil {
|
if err := os.Chdir(dir); err != nil {
|
||||||
logger.Fatalf("Could not find migration directory: %s", err)
|
beeLogger.Log.Fatalf("Could not find migration directory: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -333,7 +337,7 @@ func changeDir(dir string) {
|
||||||
func removeTempFile(dir, file string) {
|
func removeTempFile(dir, file string) {
|
||||||
changeDir(dir)
|
changeDir(dir)
|
||||||
if err := os.Remove(file); err != nil {
|
if err := os.Remove(file); err != nil {
|
||||||
logger.Warnf("Could not remove temporary file: %s", err)
|
beeLogger.Log.Warnf("Could not remove temporary file: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -341,7 +345,7 @@ func removeTempFile(dir, file string) {
|
||||||
func formatShellErrOutput(o string) {
|
func formatShellErrOutput(o string) {
|
||||||
for _, line := range strings.Split(o, "\n") {
|
for _, line := range strings.Split(o, "\n") {
|
||||||
if line != "" {
|
if line != "" {
|
||||||
logger.Errorf("|> %s", line)
|
beeLogger.Log.Errorf("|> %s", line)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -350,7 +354,7 @@ func formatShellErrOutput(o string) {
|
||||||
func formatShellOutput(o string) {
|
func formatShellOutput(o string) {
|
||||||
for _, line := range strings.Split(o, "\n") {
|
for _, line := range strings.Split(o, "\n") {
|
||||||
if line != "" {
|
if line != "" {
|
||||||
logger.Infof("|> %s", line)
|
beeLogger.Log.Infof("|> %s", line)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -362,11 +366,10 @@ const (
|
||||||
import(
|
import(
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
"github.com/astaxie/beego/orm"
|
"github.com/beego/beego/v2/client/orm"
|
||||||
"github.com/astaxie/beego/migration"
|
"github.com/beego/beego/v2/client/orm/migration"
|
||||||
|
|
||||||
_ "github.com/go-sql-driver/mysql"
|
_ "{{DriverRepo}}"
|
||||||
_ "github.com/lib/pq"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func init(){
|
func init(){
|
||||||
|
@ -421,3 +424,23 @@ CREATE TABLE migrations (
|
||||||
status migrations_status
|
status migrations_status
|
||||||
)`
|
)`
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// MigrateUpdate does the schema update
|
||||||
|
func MigrateUpdate(currpath, driver, connStr, dir string) {
|
||||||
|
migrate("upgrade", currpath, driver, connStr, dir)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MigrateRollback rolls back the latest migration
|
||||||
|
func MigrateRollback(currpath, driver, connStr, dir string) {
|
||||||
|
migrate("rollback", currpath, driver, connStr, dir)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MigrateReset rolls back all migrations
|
||||||
|
func MigrateReset(currpath, driver, connStr, dir string) {
|
||||||
|
migrate("reset", currpath, driver, connStr, dir)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MigrateRefresh rolls back all migrations and start over again
|
||||||
|
func MigrateRefresh(currpath, driver, connStr, dir string) {
|
||||||
|
migrate("refresh", currpath, driver, connStr, dir)
|
||||||
|
}
|
|
@ -12,24 +12,34 @@
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
|
|
||||||
package main
|
package new
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
path "path/filepath"
|
path "path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
var cmdNew = &Command{
|
var gopath utils.DocValue
|
||||||
UsageLine: "new [appname]",
|
var beegoVersion utils.DocValue
|
||||||
|
|
||||||
|
var CmdNew = &commands.Command{
|
||||||
|
UsageLine: "new [appname] [-gopath=false] [-beego=v2.1.0]",
|
||||||
Short: "Creates a Beego application",
|
Short: "Creates a Beego application",
|
||||||
Long: `
|
Long: `
|
||||||
Creates a Beego application for the given app name in the current directory.
|
Creates a Beego application for the given app name in the current directory.
|
||||||
|
now defaults to generating as a go modules project
|
||||||
The command 'new' creates a folder named [appname] and generates the following structure:
|
The command 'new' creates a folder named [appname] [-gopath=false] [-beego=v1.12.3] and generates the following structure:
|
||||||
|
|
||||||
├── main.go
|
├── main.go
|
||||||
|
├── go.mod
|
||||||
├── {{"conf"|foldername}}
|
├── {{"conf"|foldername}}
|
||||||
│ └── app.conf
|
│ └── app.conf
|
||||||
├── {{"controllers"|foldername}}
|
├── {{"controllers"|foldername}}
|
||||||
|
@ -47,73 +57,8 @@ Creates a Beego application for the given app name in the current directory.
|
||||||
└── index.tpl
|
└── index.tpl
|
||||||
|
|
||||||
`,
|
`,
|
||||||
PreRun: func(cmd *Command, args []string) { ShowShortVersionBanner() },
|
PreRun: nil,
|
||||||
Run: createApp,
|
Run: CreateApp,
|
||||||
}
|
|
||||||
|
|
||||||
func createApp(cmd *Command, args []string) int {
|
|
||||||
output := cmd.Out()
|
|
||||||
if len(args) != 1 {
|
|
||||||
logger.Fatal("Argument [appname] is missing")
|
|
||||||
}
|
|
||||||
|
|
||||||
apppath, packpath, err := checkEnv(args[0])
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("%s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if isExist(apppath) {
|
|
||||||
logger.Errorf(bold("Application '%s' already exists"), apppath)
|
|
||||||
logger.Warn(bold("Do you want to overwrite it? [Yes|No] "))
|
|
||||||
if !askForConfirmation() {
|
|
||||||
os.Exit(2)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
logger.Info("Creating application...")
|
|
||||||
|
|
||||||
os.MkdirAll(apppath, 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", apppath+string(path.Separator), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "conf"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "conf")+string(path.Separator), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "controllers"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "controllers")+string(path.Separator), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "models"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models")+string(path.Separator), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "routers"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "routers")+string(path.Separator), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "tests"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "tests")+string(path.Separator), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "static"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "static")+string(path.Separator), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "static", "js"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "static", "js")+string(path.Separator), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "static", "css"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "static", "css")+string(path.Separator), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "static", "img"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "static", "img")+string(path.Separator), "\x1b[0m")
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "views")+string(path.Separator), "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "views"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "conf", "app.conf"), "\x1b[0m")
|
|
||||||
WriteToFile(path.Join(apppath, "conf", "app.conf"), strings.Replace(appconf, "{{.Appname}}", path.Base(args[0]), -1))
|
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "controllers", "default.go"), "\x1b[0m")
|
|
||||||
WriteToFile(path.Join(apppath, "controllers", "default.go"), controllers)
|
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "views", "index.tpl"), "\x1b[0m")
|
|
||||||
WriteToFile(path.Join(apppath, "views", "index.tpl"), indextpl)
|
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "routers", "router.go"), "\x1b[0m")
|
|
||||||
WriteToFile(path.Join(apppath, "routers", "router.go"), strings.Replace(router, "{{.Appname}}", packpath, -1))
|
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "tests", "default_test.go"), "\x1b[0m")
|
|
||||||
WriteToFile(path.Join(apppath, "tests", "default_test.go"), strings.Replace(test, "{{.Appname}}", packpath, -1))
|
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "main.go"), "\x1b[0m")
|
|
||||||
WriteToFile(path.Join(apppath, "main.go"), strings.Replace(maingo, "{{.Appname}}", packpath, -1))
|
|
||||||
|
|
||||||
logger.Success("New application successfully created!")
|
|
||||||
return 0
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var appconf = `appname = {{.Appname}}
|
var appconf = `appname = {{.Appname}}
|
||||||
|
@ -125,7 +70,7 @@ var maingo = `package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
_ "{{.Appname}}/routers"
|
_ "{{.Appname}}/routers"
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
@ -137,14 +82,20 @@ var router = `package routers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"{{.Appname}}/controllers"
|
"{{.Appname}}/controllers"
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
beego.Router("/", &controllers.MainController{})
|
beego.Router("/", &controllers.MainController{})
|
||||||
}
|
}
|
||||||
`
|
`
|
||||||
|
var goMod = `module %s
|
||||||
|
|
||||||
|
go %s
|
||||||
|
|
||||||
|
require github.com/beego/beego/v2 %s
|
||||||
|
require github.com/smartystreets/goconvey v1.6.4
|
||||||
|
`
|
||||||
var test = `package test
|
var test = `package test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
@ -153,14 +104,17 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
"runtime"
|
"runtime"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
|
||||||
|
"github.com/beego/beego/v2/core/logs"
|
||||||
|
|
||||||
_ "{{.Appname}}/routers"
|
_ "{{.Appname}}/routers"
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
. "github.com/smartystreets/goconvey/convey"
|
. "github.com/smartystreets/goconvey/convey"
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
_, file, _, _ := runtime.Caller(1)
|
_, file, _, _ := runtime.Caller(0)
|
||||||
apppath, _ := filepath.Abs(filepath.Dir(filepath.Join(file, ".." + string(filepath.Separator))))
|
apppath, _ := filepath.Abs(filepath.Dir(filepath.Join(file, ".." + string(filepath.Separator))))
|
||||||
beego.TestBeegoInit(apppath)
|
beego.TestBeegoInit(apppath)
|
||||||
}
|
}
|
||||||
|
@ -172,7 +126,7 @@ func TestBeego(t *testing.T) {
|
||||||
w := httptest.NewRecorder()
|
w := httptest.NewRecorder()
|
||||||
beego.BeeApp.Handlers.ServeHTTP(w, r)
|
beego.BeeApp.Handlers.ServeHTTP(w, r)
|
||||||
|
|
||||||
beego.Trace("testing", "TestBeego", "Code[%d]\n%s", w.Code, w.Body.String())
|
logs.Trace("testing", "TestBeego", "Code[%d]\n%s", w.Code, w.Body.String())
|
||||||
|
|
||||||
Convey("Subject: Test Station Endpoint\n", t, func() {
|
Convey("Subject: Test Station Endpoint\n", t, func() {
|
||||||
Convey("Status Code Should Be 200", func() {
|
Convey("Status Code Should Be 200", func() {
|
||||||
|
@ -189,7 +143,7 @@ func TestBeego(t *testing.T) {
|
||||||
var controllers = `package controllers
|
var controllers = `package controllers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
type MainController struct {
|
type MainController struct {
|
||||||
|
@ -197,7 +151,7 @@ type MainController struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *MainController) Get() {
|
func (c *MainController) Get() {
|
||||||
c.Data["Website"] = "beego.me"
|
c.Data["Website"] = "beego.vip"
|
||||||
c.Data["Email"] = "astaxie@gmail.com"
|
c.Data["Email"] = "astaxie@gmail.com"
|
||||||
c.TplName = "index.tpl"
|
c.TplName = "index.tpl"
|
||||||
}
|
}
|
||||||
|
@ -294,6 +248,112 @@ var indextpl = `<!DOCTYPE html>
|
||||||
</div>
|
</div>
|
||||||
</footer>
|
</footer>
|
||||||
<div class="backdrop"></div>
|
<div class="backdrop"></div>
|
||||||
|
|
||||||
|
<script src="/static/js/reload.min.js"></script>
|
||||||
</body>
|
</body>
|
||||||
</html>
|
</html>
|
||||||
`
|
`
|
||||||
|
|
||||||
|
var reloadJsClient = `function b(a){var c=new WebSocket(a);c.onclose=function(){setTimeout(function(){b(a)},2E3)};c.onmessage=function(){location.reload()}}try{if(window.WebSocket)try{b("ws://localhost:12450/reload")}catch(a){console.error(a)}else console.log("Your browser does not support WebSockets.")}catch(a){console.error("Exception during connecting to Reload:",a)};
|
||||||
|
`
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
CmdNew.Flag.Var(&gopath, "gopath", "Support go path,default false")
|
||||||
|
CmdNew.Flag.Var(&beegoVersion, "beego", "set beego version,only take effect by go mod")
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdNew)
|
||||||
|
}
|
||||||
|
|
||||||
|
func CreateApp(cmd *commands.Command, args []string) int {
|
||||||
|
output := cmd.Out()
|
||||||
|
if len(args) == 0 {
|
||||||
|
beeLogger.Log.Fatal("Argument [appname] is missing")
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(args) >= 2 {
|
||||||
|
err := cmd.Flag.Parse(args[1:])
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatal("Parse args err " + err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
var appPath string
|
||||||
|
var packPath string
|
||||||
|
var err error
|
||||||
|
if gopath == `true` {
|
||||||
|
beeLogger.Log.Info("Generate new project support GOPATH")
|
||||||
|
version.ShowShortVersionBanner()
|
||||||
|
appPath, packPath, err = utils.CheckEnv(args[0])
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("%s", err)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Info("Generate new project support go modules.")
|
||||||
|
appPath = path.Join(utils.GetBeeWorkPath(), args[0])
|
||||||
|
packPath = args[0]
|
||||||
|
if beegoVersion.String() == `` {
|
||||||
|
beegoVersion.Set(utils.BEEGO_VERSION)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if utils.IsExist(appPath) {
|
||||||
|
beeLogger.Log.Errorf(colors.Bold("Application '%s' already exists"), appPath)
|
||||||
|
beeLogger.Log.Warn(colors.Bold("Do you want to overwrite it? [Yes|No] "))
|
||||||
|
if !utils.AskForConfirmation() {
|
||||||
|
os.Exit(2)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
beeLogger.Log.Info("Creating application...")
|
||||||
|
|
||||||
|
// If it is the current directory, select the current folder name to package path
|
||||||
|
if packPath == "." {
|
||||||
|
packPath = path.Base(appPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
os.MkdirAll(appPath, 0755)
|
||||||
|
if gopath != `true` {
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "go.mod"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(appPath, "go.mod"), fmt.Sprintf(goMod, packPath, utils.GetGoVersionSkipMinor(), beegoVersion.String()))
|
||||||
|
}
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", appPath+string(path.Separator), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "conf"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "conf")+string(path.Separator), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "controllers"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "controllers")+string(path.Separator), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "models"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "models")+string(path.Separator), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "routers"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "routers")+string(path.Separator), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "tests"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "tests")+string(path.Separator), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "static"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "static")+string(path.Separator), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "static", "js"), 0755)
|
||||||
|
utils.WriteToFile(path.Join(appPath, "static", "js", "reload.min.js"), reloadJsClient)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "static", "js")+string(path.Separator), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "static", "css"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "static", "css")+string(path.Separator), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "static", "img"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "static", "img")+string(path.Separator), "\x1b[0m")
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "views")+string(path.Separator), "\x1b[0m")
|
||||||
|
os.Mkdir(path.Join(appPath, "views"), 0755)
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "conf", "app.conf"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(appPath, "conf", "app.conf"), strings.Replace(appconf, "{{.Appname}}", path.Base(args[0]), -1))
|
||||||
|
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "controllers", "default.go"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(appPath, "controllers", "default.go"), controllers)
|
||||||
|
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "views", "index.tpl"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(appPath, "views", "index.tpl"), indextpl)
|
||||||
|
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "routers", "router.go"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(appPath, "routers", "router.go"), strings.Replace(router, "{{.Appname}}", packPath, -1))
|
||||||
|
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "tests", "default_test.go"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(appPath, "tests", "default_test.go"), strings.Replace(test, "{{.Appname}}", packPath, -1))
|
||||||
|
|
||||||
|
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(appPath, "main.go"), "\x1b[0m")
|
||||||
|
utils.WriteToFile(path.Join(appPath, "main.go"), strings.Replace(maingo, "{{.Appname}}", packPath, -1))
|
||||||
|
|
||||||
|
beeLogger.Log.Success("New application successfully created!")
|
||||||
|
return 0
|
||||||
|
}
|
|
@ -1,18 +1,4 @@
|
||||||
// Copyright 2013 bee authors
|
package pack
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"archive/tar"
|
"archive/tar"
|
||||||
|
@ -31,9 +17,14 @@ import (
|
||||||
"strings"
|
"strings"
|
||||||
"syscall"
|
"syscall"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
var cmdPack = &Command{
|
var CmdPack = &commands.Command{
|
||||||
CustomFlags: true,
|
CustomFlags: true,
|
||||||
UsageLine: "pack",
|
UsageLine: "pack",
|
||||||
Short: "Compresses a Beego application into a single file",
|
Short: "Compresses a Beego application into a single file",
|
||||||
|
@ -43,40 +34,31 @@ var cmdPack = &Command{
|
||||||
{{"Example:"|bold}}
|
{{"Example:"|bold}}
|
||||||
$ bee pack -v -ba="-ldflags '-s -w'"
|
$ bee pack -v -ba="-ldflags '-s -w'"
|
||||||
`,
|
`,
|
||||||
PreRun: func(cmd *Command, args []string) { ShowShortVersionBanner() },
|
PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() },
|
||||||
Run: packApp,
|
Run: packApp,
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
var (
|
||||||
appPath string
|
appPath string
|
||||||
|
appName string
|
||||||
excludeP string
|
excludeP string
|
||||||
excludeS string
|
excludeS string
|
||||||
outputP string
|
outputP string
|
||||||
excludeR ListOpts
|
excludeR utils.ListOpts
|
||||||
fsym bool
|
fsym bool
|
||||||
ssym bool
|
ssym bool
|
||||||
build bool
|
build bool
|
||||||
buildArgs string
|
buildArgs string
|
||||||
buildEnvs ListOpts
|
buildEnvs utils.ListOpts
|
||||||
verbose bool
|
verbose bool
|
||||||
format string
|
format string
|
||||||
)
|
)
|
||||||
|
|
||||||
type ListOpts []string
|
|
||||||
|
|
||||||
func (opts *ListOpts) String() string {
|
|
||||||
return fmt.Sprint(*opts)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (opts *ListOpts) Set(value string) error {
|
|
||||||
*opts = append(*opts, value)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
fs := flag.NewFlagSet("pack", flag.ContinueOnError)
|
fs := flag.NewFlagSet("pack", flag.ContinueOnError)
|
||||||
fs.StringVar(&appPath, "p", "", "Set the application path. Defaults to the current path.")
|
fs.StringVar(&appPath, "p", "", "Set the application path. Defaults to the current path.")
|
||||||
fs.BoolVar(&build, "b", true, "Tell the command to do a build for the current platform. Defaults to true.")
|
fs.BoolVar(&build, "b", true, "Tell the command to do a build for the current platform. Defaults to true.")
|
||||||
|
fs.StringVar(&appName, "a", "", "Set the application name. Defaults to the dir name.")
|
||||||
fs.StringVar(&buildArgs, "ba", "", "Specify additional args for Go build.")
|
fs.StringVar(&buildArgs, "ba", "", "Specify additional args for Go build.")
|
||||||
fs.Var(&buildEnvs, "be", "Specify additional env variables for Go build. e.g. GOARCH=arm.")
|
fs.Var(&buildEnvs, "be", "Specify additional env variables for Go build. e.g. GOARCH=arm.")
|
||||||
fs.StringVar(&outputP, "o", "", "Set the compressed file output path. Defaults to the current path.")
|
fs.StringVar(&outputP, "o", "", "Set the compressed file output path. Defaults to the current path.")
|
||||||
|
@ -87,7 +69,8 @@ func init() {
|
||||||
fs.BoolVar(&fsym, "fs", false, "Tell the command to follow symlinks. Defaults to false.")
|
fs.BoolVar(&fsym, "fs", false, "Tell the command to follow symlinks. Defaults to false.")
|
||||||
fs.BoolVar(&ssym, "ss", false, "Tell the command to skip symlinks. Defaults to false.")
|
fs.BoolVar(&ssym, "ss", false, "Tell the command to skip symlinks. Defaults to false.")
|
||||||
fs.BoolVar(&verbose, "v", false, "Be more verbose during the operation. Defaults to false.")
|
fs.BoolVar(&verbose, "v", false, "Be more verbose during the operation. Defaults to false.")
|
||||||
cmdPack.Flag = *fs
|
CmdPack.Flag = *fs
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdPack)
|
||||||
}
|
}
|
||||||
|
|
||||||
type walker interface {
|
type walker interface {
|
||||||
|
@ -115,10 +98,6 @@ type walkFileTree struct {
|
||||||
output *io.Writer
|
output *io.Writer
|
||||||
}
|
}
|
||||||
|
|
||||||
func (wft *walkFileTree) setPrefix(prefix string) {
|
|
||||||
wft.prefix = prefix
|
|
||||||
}
|
|
||||||
|
|
||||||
func (wft *walkFileTree) isExclude(fPath string) bool {
|
func (wft *walkFileTree) isExclude(fPath string) bool {
|
||||||
if fPath == "" {
|
if fPath == "" {
|
||||||
return true
|
return true
|
||||||
|
@ -316,12 +295,12 @@ func (wft *tarWalk) compress(name, fpath string, fi os.FileInfo) (bool, error) {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if isSym == false {
|
if !isSym {
|
||||||
fr, err := os.Open(fpath)
|
fr, err := os.Open(fpath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
defer CloseFile(fr)
|
defer utils.CloseFile(fr)
|
||||||
_, err = io.Copy(tw, fr)
|
_, err = io.Copy(tw, fr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
|
@ -352,12 +331,12 @@ func (wft *zipWalk) compress(name, fpath string, fi os.FileInfo) (bool, error) {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if isSym == false {
|
if !isSym {
|
||||||
fr, err := os.Open(fpath)
|
fr, err := os.Open(fpath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
defer CloseFile(fr)
|
defer utils.CloseFile(fr)
|
||||||
_, err = io.Copy(w, fr)
|
_, err = io.Copy(w, fr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
|
@ -379,10 +358,10 @@ func (wft *zipWalk) compress(name, fpath string, fi os.FileInfo) (bool, error) {
|
||||||
func packDirectory(output io.Writer, excludePrefix []string, excludeSuffix []string,
|
func packDirectory(output io.Writer, excludePrefix []string, excludeSuffix []string,
|
||||||
excludeRegexp []*regexp.Regexp, includePath ...string) (err error) {
|
excludeRegexp []*regexp.Regexp, includePath ...string) (err error) {
|
||||||
|
|
||||||
logger.Infof("Excluding relpath prefix: %s", strings.Join(excludePrefix, ":"))
|
beeLogger.Log.Infof("Excluding relpath prefix: %s", strings.Join(excludePrefix, ":"))
|
||||||
logger.Infof("Excluding relpath suffix: %s", strings.Join(excludeSuffix, ":"))
|
beeLogger.Log.Infof("Excluding relpath suffix: %s", strings.Join(excludeSuffix, ":"))
|
||||||
if len(excludeRegexp) > 0 {
|
if len(excludeRegexp) > 0 {
|
||||||
logger.Infof("Excluding filename regex: `%s`", strings.Join(excludeR, "`, `"))
|
beeLogger.Log.Infof("Excluding filename regex: `%s`", strings.Join(excludeR, "`, `"))
|
||||||
}
|
}
|
||||||
|
|
||||||
w, err := os.OpenFile(outputP, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
|
w, err := os.OpenFile(outputP, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
|
||||||
|
@ -437,10 +416,10 @@ func packDirectory(output io.Writer, excludePrefix []string, excludeSuffix []str
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func packApp(cmd *Command, args []string) int {
|
func packApp(cmd *commands.Command, args []string) int {
|
||||||
output := cmd.Out()
|
output := cmd.Out()
|
||||||
curPath, _ := os.Getwd()
|
curPath, _ := os.Getwd()
|
||||||
thePath := ""
|
var thePath string
|
||||||
|
|
||||||
nArgs := []string{}
|
nArgs := []string{}
|
||||||
has := false
|
has := false
|
||||||
|
@ -454,21 +433,23 @@ func packApp(cmd *Command, args []string) int {
|
||||||
}
|
}
|
||||||
cmd.Flag.Parse(nArgs)
|
cmd.Flag.Parse(nArgs)
|
||||||
|
|
||||||
if path.IsAbs(appPath) == false {
|
if !path.IsAbs(appPath) {
|
||||||
appPath = path.Join(curPath, appPath)
|
appPath = path.Join(curPath, appPath)
|
||||||
}
|
}
|
||||||
|
|
||||||
thePath, err := path.Abs(appPath)
|
thePath, err := path.Abs(appPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Wrong application path: %s", thePath)
|
beeLogger.Log.Fatalf("Wrong application path: %s", thePath)
|
||||||
}
|
}
|
||||||
if stat, err := os.Stat(thePath); os.IsNotExist(err) || stat.IsDir() == false {
|
if stat, err := os.Stat(thePath); os.IsNotExist(err) || !stat.IsDir() {
|
||||||
logger.Fatalf("Application path does not exist: %s", thePath)
|
beeLogger.Log.Fatalf("Application path does not exist: %s", thePath)
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Infof("Packaging application on '%s'...", thePath)
|
beeLogger.Log.Infof("Packaging application on '%s'...", thePath)
|
||||||
|
|
||||||
appName := path.Base(thePath)
|
if len(appName) == 0 {
|
||||||
|
appName = path.Base(thePath)
|
||||||
|
}
|
||||||
|
|
||||||
goos := runtime.GOOS
|
goos := runtime.GOOS
|
||||||
if v, found := syscall.Getenv("GOOS"); found {
|
if v, found := syscall.Getenv("GOOS"); found {
|
||||||
|
@ -488,12 +469,12 @@ func packApp(cmd *Command, args []string) int {
|
||||||
// Remove the tmpdir once bee pack is done
|
// Remove the tmpdir once bee pack is done
|
||||||
err := os.RemoveAll(tmpdir)
|
err := os.RemoveAll(tmpdir)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Error("Failed to remove the generated temp dir")
|
beeLogger.Log.Error("Failed to remove the generated temp dir")
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
if build {
|
if build {
|
||||||
logger.Info("Building application...")
|
beeLogger.Log.Infof("Building application (%v)...", appName)
|
||||||
var envs []string
|
var envs []string
|
||||||
for _, env := range buildEnvs {
|
for _, env := range buildEnvs {
|
||||||
parts := strings.SplitN(env, "=", 2)
|
parts := strings.SplitN(env, "=", 2)
|
||||||
|
@ -515,7 +496,7 @@ func packApp(cmd *Command, args []string) int {
|
||||||
os.Setenv("GOOS", goos)
|
os.Setenv("GOOS", goos)
|
||||||
os.Setenv("GOARCH", goarch)
|
os.Setenv("GOARCH", goarch)
|
||||||
|
|
||||||
logger.Infof("Using: GOOS=%s GOARCH=%s", goos, goarch)
|
beeLogger.Log.Infof("Using: GOOS=%s GOARCH=%s", goos, goarch)
|
||||||
|
|
||||||
binPath := path.Join(tmpdir, appName)
|
binPath := path.Join(tmpdir, appName)
|
||||||
if goos == "windows" {
|
if goos == "windows" {
|
||||||
|
@ -538,10 +519,10 @@ func packApp(cmd *Command, args []string) int {
|
||||||
execmd.Dir = thePath
|
execmd.Dir = thePath
|
||||||
err = execmd.Run()
|
err = execmd.Run()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatal(err.Error())
|
beeLogger.Log.Fatal(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Success("Build Successful!")
|
beeLogger.Log.Success("Build Successful!")
|
||||||
}
|
}
|
||||||
|
|
||||||
switch format {
|
switch format {
|
||||||
|
@ -552,14 +533,14 @@ func packApp(cmd *Command, args []string) int {
|
||||||
|
|
||||||
outputN := appName + "." + format
|
outputN := appName + "." + format
|
||||||
|
|
||||||
if outputP == "" || path.IsAbs(outputP) == false {
|
if outputP == "" || !path.IsAbs(outputP) {
|
||||||
outputP = path.Join(curPath, outputP)
|
outputP = path.Join(curPath, outputP)
|
||||||
}
|
}
|
||||||
|
|
||||||
if _, err := os.Stat(outputP); err != nil {
|
if _, err := os.Stat(outputP); err != nil {
|
||||||
err = os.MkdirAll(outputP, 0755)
|
err = os.MkdirAll(outputP, 0755)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatal(err.Error())
|
beeLogger.Log.Fatal(err.Error())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -576,25 +557,26 @@ func packApp(cmd *Command, args []string) int {
|
||||||
exs = append(exs, p)
|
exs = append(exs, p)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
exs = append(exs, `go.mod`)
|
||||||
|
exs = append(exs, `go.sum`)
|
||||||
var exr []*regexp.Regexp
|
var exr []*regexp.Regexp
|
||||||
for _, r := range excludeR {
|
for _, r := range excludeR {
|
||||||
if len(r) > 0 {
|
if len(r) > 0 {
|
||||||
if re, err := regexp.Compile(r); err != nil {
|
if re, err := regexp.Compile(r); err != nil {
|
||||||
logger.Fatal(err.Error())
|
beeLogger.Log.Fatal(err.Error())
|
||||||
} else {
|
} else {
|
||||||
exr = append(exr, re)
|
exr = append(exr, re)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Infof("Writing to output: %s", outputP)
|
beeLogger.Log.Infof("Writing to output: %s", outputP)
|
||||||
|
|
||||||
err = packDirectory(output, exp, exs, exr, tmpdir, thePath)
|
err = packDirectory(output, exp, exs, exr, tmpdir, thePath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatal(err.Error())
|
beeLogger.Log.Fatal(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Success("Application packed!")
|
beeLogger.Log.Success("Application packed!")
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
102
cmd/commands/rs/rs.go
Normal file
102
cmd/commands/rs/rs.go
Normal file
|
@ -0,0 +1,102 @@
|
||||||
|
// Copyright 2017 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
|
||||||
|
// Package rs ...
|
||||||
|
package rs
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"runtime"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
"github.com/beego/bee/v2/config"
|
||||||
|
"github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
var cmdRs = &commands.Command{
|
||||||
|
UsageLine: "rs",
|
||||||
|
Short: "Run customized scripts",
|
||||||
|
Long: `Run script allows you to run arbitrary commands using Bee.
|
||||||
|
Custom commands are provided from the "scripts" object inside bee.json or Beefile.
|
||||||
|
|
||||||
|
To run a custom command, use: {{"$ bee rs mycmd ARGS" | bold}}
|
||||||
|
{{if len .}}
|
||||||
|
{{"AVAILABLE SCRIPTS"|headline}}{{range $cmdName, $cmd := .}}
|
||||||
|
{{$cmdName | bold}}
|
||||||
|
{{$cmd}}{{end}}{{end}}
|
||||||
|
`,
|
||||||
|
PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() },
|
||||||
|
Run: runScript,
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
config.LoadConfig()
|
||||||
|
cmdRs.Long = utils.TmplToString(cmdRs.Long, config.Conf.Scripts)
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, cmdRs)
|
||||||
|
}
|
||||||
|
|
||||||
|
func runScript(cmd *commands.Command, args []string) int {
|
||||||
|
if len(args) == 0 {
|
||||||
|
cmd.Usage()
|
||||||
|
}
|
||||||
|
|
||||||
|
start := time.Now()
|
||||||
|
script, args := args[0], args[1:]
|
||||||
|
|
||||||
|
if c, exist := config.Conf.Scripts[script]; exist {
|
||||||
|
command := customCommand{
|
||||||
|
Name: script,
|
||||||
|
Command: c,
|
||||||
|
Args: args,
|
||||||
|
}
|
||||||
|
if err := command.run(); err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Errorf("Command '%s' not found in Beefile/bee.json", script)
|
||||||
|
}
|
||||||
|
elapsed := time.Since(start)
|
||||||
|
fmt.Println(colors.GreenBold(fmt.Sprintf("Finished in %s.", elapsed)))
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
type customCommand struct {
|
||||||
|
Name string
|
||||||
|
Command string
|
||||||
|
Args []string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *customCommand) run() error {
|
||||||
|
beeLogger.Log.Info(colors.GreenBold(fmt.Sprintf("Running '%s'...", c.Name)))
|
||||||
|
var cmd *exec.Cmd
|
||||||
|
switch runtime.GOOS {
|
||||||
|
case "darwin", "linux":
|
||||||
|
args := append([]string{c.Command}, c.Args...)
|
||||||
|
cmd = exec.Command("sh", "-c", strings.Join(args, " "))
|
||||||
|
case "windows":
|
||||||
|
args := append([]string{c.Command}, c.Args...)
|
||||||
|
cmd = exec.Command("cmd", "/C", strings.Join(args, " "))
|
||||||
|
}
|
||||||
|
cmd.Stdout = os.Stdout
|
||||||
|
cmd.Stderr = os.Stderr
|
||||||
|
return cmd.Run()
|
||||||
|
}
|
88
cmd/commands/run/docs.go
Normal file
88
cmd/commands/run/docs.go
Normal file
|
@ -0,0 +1,88 @@
|
||||||
|
package run
|
||||||
|
|
||||||
|
import (
|
||||||
|
"archive/zip"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
swaggerVersion = "4.6.1"
|
||||||
|
swaggerlink = "https://codeload.github.com/beego/swagger/zip/refs/tags/v" + swaggerVersion
|
||||||
|
)
|
||||||
|
|
||||||
|
func downloadFromURL(url, fileName string) {
|
||||||
|
var down bool
|
||||||
|
if fd, err := os.Stat(fileName); err != nil && os.IsNotExist(err) {
|
||||||
|
down = true
|
||||||
|
} else if fd.Size() == int64(0) {
|
||||||
|
down = true
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Infof("'%s' already exists", fileName)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if down {
|
||||||
|
beeLogger.Log.Infof("Downloading '%s' to '%s'...", url, fileName)
|
||||||
|
output, err := os.Create(fileName)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Error while creating '%s': %s", fileName, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer output.Close()
|
||||||
|
|
||||||
|
response, err := http.Get(url)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Error while downloading '%s': %s", url, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer response.Body.Close()
|
||||||
|
|
||||||
|
n, err := io.Copy(output, response.Body)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Error while downloading '%s': %s", url, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
beeLogger.Log.Successf("%d bytes downloaded!", n)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func unzipAndDelete(src string) error {
|
||||||
|
beeLogger.Log.Infof("Unzipping '%s'...", src)
|
||||||
|
r, err := zip.OpenReader(src)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer r.Close()
|
||||||
|
|
||||||
|
rp := strings.NewReplacer("swagger-"+swaggerVersion, "swagger")
|
||||||
|
for _, f := range r.File {
|
||||||
|
rc, err := f.Open()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer rc.Close()
|
||||||
|
|
||||||
|
fname := rp.Replace(f.Name)
|
||||||
|
if f.FileInfo().IsDir() {
|
||||||
|
os.MkdirAll(fname, f.Mode())
|
||||||
|
} else {
|
||||||
|
f, err := os.OpenFile(
|
||||||
|
fname, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
_, err = io.Copy(f, rc)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
beeLogger.Log.Successf("Done! Deleting '%s'...", src)
|
||||||
|
return os.RemoveAll(src)
|
||||||
|
}
|
192
cmd/commands/run/reload.go
Normal file
192
cmd/commands/run/reload.go
Normal file
|
@ -0,0 +1,192 @@
|
||||||
|
// Copyright 2017 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
package run
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/gorilla/websocket"
|
||||||
|
)
|
||||||
|
|
||||||
|
// wsBroker maintains the set of active clients and broadcasts messages to the clients.
|
||||||
|
type wsBroker struct {
|
||||||
|
clients map[*wsClient]bool // Registered clients.
|
||||||
|
broadcast chan []byte // Inbound messages from the clients.
|
||||||
|
register chan *wsClient // Register requests from the clients.
|
||||||
|
unregister chan *wsClient // Unregister requests from clients.
|
||||||
|
}
|
||||||
|
|
||||||
|
func (br *wsBroker) run() {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case client := <-br.register:
|
||||||
|
br.clients[client] = true
|
||||||
|
case client := <-br.unregister:
|
||||||
|
if _, ok := br.clients[client]; ok {
|
||||||
|
delete(br.clients, client)
|
||||||
|
close(client.send)
|
||||||
|
}
|
||||||
|
case message := <-br.broadcast:
|
||||||
|
for client := range br.clients {
|
||||||
|
select {
|
||||||
|
case client.send <- message:
|
||||||
|
default:
|
||||||
|
close(client.send)
|
||||||
|
delete(br.clients, client)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// wsClient represents the end-client.
|
||||||
|
type wsClient struct {
|
||||||
|
broker *wsBroker // The broker.
|
||||||
|
conn *websocket.Conn // The websocket connection.
|
||||||
|
send chan []byte // Buffered channel of outbound messages.
|
||||||
|
}
|
||||||
|
|
||||||
|
// readPump pumps messages from the websocket connection to the broker.
|
||||||
|
func (c *wsClient) readPump() {
|
||||||
|
defer func() {
|
||||||
|
c.broker.unregister <- c
|
||||||
|
c.conn.Close()
|
||||||
|
}()
|
||||||
|
|
||||||
|
for {
|
||||||
|
_, _, err := c.conn.ReadMessage()
|
||||||
|
if err != nil {
|
||||||
|
if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) {
|
||||||
|
beeLogger.Log.Errorf("An error happened when reading from the Websocket client: %v", err)
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// write writes a message with the given message type and payload.
|
||||||
|
func (c *wsClient) write(mt int, payload []byte) error {
|
||||||
|
c.conn.SetWriteDeadline(time.Now().Add(writeWait))
|
||||||
|
return c.conn.WriteMessage(mt, payload)
|
||||||
|
}
|
||||||
|
|
||||||
|
// writePump pumps messages from the broker to the websocket connection.
|
||||||
|
func (c *wsClient) writePump() {
|
||||||
|
ticker := time.NewTicker(pingPeriod)
|
||||||
|
defer func() {
|
||||||
|
ticker.Stop()
|
||||||
|
c.conn.Close()
|
||||||
|
}()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case message, ok := <-c.send:
|
||||||
|
if !ok {
|
||||||
|
// The broker closed the channel.
|
||||||
|
c.write(websocket.CloseMessage, []byte{})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
c.conn.SetWriteDeadline(time.Now().Add(writeWait))
|
||||||
|
w, err := c.conn.NextWriter(websocket.TextMessage)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
w.Write(message)
|
||||||
|
|
||||||
|
n := len(c.send)
|
||||||
|
for i := 0; i < n; i++ {
|
||||||
|
w.Write([]byte("/n"))
|
||||||
|
w.Write(<-c.send)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := w.Close(); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
case <-ticker.C:
|
||||||
|
if err := c.write(websocket.PingMessage, []byte{}); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
broker *wsBroker // The broker.
|
||||||
|
reloadAddress = ":12450" // The port on which the reload server will listen to.
|
||||||
|
|
||||||
|
upgrader = websocket.Upgrader{
|
||||||
|
ReadBufferSize: 1024,
|
||||||
|
WriteBufferSize: 1024,
|
||||||
|
CheckOrigin: func(r *http.Request) bool { return true },
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
writeWait = 10 * time.Second // Time allowed to write a message to the peer.
|
||||||
|
pongWait = 60 * time.Second // Time allowed to read the next pong message from the peer.
|
||||||
|
pingPeriod = (pongWait * 9) / 10 // Send pings to peer with this period. Must be less than pongWait.
|
||||||
|
)
|
||||||
|
|
||||||
|
func startReloadServer() {
|
||||||
|
broker = &wsBroker{
|
||||||
|
broadcast: make(chan []byte),
|
||||||
|
register: make(chan *wsClient),
|
||||||
|
unregister: make(chan *wsClient),
|
||||||
|
clients: make(map[*wsClient]bool),
|
||||||
|
}
|
||||||
|
|
||||||
|
go broker.run()
|
||||||
|
http.HandleFunc("/reload", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
handleWsRequest(broker, w, r)
|
||||||
|
})
|
||||||
|
|
||||||
|
go startServer()
|
||||||
|
beeLogger.Log.Infof("Reload server listening at %s", reloadAddress)
|
||||||
|
}
|
||||||
|
|
||||||
|
func startServer() {
|
||||||
|
err := http.ListenAndServe(reloadAddress, nil)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Failed to start up the Reload server: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func sendReload(payload string) {
|
||||||
|
message := bytes.TrimSpace([]byte(payload))
|
||||||
|
broker.broadcast <- message
|
||||||
|
}
|
||||||
|
|
||||||
|
// handleWsRequest handles websocket requests from the peer.
|
||||||
|
func handleWsRequest(broker *wsBroker, w http.ResponseWriter, r *http.Request) {
|
||||||
|
conn, err := upgrader.Upgrade(w, r, nil)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("error while upgrading server connection: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
client := &wsClient{
|
||||||
|
broker: broker,
|
||||||
|
conn: conn,
|
||||||
|
send: make(chan []byte, 256),
|
||||||
|
}
|
||||||
|
client.broker.register <- client
|
||||||
|
|
||||||
|
go client.writePump()
|
||||||
|
client.readPump()
|
||||||
|
}
|
256
cmd/commands/run/run.go
Normal file
256
cmd/commands/run/run.go
Normal file
|
@ -0,0 +1,256 @@
|
||||||
|
// Copyright 2013 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
package run
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
path "path/filepath"
|
||||||
|
"runtime"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
"github.com/beego/bee/v2/config"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
var CmdRun = &commands.Command{
|
||||||
|
UsageLine: "run [appname] [watchall] [-main=*.go] [-downdoc=true] [-gendoc=true] [-vendor=true] [-e=folderToExclude] [-ex=extraPackageToWatch] [-tags=goBuildTags] [-runmode=BEEGO_RUNMODE]",
|
||||||
|
Short: "Run the application by starting a local development server",
|
||||||
|
Long: `
|
||||||
|
Run command will supervise the filesystem of the application for any changes, and recompile/restart it.
|
||||||
|
|
||||||
|
`,
|
||||||
|
PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() },
|
||||||
|
Run: RunApp,
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
mainFiles utils.ListOpts
|
||||||
|
downdoc utils.DocValue
|
||||||
|
gendoc utils.DocValue
|
||||||
|
// The flags list of the paths excluded from watching
|
||||||
|
excludedPaths utils.StrFlags
|
||||||
|
// Pass through to -tags arg of "go build"
|
||||||
|
buildTags string
|
||||||
|
// Pass through to -ldflags arg of "go build"
|
||||||
|
buildLDFlags string
|
||||||
|
// Application path
|
||||||
|
currpath string
|
||||||
|
// Application name
|
||||||
|
appname string
|
||||||
|
// Channel to signal an Exit
|
||||||
|
exit chan bool
|
||||||
|
// Flag to watch the vendor folder
|
||||||
|
vendorWatch bool
|
||||||
|
// Current user workspace
|
||||||
|
currentGoPath string
|
||||||
|
// Current runmode
|
||||||
|
runmode string
|
||||||
|
// Extra args to run application
|
||||||
|
runargs string
|
||||||
|
// Extra directories
|
||||||
|
extraPackages utils.StrFlags
|
||||||
|
)
|
||||||
|
var started = make(chan bool)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
CmdRun.Flag.Var(&mainFiles, "main", "Specify main go files.")
|
||||||
|
CmdRun.Flag.Var(&gendoc, "gendoc", "Enable auto-generate the docs.")
|
||||||
|
CmdRun.Flag.Var(&downdoc, "downdoc", "Enable auto-download of the swagger file if it does not exist.")
|
||||||
|
CmdRun.Flag.Var(&excludedPaths, "e", "List of paths to exclude.")
|
||||||
|
CmdRun.Flag.BoolVar(&vendorWatch, "vendor", false, "Enable watch vendor folder.")
|
||||||
|
CmdRun.Flag.StringVar(&buildTags, "tags", "", "Set the build tags. See: https://golang.org/pkg/go/build/")
|
||||||
|
CmdRun.Flag.StringVar(&buildLDFlags, "ldflags", "", "Set the build ldflags. See: https://golang.org/pkg/go/build/")
|
||||||
|
CmdRun.Flag.StringVar(&runmode, "runmode", "", "Set the Beego run mode.")
|
||||||
|
CmdRun.Flag.StringVar(&runargs, "runargs", "", "Extra args to run application")
|
||||||
|
CmdRun.Flag.Var(&extraPackages, "ex", "List of extra package to watch.")
|
||||||
|
exit = make(chan bool)
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdRun)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RunApp locates files to watch, and starts the beego application
|
||||||
|
func RunApp(cmd *commands.Command, args []string) int {
|
||||||
|
// The default app path is the current working directory
|
||||||
|
appPath, _ := os.Getwd()
|
||||||
|
|
||||||
|
// If an argument is presented, we use it as the app path
|
||||||
|
if len(args) != 0 && args[0] != "watchall" {
|
||||||
|
if path.IsAbs(args[0]) {
|
||||||
|
appPath = args[0]
|
||||||
|
} else {
|
||||||
|
appPath = path.Join(appPath, args[0])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if utils.IsInGOPATH(appPath) {
|
||||||
|
if found, _gopath, _path := utils.SearchGOPATHs(appPath); found {
|
||||||
|
appPath = _path
|
||||||
|
appname = path.Base(appPath)
|
||||||
|
currentGoPath = _gopath
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Fatalf("No application '%s' found in your GOPATH", appPath)
|
||||||
|
}
|
||||||
|
if strings.HasSuffix(appname, ".go") && utils.IsExist(appPath) {
|
||||||
|
beeLogger.Log.Warnf("The appname is in conflict with file's current path. Do you want to build appname as '%s'", appname)
|
||||||
|
beeLogger.Log.Info("Do you want to overwrite it? [yes|no] ")
|
||||||
|
if !utils.AskForConfirmation() {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Warn("Running application outside of GOPATH")
|
||||||
|
appname = path.Base(appPath)
|
||||||
|
currentGoPath = appPath
|
||||||
|
}
|
||||||
|
|
||||||
|
beeLogger.Log.Infof("Using '%s' as 'appname'", appname)
|
||||||
|
|
||||||
|
beeLogger.Log.Debugf("Current path: %s", utils.FILE(), utils.LINE(), appPath)
|
||||||
|
|
||||||
|
if runmode == "prod" || runmode == "dev" {
|
||||||
|
os.Setenv("BEEGO_RUNMODE", runmode)
|
||||||
|
beeLogger.Log.Infof("Using '%s' as 'runmode'", os.Getenv("BEEGO_RUNMODE"))
|
||||||
|
} else if runmode != "" {
|
||||||
|
os.Setenv("BEEGO_RUNMODE", runmode)
|
||||||
|
beeLogger.Log.Warnf("Using '%s' as 'runmode'", os.Getenv("BEEGO_RUNMODE"))
|
||||||
|
} else if os.Getenv("BEEGO_RUNMODE") != "" {
|
||||||
|
beeLogger.Log.Warnf("Using '%s' as 'runmode'", os.Getenv("BEEGO_RUNMODE"))
|
||||||
|
}
|
||||||
|
|
||||||
|
var paths []string
|
||||||
|
readAppDirectories(appPath, &paths)
|
||||||
|
|
||||||
|
// Because monitor files has some issues, we watch current directory
|
||||||
|
// and ignore non-go files.
|
||||||
|
for _, p := range config.Conf.DirStruct.Others {
|
||||||
|
paths = append(paths, strings.Replace(p, "$GOPATH", currentGoPath, -1))
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(extraPackages) > 0 {
|
||||||
|
// get the full path
|
||||||
|
for _, packagePath := range extraPackages {
|
||||||
|
if found, _, _fullPath := utils.SearchGOPATHs(packagePath); found {
|
||||||
|
readAppDirectories(_fullPath, &paths)
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Warnf("No extra package '%s' found in your GOPATH", packagePath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// let paths unique
|
||||||
|
strSet := make(map[string]struct{})
|
||||||
|
for _, p := range paths {
|
||||||
|
strSet[p] = struct{}{}
|
||||||
|
}
|
||||||
|
paths = make([]string, len(strSet))
|
||||||
|
index := 0
|
||||||
|
for i := range strSet {
|
||||||
|
paths[index] = i
|
||||||
|
index++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
files := []string{}
|
||||||
|
for _, arg := range mainFiles {
|
||||||
|
if len(arg) > 0 {
|
||||||
|
files = append(files, arg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if downdoc == "true" {
|
||||||
|
if _, err := os.Stat(path.Join(appPath, "swagger", "index.html")); err != nil {
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
downloadFromURL(swaggerlink, "swagger.zip")
|
||||||
|
unzipAndDelete("swagger.zip")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start the Reload server (if enabled)
|
||||||
|
if config.Conf.EnableReload {
|
||||||
|
startReloadServer()
|
||||||
|
}
|
||||||
|
if gendoc == "true" {
|
||||||
|
NewWatcher(paths, files, true)
|
||||||
|
AutoBuild(files, true)
|
||||||
|
} else {
|
||||||
|
NewWatcher(paths, files, false)
|
||||||
|
AutoBuild(files, false)
|
||||||
|
}
|
||||||
|
|
||||||
|
for {
|
||||||
|
<-exit
|
||||||
|
runtime.Goexit()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func readAppDirectories(directory string, paths *[]string) {
|
||||||
|
fileInfos, err := ioutil.ReadDir(directory)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
useDirectory := false
|
||||||
|
for _, fileInfo := range fileInfos {
|
||||||
|
if strings.HasSuffix(fileInfo.Name(), "docs") {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if strings.HasSuffix(fileInfo.Name(), "swagger") {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if !vendorWatch && strings.HasSuffix(fileInfo.Name(), "vendor") {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if isExcluded(path.Join(directory, fileInfo.Name())) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if fileInfo.IsDir() && fileInfo.Name()[0] != '.' {
|
||||||
|
readAppDirectories(directory+"/"+fileInfo.Name(), paths)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if useDirectory {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if path.Ext(fileInfo.Name()) == ".go" || (ifStaticFile(fileInfo.Name()) && config.Conf.EnableReload) {
|
||||||
|
*paths = append(*paths, directory)
|
||||||
|
useDirectory = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// If a file is excluded
|
||||||
|
func isExcluded(filePath string) bool {
|
||||||
|
for _, p := range excludedPaths {
|
||||||
|
absP, err := path.Abs(p)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Cannot get absolute path of '%s'", p)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
absFilePath, err := path.Abs(filePath)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Cannot get absolute path of '%s'", filePath)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if strings.HasPrefix(absFilePath, absP) {
|
||||||
|
beeLogger.Log.Infof("'%s' is not being watched", filePath)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
|
@ -12,7 +12,7 @@
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
|
|
||||||
package main
|
package run
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
@ -24,95 +24,91 @@ import (
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/config"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
"github.com/fsnotify/fsnotify"
|
"github.com/fsnotify/fsnotify"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
cmd *exec.Cmd
|
cmd *exec.Cmd
|
||||||
state sync.Mutex
|
state sync.Mutex
|
||||||
eventTime = make(map[string]int64)
|
eventTime = make(map[string]int64)
|
||||||
scheduleTime time.Time
|
scheduleTime time.Time
|
||||||
|
watchExts = config.Conf.WatchExts
|
||||||
|
watchExtsStatic = config.Conf.WatchExtsStatic
|
||||||
|
ignoredFilesRegExps = []string{
|
||||||
|
`.#(\w+).go$`,
|
||||||
|
`.(\w+).go.swp$`,
|
||||||
|
`(\w+).go~$`,
|
||||||
|
`(\w+).tmp$`,
|
||||||
|
`commentsRouter_controllers.go$`,
|
||||||
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
// NewWatcher starts an fsnotify Watcher on the specified paths
|
// NewWatcher starts an fsnotify Watcher on the specified paths
|
||||||
func NewWatcher(paths []string, files []string, isgenerate bool) {
|
func NewWatcher(paths []string, files []string, isgenerate bool) {
|
||||||
watcher, err := fsnotify.NewWatcher()
|
watcher, err := fsnotify.NewWatcher()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Failed to create watcher: %s", err)
|
beeLogger.Log.Fatalf("Failed to create watcher: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
for {
|
||||||
select {
|
select {
|
||||||
case e := <-watcher.Events:
|
case e := <-watcher.Events:
|
||||||
isbuild := true
|
isBuild := true
|
||||||
|
|
||||||
|
if ifStaticFile(e.Name) && config.Conf.EnableReload {
|
||||||
|
sendReload(e.String())
|
||||||
|
continue
|
||||||
|
}
|
||||||
// Skip ignored files
|
// Skip ignored files
|
||||||
if shouldIgnoreFile(e.Name) {
|
if shouldIgnoreFile(e.Name) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if !checkIfWatchExt(e.Name) {
|
if !shouldWatchFileWithExtension(e.Name) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
mt := getFileModTime(e.Name)
|
mt := utils.GetFileModTime(e.Name)
|
||||||
if t := eventTime[e.Name]; mt == t {
|
if t := eventTime[e.Name]; mt == t {
|
||||||
logger.Infof(bold("Skipping: ")+"%s", e.String())
|
beeLogger.Log.Hintf(colors.Bold("Skipping: ")+"%s", e.String())
|
||||||
isbuild = false
|
isBuild = false
|
||||||
}
|
}
|
||||||
|
|
||||||
eventTime[e.Name] = mt
|
eventTime[e.Name] = mt
|
||||||
|
|
||||||
if isbuild {
|
if isBuild {
|
||||||
logger.Infof("Event fired: %s", e)
|
beeLogger.Log.Hintf("Event fired: %s", e)
|
||||||
go func() {
|
go func() {
|
||||||
// Wait 1s before autobuild util there is no file change.
|
// Wait 1s before autobuild until there is no file change.
|
||||||
scheduleTime = time.Now().Add(1 * time.Second)
|
scheduleTime = time.Now().Add(1 * time.Second)
|
||||||
for {
|
time.Sleep(time.Until(scheduleTime))
|
||||||
time.Sleep(scheduleTime.Sub(time.Now()))
|
|
||||||
if time.Now().After(scheduleTime) {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
AutoBuild(files, isgenerate)
|
AutoBuild(files, isgenerate)
|
||||||
|
|
||||||
|
if config.Conf.EnableReload {
|
||||||
|
// Wait 100ms more before refreshing the browser
|
||||||
|
time.Sleep(100 * time.Millisecond)
|
||||||
|
sendReload(e.String())
|
||||||
|
}
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
case err := <-watcher.Errors:
|
case err := <-watcher.Errors:
|
||||||
logger.Warnf("Watcher error: %s", err.Error()) // No need to exit here
|
beeLogger.Log.Warnf("Watcher error: %s", err.Error()) // No need to exit here
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
logger.Info("Initializing watcher...")
|
beeLogger.Log.Info("Initializing watcher...")
|
||||||
for _, path := range paths {
|
for _, path := range paths {
|
||||||
logger.Infof(bold("Watching: ")+"%s", path)
|
beeLogger.Log.Hintf(colors.Bold("Watching: ")+"%s", path)
|
||||||
err = watcher.Add(path)
|
err = watcher.Add(path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Failed to watch directory: %s", err)
|
beeLogger.Log.Fatalf("Failed to watch directory: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// getFileModTime returns unix timestamp of `os.File.ModTime` for the given path.
|
|
||||||
func getFileModTime(path string) int64 {
|
|
||||||
path = strings.Replace(path, "\\", "/", -1)
|
|
||||||
f, err := os.Open(path)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Failed to open file on '%s': %s", path, err)
|
|
||||||
return time.Now().Unix()
|
|
||||||
}
|
|
||||||
defer f.Close()
|
|
||||||
|
|
||||||
fi, err := f.Stat()
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Failed to get file stats: %s", err)
|
|
||||||
return time.Now().Unix()
|
|
||||||
}
|
|
||||||
|
|
||||||
return fi.ModTime().Unix()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// AutoBuild builds the specified set of files
|
// AutoBuild builds the specified set of files
|
||||||
|
@ -123,61 +119,36 @@ func AutoBuild(files []string, isgenerate bool) {
|
||||||
os.Chdir(currpath)
|
os.Chdir(currpath)
|
||||||
|
|
||||||
cmdName := "go"
|
cmdName := "go"
|
||||||
if conf.Gopm.Enable {
|
|
||||||
cmdName = "gopm"
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
var (
|
||||||
err error
|
err error
|
||||||
stderr bytes.Buffer
|
stderr bytes.Buffer
|
||||||
stdout bytes.Buffer
|
|
||||||
)
|
)
|
||||||
// For applications use full import path like "github.com/.../.."
|
// For applications use full import path like "github.com/.../.."
|
||||||
// are able to use "go install" to reduce build time.
|
// are able to use "go install" to reduce build time.
|
||||||
if conf.GoInstall {
|
if config.Conf.GoInstall {
|
||||||
icmd := exec.Command(cmdName, "install", "-v")
|
icmd := exec.Command(cmdName, "install", "-v")
|
||||||
icmd.Stdout = os.Stdout
|
icmd.Stdout = os.Stdout
|
||||||
icmd.Stderr = os.Stderr
|
icmd.Stderr = os.Stderr
|
||||||
icmd.Env = append(os.Environ(), "GOGC=off")
|
icmd.Env = append(os.Environ(), "GOGC=off")
|
||||||
icmd.Run()
|
icmd.Run()
|
||||||
}
|
}
|
||||||
if conf.Gopm.Install {
|
|
||||||
icmd := exec.Command("go", "list", "./...")
|
|
||||||
icmd.Stdout = &stdout
|
|
||||||
icmd.Env = append(os.Environ(), "GOGC=off")
|
|
||||||
err = icmd.Run()
|
|
||||||
if err == nil {
|
|
||||||
list := strings.Split(stdout.String(), "\n")[1:]
|
|
||||||
for _, pkg := range list {
|
|
||||||
if len(pkg) == 0 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
icmd = exec.Command(cmdName, "install", pkg)
|
|
||||||
icmd.Stdout = os.Stdout
|
|
||||||
icmd.Stderr = os.Stderr
|
|
||||||
icmd.Env = append(os.Environ(), "GOGC=off")
|
|
||||||
err = icmd.Run()
|
|
||||||
if err != nil {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if isgenerate {
|
if isgenerate {
|
||||||
logger.Info("Generating the docs...")
|
beeLogger.Log.Info("Generating the docs...")
|
||||||
icmd := exec.Command("bee", "generate", "docs")
|
icmd := exec.Command("bee", "generate", "docs")
|
||||||
icmd.Env = append(os.Environ(), "GOGC=off")
|
icmd.Env = append(os.Environ(), "GOGC=off")
|
||||||
err = icmd.Run()
|
err = icmd.Run()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to generate the docs.")
|
utils.Notify("", "Failed to generate the docs.")
|
||||||
|
beeLogger.Log.Errorf("Failed to generate the docs.")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
logger.Success("Docs generated!")
|
beeLogger.Log.Success("Docs generated!")
|
||||||
}
|
}
|
||||||
|
appName := appname
|
||||||
if err == nil {
|
if err == nil {
|
||||||
appName := appname
|
|
||||||
if runtime.GOOS == "windows" {
|
if runtime.GOOS == "windows" {
|
||||||
appName += ".exe"
|
appName += ".exe"
|
||||||
}
|
}
|
||||||
|
@ -187,6 +158,9 @@ func AutoBuild(files []string, isgenerate bool) {
|
||||||
if buildTags != "" {
|
if buildTags != "" {
|
||||||
args = append(args, "-tags", buildTags)
|
args = append(args, "-tags", buildTags)
|
||||||
}
|
}
|
||||||
|
if buildLDFlags != "" {
|
||||||
|
args = append(args, "-ldflags", buildLDFlags)
|
||||||
|
}
|
||||||
args = append(args, files...)
|
args = append(args, files...)
|
||||||
|
|
||||||
bcmd := exec.Command(cmdName, args...)
|
bcmd := exec.Command(cmdName, args...)
|
||||||
|
@ -194,62 +168,98 @@ func AutoBuild(files []string, isgenerate bool) {
|
||||||
bcmd.Stderr = &stderr
|
bcmd.Stderr = &stderr
|
||||||
err = bcmd.Run()
|
err = bcmd.Run()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Errorf("Failed to build the application: %s", stderr.String())
|
utils.Notify(stderr.String(), "Build Failed")
|
||||||
|
beeLogger.Log.Errorf("Failed to build the application: %s", stderr.String())
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Success("Built Successfully!")
|
beeLogger.Log.Success("Built Successfully!")
|
||||||
Restart(appname)
|
Restart(appName)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Kill kills the running command process
|
// Kill kills the running command process
|
||||||
func Kill() {
|
func Kill() {
|
||||||
defer func() {
|
defer func() {
|
||||||
if e := recover(); e != nil {
|
if e := recover(); e != nil {
|
||||||
logger.Infof("Kill recover: %s", e)
|
beeLogger.Log.Infof("Kill recover: %s", e)
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
if cmd != nil && cmd.Process != nil {
|
if cmd != nil && cmd.Process != nil {
|
||||||
err := cmd.Process.Kill()
|
// Windows does not support Interrupt
|
||||||
if err != nil {
|
if runtime.GOOS == "windows" {
|
||||||
logger.Errorf("Error while killing cmd process: %s", err)
|
cmd.Process.Signal(os.Kill)
|
||||||
|
} else {
|
||||||
|
cmd.Process.Signal(os.Interrupt)
|
||||||
|
}
|
||||||
|
|
||||||
|
ch := make(chan struct{}, 1)
|
||||||
|
go func() {
|
||||||
|
cmd.Wait()
|
||||||
|
ch <- struct{}{}
|
||||||
|
}()
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-ch:
|
||||||
|
return
|
||||||
|
case <-time.After(10 * time.Second):
|
||||||
|
beeLogger.Log.Info("Timeout. Force kill cmd process")
|
||||||
|
err := cmd.Process.Kill()
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Error while killing cmd process: %s", err)
|
||||||
|
}
|
||||||
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Restart kills the running command process and starts it again
|
// Restart kills the running command process and starts it again
|
||||||
func Restart(appname string) {
|
func Restart(appname string) {
|
||||||
logger.Debugf("Kill running process", __FILE__(), __LINE__())
|
beeLogger.Log.Debugf("Kill running process", utils.FILE(), utils.LINE())
|
||||||
Kill()
|
Kill()
|
||||||
go Start(appname)
|
go Start(appname)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Start starts the command process
|
// Start starts the command process
|
||||||
func Start(appname string) {
|
func Start(appname string) {
|
||||||
logger.Infof("Restarting '%s'...", appname)
|
beeLogger.Log.Infof("Restarting '%s'...", appname)
|
||||||
if strings.Index(appname, "./") == -1 {
|
if !strings.Contains(appname, "./") {
|
||||||
appname = "./" + appname
|
appname = "./" + appname
|
||||||
}
|
}
|
||||||
|
|
||||||
cmd = exec.Command(appname)
|
cmd = exec.Command(appname)
|
||||||
cmd.Stdout = os.Stdout
|
cmd.Stdout = os.Stdout
|
||||||
cmd.Stderr = os.Stderr
|
cmd.Stderr = os.Stderr
|
||||||
cmd.Args = append([]string{appname}, conf.CmdArgs...)
|
if runargs != "" {
|
||||||
cmd.Env = append(os.Environ(), conf.Envs...)
|
r := regexp.MustCompile("'.+'|\".+\"|\\S+")
|
||||||
|
m := r.FindAllString(runargs, -1)
|
||||||
|
cmd.Args = append([]string{appname}, m...)
|
||||||
|
} else {
|
||||||
|
cmd.Args = append([]string{appname}, config.Conf.CmdArgs...)
|
||||||
|
}
|
||||||
|
cmd.Env = append(os.Environ(), config.Conf.Envs...)
|
||||||
|
|
||||||
go cmd.Run()
|
go cmd.Run()
|
||||||
logger.Successf("'%s' is running...", appname)
|
beeLogger.Log.Successf("'%s' is running...", appname)
|
||||||
started <- true
|
started <- true
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func ifStaticFile(filename string) bool {
|
||||||
|
for _, s := range watchExtsStatic {
|
||||||
|
if strings.HasSuffix(filename, s) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
// shouldIgnoreFile ignores filenames generated by Emacs, Vim or SublimeText.
|
// shouldIgnoreFile ignores filenames generated by Emacs, Vim or SublimeText.
|
||||||
// It returns true if the file should be ignored, false otherwise.
|
// It returns true if the file should be ignored, false otherwise.
|
||||||
func shouldIgnoreFile(filename string) bool {
|
func shouldIgnoreFile(filename string) bool {
|
||||||
for _, regex := range ignoredFilesRegExps {
|
for _, regex := range ignoredFilesRegExps {
|
||||||
r, err := regexp.Compile(regex)
|
r, err := regexp.Compile(regex)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Could not compile regular expression: %s", err)
|
beeLogger.Log.Fatalf("Could not compile regular expression: %s", err)
|
||||||
}
|
}
|
||||||
if r.MatchString(filename) {
|
if r.MatchString(filename) {
|
||||||
return true
|
return true
|
||||||
|
@ -259,16 +269,9 @@ func shouldIgnoreFile(filename string) bool {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
var watchExts = []string{".go"}
|
// shouldWatchFileWithExtension returns true if the name of the file
|
||||||
var ignoredFilesRegExps = []string{
|
// hash a suffix that should be watched.
|
||||||
`.#(\w+).go`,
|
func shouldWatchFileWithExtension(name string) bool {
|
||||||
`.(\w+).go.swp`,
|
|
||||||
`(\w+).go~`,
|
|
||||||
`(\w+).tmp`,
|
|
||||||
}
|
|
||||||
|
|
||||||
// checkIfWatchExt returns true if the name HasSuffix <watch_ext>.
|
|
||||||
func checkIfWatchExt(name string) bool {
|
|
||||||
for _, s := range watchExts {
|
for _, s := range watchExts {
|
||||||
if strings.HasSuffix(name, s) {
|
if strings.HasSuffix(name, s) {
|
||||||
return true
|
return true
|
76
cmd/commands/server/server.go
Normal file
76
cmd/commands/server/server.go
Normal file
|
@ -0,0 +1,76 @@
|
||||||
|
// Copyright 2013 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
|
||||||
|
package apiapp
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/version"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
var CmdServer = &commands.Command{
|
||||||
|
// CustomFlags: true,
|
||||||
|
UsageLine: "server [port]",
|
||||||
|
Short: "serving static content over HTTP on port",
|
||||||
|
Long: `
|
||||||
|
The command 'server' creates a Beego API application.
|
||||||
|
`,
|
||||||
|
PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() },
|
||||||
|
Run: createAPI,
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
a utils.DocValue
|
||||||
|
p utils.DocValue
|
||||||
|
f utils.DocValue
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
CmdServer.Flag.Var(&a, "a", "Listen address")
|
||||||
|
CmdServer.Flag.Var(&p, "p", "Listen port")
|
||||||
|
CmdServer.Flag.Var(&f, "f", "Static files fold")
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdServer)
|
||||||
|
}
|
||||||
|
|
||||||
|
func createAPI(cmd *commands.Command, args []string) int {
|
||||||
|
if len(args) > 0 {
|
||||||
|
err := cmd.Flag.Parse(args[1:])
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if a == "" {
|
||||||
|
a = "127.0.0.1"
|
||||||
|
}
|
||||||
|
if p == "" {
|
||||||
|
p = "8080"
|
||||||
|
}
|
||||||
|
if f == "" {
|
||||||
|
cwd, _ := os.Getwd()
|
||||||
|
f = utils.DocValue(cwd)
|
||||||
|
}
|
||||||
|
beeLogger.Log.Infof("Start server on http://%s:%s, static file %s", a, p, f)
|
||||||
|
err := http.ListenAndServe(string(a)+":"+string(p), http.FileServer(http.Dir(f)))
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
41
cmd/commands/update/update.go
Normal file
41
cmd/commands/update/update.go
Normal file
|
@ -0,0 +1,41 @@
|
||||||
|
package update
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/config"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
var CmdUpdate = &commands.Command{
|
||||||
|
UsageLine: "update",
|
||||||
|
Short: "Update Bee",
|
||||||
|
Long: `
|
||||||
|
Automatic run command "go get -u github.com/beego/bee/v2" for selfupdate
|
||||||
|
`,
|
||||||
|
Run: updateBee,
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
fs := flag.NewFlagSet("update", flag.ContinueOnError)
|
||||||
|
CmdUpdate.Flag = *fs
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdUpdate)
|
||||||
|
}
|
||||||
|
|
||||||
|
func updateBee(cmd *commands.Command, args []string) int {
|
||||||
|
beeLogger.Log.Info("Updating")
|
||||||
|
beePath := config.GitRemotePath
|
||||||
|
cmdUp := exec.Command("go", "get", "-u", beePath)
|
||||||
|
cmdUp.Stdout = os.Stdout
|
||||||
|
cmdUp.Stderr = os.Stderr
|
||||||
|
if err := cmdUp.Run(); err != nil {
|
||||||
|
beeLogger.Log.Warnf("Run cmd err:%s", err)
|
||||||
|
}
|
||||||
|
// update the Time when updateBee every time
|
||||||
|
utils.UpdateLastPublishedTime()
|
||||||
|
return 0
|
||||||
|
}
|
73
cmd/commands/version/banner.go
Normal file
73
cmd/commands/version/banner.go
Normal file
|
@ -0,0 +1,73 @@
|
||||||
|
package version
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"runtime"
|
||||||
|
"text/template"
|
||||||
|
|
||||||
|
"time"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
// RuntimeInfo holds information about the current runtime.
|
||||||
|
type RuntimeInfo struct {
|
||||||
|
GoVersion string
|
||||||
|
GOOS string
|
||||||
|
GOARCH string
|
||||||
|
NumCPU int
|
||||||
|
GOPATH string
|
||||||
|
GOROOT string
|
||||||
|
Compiler string
|
||||||
|
BeeVersion string
|
||||||
|
Published string
|
||||||
|
}
|
||||||
|
|
||||||
|
// InitBanner loads the banner and prints it to output
|
||||||
|
// All errors are ignored, the application will not
|
||||||
|
// print the banner in case of error.
|
||||||
|
func InitBanner(out io.Writer, in io.Reader) {
|
||||||
|
if in == nil {
|
||||||
|
beeLogger.Log.Fatal("The input is nil")
|
||||||
|
}
|
||||||
|
|
||||||
|
banner, err := ioutil.ReadAll(in)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Error while trying to read the banner: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
show(out, string(banner))
|
||||||
|
}
|
||||||
|
|
||||||
|
func show(out io.Writer, content string) {
|
||||||
|
t, err := template.New("banner").
|
||||||
|
Funcs(template.FuncMap{"Now": Now}).
|
||||||
|
Parse(content)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Cannot parse the banner template: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = t.Execute(out, RuntimeInfo{
|
||||||
|
runtime.Version(),
|
||||||
|
runtime.GOOS,
|
||||||
|
runtime.GOARCH,
|
||||||
|
runtime.NumCPU(),
|
||||||
|
os.Getenv("GOPATH"),
|
||||||
|
runtime.GOROOT(),
|
||||||
|
runtime.Compiler,
|
||||||
|
version,
|
||||||
|
utils.GetLastPublishedTime(),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Now returns the current local time in the specified layout
|
||||||
|
func Now(layout string) string {
|
||||||
|
return time.Now().Format(layout)
|
||||||
|
}
|
110
cmd/commands/version/version.go
Normal file
110
cmd/commands/version/version.go
Normal file
|
@ -0,0 +1,110 @@
|
||||||
|
package version
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"gopkg.in/yaml.v2"
|
||||||
|
"os"
|
||||||
|
"runtime"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/config"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
)
|
||||||
|
|
||||||
|
const verboseVersionBanner string = `%s%s______
|
||||||
|
| ___ \
|
||||||
|
| |_/ / ___ ___
|
||||||
|
| ___ \ / _ \ / _ \
|
||||||
|
| |_/ /| __/| __/
|
||||||
|
\____/ \___| \___| v{{ .BeeVersion }}%s
|
||||||
|
%s%s
|
||||||
|
├── GoVersion : {{ .GoVersion }}
|
||||||
|
├── GOOS : {{ .GOOS }}
|
||||||
|
├── GOARCH : {{ .GOARCH }}
|
||||||
|
├── NumCPU : {{ .NumCPU }}
|
||||||
|
├── GOPATH : {{ .GOPATH }}
|
||||||
|
├── GOROOT : {{ .GOROOT }}
|
||||||
|
├── Compiler : {{ .Compiler }}
|
||||||
|
└── Date : {{ Now "Monday, 2 Jan 2006" }}%s
|
||||||
|
`
|
||||||
|
|
||||||
|
const shortVersionBanner = `______
|
||||||
|
| ___ \
|
||||||
|
| |_/ / ___ ___
|
||||||
|
| ___ \ / _ \ / _ \
|
||||||
|
| |_/ /| __/| __/
|
||||||
|
\____/ \___| \___| v{{ .BeeVersion }}
|
||||||
|
`
|
||||||
|
|
||||||
|
var CmdVersion = &commands.Command{
|
||||||
|
UsageLine: "version",
|
||||||
|
Short: "Prints the current Bee version",
|
||||||
|
Long: `
|
||||||
|
Prints the current Bee, Beego and Go version alongside the platform information.
|
||||||
|
`,
|
||||||
|
Run: versionCmd,
|
||||||
|
}
|
||||||
|
var outputFormat string
|
||||||
|
|
||||||
|
const version = config.Version
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
fs := flag.NewFlagSet("version", flag.ContinueOnError)
|
||||||
|
fs.StringVar(&outputFormat, "o", "", "Set the output format. Either json or yaml.")
|
||||||
|
CmdVersion.Flag = *fs
|
||||||
|
commands.AvailableCommands = append(commands.AvailableCommands, CmdVersion)
|
||||||
|
}
|
||||||
|
|
||||||
|
func versionCmd(cmd *commands.Command, args []string) int {
|
||||||
|
|
||||||
|
cmd.Flag.Parse(args)
|
||||||
|
stdout := cmd.Out()
|
||||||
|
|
||||||
|
if outputFormat != "" {
|
||||||
|
runtimeInfo := RuntimeInfo{
|
||||||
|
GoVersion: runtime.Version(),
|
||||||
|
GOOS: runtime.GOOS,
|
||||||
|
GOARCH: runtime.GOARCH,
|
||||||
|
NumCPU: runtime.NumCPU(),
|
||||||
|
GOPATH: os.Getenv("GOPATH"),
|
||||||
|
GOROOT: runtime.GOROOT(),
|
||||||
|
Compiler: runtime.Compiler,
|
||||||
|
BeeVersion: version,
|
||||||
|
}
|
||||||
|
switch outputFormat {
|
||||||
|
case "json":
|
||||||
|
{
|
||||||
|
b, err := json.MarshalIndent(runtimeInfo, "", " ")
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
}
|
||||||
|
fmt.Println(string(b))
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
case "yaml":
|
||||||
|
{
|
||||||
|
b, err := yaml.Marshal(&runtimeInfo)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
}
|
||||||
|
fmt.Println(string(b))
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
coloredBanner := fmt.Sprintf(verboseVersionBanner, "\x1b[35m", "\x1b[1m",
|
||||||
|
"\x1b[0m", "\x1b[32m", "\x1b[1m", "\x1b[0m")
|
||||||
|
InitBanner(stdout, bytes.NewBufferString(coloredBanner))
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// ShowShortVersionBanner prints the short version banner.
|
||||||
|
func ShowShortVersionBanner() {
|
||||||
|
output := colors.NewColorWriter(os.Stdout)
|
||||||
|
InitBanner(output, bytes.NewBufferString(colors.MagentaBold(shortVersionBanner)))
|
||||||
|
}
|
261
code.go
261
code.go
|
@ -1,261 +0,0 @@
|
||||||
// Copyright 2011 Gary Burd
|
|
||||||
// Copyright 2013 Unknown
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"go/ast"
|
|
||||||
"go/printer"
|
|
||||||
"go/scanner"
|
|
||||||
"go/token"
|
|
||||||
"math"
|
|
||||||
"strconv"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
notPredeclared = iota
|
|
||||||
predeclaredType
|
|
||||||
predeclaredConstant
|
|
||||||
predeclaredFunction
|
|
||||||
)
|
|
||||||
|
|
||||||
// predeclared represents the set of all predeclared identifiers.
|
|
||||||
var predeclared = map[string]int{
|
|
||||||
"bool": predeclaredType,
|
|
||||||
"byte": predeclaredType,
|
|
||||||
"complex128": predeclaredType,
|
|
||||||
"complex64": predeclaredType,
|
|
||||||
"error": predeclaredType,
|
|
||||||
"float32": predeclaredType,
|
|
||||||
"float64": predeclaredType,
|
|
||||||
"int16": predeclaredType,
|
|
||||||
"int32": predeclaredType,
|
|
||||||
"int64": predeclaredType,
|
|
||||||
"int8": predeclaredType,
|
|
||||||
"int": predeclaredType,
|
|
||||||
"rune": predeclaredType,
|
|
||||||
"string": predeclaredType,
|
|
||||||
"uint16": predeclaredType,
|
|
||||||
"uint32": predeclaredType,
|
|
||||||
"uint64": predeclaredType,
|
|
||||||
"uint8": predeclaredType,
|
|
||||||
"uint": predeclaredType,
|
|
||||||
"uintptr": predeclaredType,
|
|
||||||
|
|
||||||
"true": predeclaredConstant,
|
|
||||||
"false": predeclaredConstant,
|
|
||||||
"iota": predeclaredConstant,
|
|
||||||
"nil": predeclaredConstant,
|
|
||||||
|
|
||||||
"append": predeclaredFunction,
|
|
||||||
"cap": predeclaredFunction,
|
|
||||||
"close": predeclaredFunction,
|
|
||||||
"complex": predeclaredFunction,
|
|
||||||
"copy": predeclaredFunction,
|
|
||||||
"delete": predeclaredFunction,
|
|
||||||
"imag": predeclaredFunction,
|
|
||||||
"len": predeclaredFunction,
|
|
||||||
"make": predeclaredFunction,
|
|
||||||
"new": predeclaredFunction,
|
|
||||||
"panic": predeclaredFunction,
|
|
||||||
"print": predeclaredFunction,
|
|
||||||
"println": predeclaredFunction,
|
|
||||||
"real": predeclaredFunction,
|
|
||||||
"recover": predeclaredFunction,
|
|
||||||
}
|
|
||||||
|
|
||||||
const (
|
|
||||||
ExportLinkAnnotation AnnotationKind = iota
|
|
||||||
AnchorAnnotation
|
|
||||||
CommentAnnotation
|
|
||||||
PackageLinkAnnotation
|
|
||||||
BuiltinAnnotation
|
|
||||||
)
|
|
||||||
|
|
||||||
// annotationVisitor collects annotations.
|
|
||||||
type annotationVisitor struct {
|
|
||||||
annotations []Annotation
|
|
||||||
}
|
|
||||||
|
|
||||||
func (v *annotationVisitor) add(kind AnnotationKind, importPath string) {
|
|
||||||
v.annotations = append(v.annotations, Annotation{Kind: kind, ImportPath: importPath})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (v *annotationVisitor) ignoreName() {
|
|
||||||
v.add(-1, "")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (v *annotationVisitor) Visit(n ast.Node) ast.Visitor {
|
|
||||||
switch n := n.(type) {
|
|
||||||
case *ast.TypeSpec:
|
|
||||||
v.ignoreName()
|
|
||||||
ast.Walk(v, n.Type)
|
|
||||||
case *ast.FuncDecl:
|
|
||||||
if n.Recv != nil {
|
|
||||||
ast.Walk(v, n.Recv)
|
|
||||||
}
|
|
||||||
v.ignoreName()
|
|
||||||
ast.Walk(v, n.Type)
|
|
||||||
case *ast.Field:
|
|
||||||
for range n.Names {
|
|
||||||
v.ignoreName()
|
|
||||||
}
|
|
||||||
ast.Walk(v, n.Type)
|
|
||||||
case *ast.ValueSpec:
|
|
||||||
for range n.Names {
|
|
||||||
v.add(AnchorAnnotation, "")
|
|
||||||
}
|
|
||||||
if n.Type != nil {
|
|
||||||
ast.Walk(v, n.Type)
|
|
||||||
}
|
|
||||||
for _, x := range n.Values {
|
|
||||||
ast.Walk(v, x)
|
|
||||||
}
|
|
||||||
case *ast.Ident:
|
|
||||||
switch {
|
|
||||||
case n.Obj == nil && predeclared[n.Name] != notPredeclared:
|
|
||||||
v.add(BuiltinAnnotation, "")
|
|
||||||
case n.Obj != nil && ast.IsExported(n.Name):
|
|
||||||
v.add(ExportLinkAnnotation, "")
|
|
||||||
default:
|
|
||||||
v.ignoreName()
|
|
||||||
}
|
|
||||||
case *ast.SelectorExpr:
|
|
||||||
if x, _ := n.X.(*ast.Ident); x != nil {
|
|
||||||
if obj := x.Obj; obj != nil && obj.Kind == ast.Pkg {
|
|
||||||
if spec, _ := obj.Decl.(*ast.ImportSpec); spec != nil {
|
|
||||||
if path, err := strconv.Unquote(spec.Path.Value); err == nil {
|
|
||||||
v.add(PackageLinkAnnotation, path)
|
|
||||||
if path == "C" {
|
|
||||||
v.ignoreName()
|
|
||||||
} else {
|
|
||||||
v.add(ExportLinkAnnotation, path)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
ast.Walk(v, n.X)
|
|
||||||
v.ignoreName()
|
|
||||||
default:
|
|
||||||
return v
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func printDecl(decl ast.Node, fset *token.FileSet, buf []byte) (Code, []byte) {
|
|
||||||
v := &annotationVisitor{}
|
|
||||||
ast.Walk(v, decl)
|
|
||||||
|
|
||||||
buf = buf[:0]
|
|
||||||
err := (&printer.Config{Mode: printer.UseSpaces, Tabwidth: 4}).Fprint(sliceWriter{&buf}, fset, decl)
|
|
||||||
if err != nil {
|
|
||||||
return Code{Text: err.Error()}, buf
|
|
||||||
}
|
|
||||||
|
|
||||||
var annotations []Annotation
|
|
||||||
var s scanner.Scanner
|
|
||||||
fset = token.NewFileSet()
|
|
||||||
file := fset.AddFile("", fset.Base(), len(buf))
|
|
||||||
s.Init(file, buf, nil, scanner.ScanComments)
|
|
||||||
loop:
|
|
||||||
for {
|
|
||||||
pos, tok, lit := s.Scan()
|
|
||||||
switch tok {
|
|
||||||
case token.EOF:
|
|
||||||
break loop
|
|
||||||
case token.COMMENT:
|
|
||||||
p := file.Offset(pos)
|
|
||||||
e := p + len(lit)
|
|
||||||
if p > math.MaxInt16 || e > math.MaxInt16 {
|
|
||||||
break loop
|
|
||||||
}
|
|
||||||
annotations = append(annotations, Annotation{Kind: CommentAnnotation, Pos: int16(p), End: int16(e)})
|
|
||||||
case token.IDENT:
|
|
||||||
if len(v.annotations) == 0 {
|
|
||||||
// Oops!
|
|
||||||
break loop
|
|
||||||
}
|
|
||||||
annotation := v.annotations[0]
|
|
||||||
v.annotations = v.annotations[1:]
|
|
||||||
if annotation.Kind == -1 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
p := file.Offset(pos)
|
|
||||||
e := p + len(lit)
|
|
||||||
if p > math.MaxInt16 || e > math.MaxInt16 {
|
|
||||||
break loop
|
|
||||||
}
|
|
||||||
annotation.Pos = int16(p)
|
|
||||||
annotation.End = int16(e)
|
|
||||||
if len(annotations) > 0 && annotation.Kind == ExportLinkAnnotation {
|
|
||||||
prev := annotations[len(annotations)-1]
|
|
||||||
if prev.Kind == PackageLinkAnnotation &&
|
|
||||||
prev.ImportPath == annotation.ImportPath &&
|
|
||||||
prev.End+1 == annotation.Pos {
|
|
||||||
// merge with previous
|
|
||||||
annotation.Pos = prev.Pos
|
|
||||||
annotations[len(annotations)-1] = annotation
|
|
||||||
continue loop
|
|
||||||
}
|
|
||||||
}
|
|
||||||
annotations = append(annotations, annotation)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return Code{Text: string(buf), Annotations: annotations}, buf
|
|
||||||
}
|
|
||||||
|
|
||||||
type AnnotationKind int16
|
|
||||||
|
|
||||||
type Annotation struct {
|
|
||||||
Pos, End int16
|
|
||||||
Kind AnnotationKind
|
|
||||||
ImportPath string
|
|
||||||
}
|
|
||||||
|
|
||||||
type Code struct {
|
|
||||||
Text string
|
|
||||||
Annotations []Annotation
|
|
||||||
}
|
|
||||||
|
|
||||||
func commentAnnotations(src string) []Annotation {
|
|
||||||
var annotations []Annotation
|
|
||||||
var s scanner.Scanner
|
|
||||||
fset := token.NewFileSet()
|
|
||||||
file := fset.AddFile("", fset.Base(), len(src))
|
|
||||||
s.Init(file, []byte(src), nil, scanner.ScanComments)
|
|
||||||
for {
|
|
||||||
pos, tok, lit := s.Scan()
|
|
||||||
switch tok {
|
|
||||||
case token.EOF:
|
|
||||||
return annotations
|
|
||||||
case token.COMMENT:
|
|
||||||
p := file.Offset(pos)
|
|
||||||
e := p + len(lit)
|
|
||||||
if p > math.MaxInt16 || e > math.MaxInt16 {
|
|
||||||
return annotations
|
|
||||||
}
|
|
||||||
annotations = append(annotations, Annotation{Kind: CommentAnnotation, Pos: int16(p), End: int16(e)})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type sliceWriter struct{ p *[]byte }
|
|
||||||
|
|
||||||
func (w sliceWriter) Write(p []byte) (int, error) {
|
|
||||||
*w.p = append(*w.p, p...)
|
|
||||||
return len(p), nil
|
|
||||||
}
|
|
175
conf.go
175
conf.go
|
@ -1,175 +0,0 @@
|
||||||
// Copyright 2013 bee authors
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"io/ioutil"
|
|
||||||
"os"
|
|
||||||
|
|
||||||
"io"
|
|
||||||
"path/filepath"
|
|
||||||
|
|
||||||
"gopkg.in/yaml.v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
const confVer = 0
|
|
||||||
|
|
||||||
var defaultConf = `{
|
|
||||||
"version": 0,
|
|
||||||
"gopm": {
|
|
||||||
"enable": false,
|
|
||||||
"install": false
|
|
||||||
},
|
|
||||||
"go_install": true,
|
|
||||||
"watch_ext": [],
|
|
||||||
"dir_structure": {
|
|
||||||
"watch_all": false,
|
|
||||||
"controllers": "",
|
|
||||||
"models": "",
|
|
||||||
"others": []
|
|
||||||
},
|
|
||||||
"cmd_args": [],
|
|
||||||
"envs": [],
|
|
||||||
"database": {
|
|
||||||
"driver": "mysql"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
`
|
|
||||||
var conf struct {
|
|
||||||
Version int
|
|
||||||
// gopm support
|
|
||||||
Gopm struct {
|
|
||||||
Enable bool
|
|
||||||
Install bool
|
|
||||||
}
|
|
||||||
// Indicates whether execute "go install" before "go build".
|
|
||||||
GoInstall bool `json:"go_install" yaml:"go_install"`
|
|
||||||
WatchExt []string `json:"watch_ext" yaml:"watch_ext"`
|
|
||||||
DirStruct struct {
|
|
||||||
WatchAll bool `json:"watch_all" yaml:"watch_all"`
|
|
||||||
Controllers string
|
|
||||||
Models string
|
|
||||||
Others []string // Other directories.
|
|
||||||
} `json:"dir_structure" yaml:"dir_structure"`
|
|
||||||
CmdArgs []string `json:"cmd_args" yaml:"cmd_args"`
|
|
||||||
Envs []string
|
|
||||||
Bale struct {
|
|
||||||
Import string
|
|
||||||
Dirs []string
|
|
||||||
IngExt []string `json:"ignore_ext" yaml:"ignore_ext"`
|
|
||||||
}
|
|
||||||
Database struct {
|
|
||||||
Driver string
|
|
||||||
Conn string
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// loadConfig loads customized configuration.
|
|
||||||
func loadConfig() (err error) {
|
|
||||||
err = filepath.Walk(".", func(path string, fileInfo os.FileInfo, err error) error {
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
if fileInfo.IsDir() {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
if fileInfo.Name() == "bee.json" {
|
|
||||||
logger.Info("Loading configuration from 'bee.json'...")
|
|
||||||
err = parseJSON(path, conf)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Failed to parse JSON file: %s", err)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return io.EOF
|
|
||||||
}
|
|
||||||
|
|
||||||
if fileInfo.Name() == "Beefile" {
|
|
||||||
logger.Info("Loading configuration from 'Beefile'...")
|
|
||||||
err = parseYAML(path, conf)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Failed to parse YAML file: %s", err)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return io.EOF
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
|
|
||||||
// In case no configuration file found or an error different than io.EOF,
|
|
||||||
// fallback to default configuration
|
|
||||||
if err != io.EOF {
|
|
||||||
logger.Info("Loading default configuration...")
|
|
||||||
err = json.Unmarshal([]byte(defaultConf), &conf)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// No need to return io.EOF error
|
|
||||||
err = nil
|
|
||||||
|
|
||||||
// Check format version
|
|
||||||
if conf.Version != confVer {
|
|
||||||
logger.Warn("Your configuration file is outdated. Please do consider updating it.")
|
|
||||||
logger.Hint("Check the latest version of bee's configuration file.")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Set variables
|
|
||||||
if len(conf.DirStruct.Controllers) == 0 {
|
|
||||||
conf.DirStruct.Controllers = "controllers"
|
|
||||||
}
|
|
||||||
if len(conf.DirStruct.Models) == 0 {
|
|
||||||
conf.DirStruct.Models = "models"
|
|
||||||
}
|
|
||||||
|
|
||||||
// Append watch exts
|
|
||||||
watchExts = append(watchExts, conf.WatchExt...)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseJSON(path string, v interface{}) error {
|
|
||||||
var (
|
|
||||||
data []byte
|
|
||||||
err error
|
|
||||||
)
|
|
||||||
data, err = ioutil.ReadFile(path)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
err = json.Unmarshal(data, &v)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseYAML(path string, v interface{}) error {
|
|
||||||
var (
|
|
||||||
data []byte
|
|
||||||
err error
|
|
||||||
)
|
|
||||||
data, err = ioutil.ReadFile(path)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
err = yaml.Unmarshal(data, &v)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
170
config/conf.go
Normal file
170
config/conf.go
Normal file
|
@ -0,0 +1,170 @@
|
||||||
|
// Copyright 2013 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
|
||||||
|
"gopkg.in/yaml.v2"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
const confVer = 0
|
||||||
|
|
||||||
|
const (
|
||||||
|
Version = "2.1.0"
|
||||||
|
GitRemotePath = "github.com/beego/bee/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
var Conf = struct {
|
||||||
|
Version int
|
||||||
|
WatchExts []string `json:"watch_ext" yaml:"watch_ext"`
|
||||||
|
WatchExtsStatic []string `json:"watch_ext_static" yaml:"watch_ext_static"`
|
||||||
|
GoInstall bool `json:"go_install" yaml:"go_install"` // Indicates whether execute "go install" before "go build".
|
||||||
|
DirStruct dirStruct `json:"dir_structure" yaml:"dir_structure"`
|
||||||
|
CmdArgs []string `json:"cmd_args" yaml:"cmd_args"`
|
||||||
|
Envs []string
|
||||||
|
Bale bale
|
||||||
|
Database database
|
||||||
|
EnableReload bool `json:"enable_reload" yaml:"enable_reload"`
|
||||||
|
EnableNotification bool `json:"enable_notification" yaml:"enable_notification"`
|
||||||
|
Scripts map[string]string `json:"scripts" yaml:"scripts"`
|
||||||
|
}{
|
||||||
|
WatchExts: []string{".go"},
|
||||||
|
WatchExtsStatic: []string{".html", ".tpl", ".js", ".css"},
|
||||||
|
GoInstall: true,
|
||||||
|
DirStruct: dirStruct{
|
||||||
|
Others: []string{},
|
||||||
|
},
|
||||||
|
CmdArgs: []string{},
|
||||||
|
Envs: []string{},
|
||||||
|
Bale: bale{
|
||||||
|
Dirs: []string{},
|
||||||
|
IngExt: []string{},
|
||||||
|
},
|
||||||
|
Database: database{
|
||||||
|
Driver: "mysql",
|
||||||
|
},
|
||||||
|
EnableNotification: true,
|
||||||
|
Scripts: map[string]string{},
|
||||||
|
}
|
||||||
|
|
||||||
|
// dirStruct describes the application's directory structure
|
||||||
|
type dirStruct struct {
|
||||||
|
WatchAll bool `json:"watch_all" yaml:"watch_all"`
|
||||||
|
Controllers string
|
||||||
|
Models string
|
||||||
|
Others []string // Other directories
|
||||||
|
}
|
||||||
|
|
||||||
|
// bale
|
||||||
|
type bale struct {
|
||||||
|
Import string
|
||||||
|
Dirs []string
|
||||||
|
IngExt []string `json:"ignore_ext" yaml:"ignore_ext"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// database holds the database connection information
|
||||||
|
type database struct {
|
||||||
|
Driver string
|
||||||
|
Conn string
|
||||||
|
Dir string
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadConfig loads the bee tool configuration.
|
||||||
|
// It looks for Beefile or bee.json in the current path,
|
||||||
|
// and falls back to default configuration in case not found.
|
||||||
|
func LoadConfig() {
|
||||||
|
currentPath, err := os.Getwd()
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
dir, err := os.Open(currentPath)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
}
|
||||||
|
defer dir.Close()
|
||||||
|
|
||||||
|
files, err := dir.Readdir(-1)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, file := range files {
|
||||||
|
switch file.Name() {
|
||||||
|
case "bee.json":
|
||||||
|
{
|
||||||
|
err = parseJSON(filepath.Join(currentPath, file.Name()), &Conf)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Failed to parse JSON file: %s", err)
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
case "Beefile":
|
||||||
|
{
|
||||||
|
err = parseYAML(filepath.Join(currentPath, file.Name()), &Conf)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Failed to parse YAML file: %s", err)
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check format version
|
||||||
|
if Conf.Version != confVer {
|
||||||
|
beeLogger.Log.Warn("Your configuration file is outdated. Please do consider updating it.")
|
||||||
|
beeLogger.Log.Hint("Check the latest version of bee's configuration file.")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Set variables
|
||||||
|
if len(Conf.DirStruct.Controllers) == 0 {
|
||||||
|
Conf.DirStruct.Controllers = "controllers"
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(Conf.DirStruct.Models) == 0 {
|
||||||
|
Conf.DirStruct.Models = "models"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseJSON(path string, v interface{}) error {
|
||||||
|
var (
|
||||||
|
data []byte
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
data, err = ioutil.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
err = json.Unmarshal(data, v)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseYAML(path string, v interface{}) error {
|
||||||
|
var (
|
||||||
|
data []byte
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
data, err = ioutil.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
err = yaml.Unmarshal(data, v)
|
||||||
|
return err
|
||||||
|
}
|
200
g.go
200
g.go
|
@ -1,200 +0,0 @@
|
||||||
// Copyright 2013 bee authors
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"os"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
var cmdGenerate = &Command{
|
|
||||||
UsageLine: "generate [command]",
|
|
||||||
Short: "Source code generator",
|
|
||||||
Long: `▶ {{"To scaffold out your entire application:"|bold}}
|
|
||||||
|
|
||||||
$ bee generate scaffold [scaffoldname] [-fields="title:string,body:text"] [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"]
|
|
||||||
|
|
||||||
▶ {{"To generate a Model based on fields:"|bold}}
|
|
||||||
|
|
||||||
$ bee generate model [modelname] [-fields="name:type"]
|
|
||||||
|
|
||||||
▶ {{"To generate a controller:"|bold}}
|
|
||||||
|
|
||||||
$ bee generate controller [controllerfile]
|
|
||||||
|
|
||||||
▶ {{"To generate a CRUD view:"|bold}}
|
|
||||||
|
|
||||||
$ bee generate view [viewpath]
|
|
||||||
|
|
||||||
▶ {{"To generate a migration file for making database schema updates:"|bold}}
|
|
||||||
|
|
||||||
$ bee generate migration [migrationfile] [-fields="name:type"]
|
|
||||||
|
|
||||||
▶ {{"To generate swagger doc file:"|bold}}
|
|
||||||
|
|
||||||
$ bee generate docs
|
|
||||||
|
|
||||||
▶ {{"To generate a test case:"|bold}}
|
|
||||||
|
|
||||||
$ bee generate test [routerfile]
|
|
||||||
|
|
||||||
▶ {{"To generate appcode based on an existing database:"|bold}}
|
|
||||||
|
|
||||||
$ bee generate appcode [-tables=""] [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"] [-level=3]
|
|
||||||
`,
|
|
||||||
PreRun: func(cmd *Command, args []string) { ShowShortVersionBanner() },
|
|
||||||
Run: generateCode,
|
|
||||||
}
|
|
||||||
|
|
||||||
var driver docValue
|
|
||||||
var conn docValue
|
|
||||||
var level docValue
|
|
||||||
var tables docValue
|
|
||||||
var fields docValue
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
cmdGenerate.Flag.Var(&tables, "tables", "List of table names separated by a comma.")
|
|
||||||
cmdGenerate.Flag.Var(&driver, "driver", "Database driver. Either mysql, postgres or sqlite.")
|
|
||||||
cmdGenerate.Flag.Var(&conn, "conn", "Connection string used by the driver to connect to a database instance.")
|
|
||||||
cmdGenerate.Flag.Var(&level, "level", "Either 1, 2 or 3. i.e. 1=models; 2=models and controllers; 3=models, controllers and routers.")
|
|
||||||
cmdGenerate.Flag.Var(&fields, "fields", "List of table fields.")
|
|
||||||
}
|
|
||||||
|
|
||||||
func generateCode(cmd *Command, args []string) int {
|
|
||||||
currpath, _ := os.Getwd()
|
|
||||||
if len(args) < 1 {
|
|
||||||
logger.Fatal("Command is missing")
|
|
||||||
}
|
|
||||||
|
|
||||||
gps := GetGOPATHs()
|
|
||||||
if len(gps) == 0 {
|
|
||||||
logger.Fatal("GOPATH environment variable is not set or empty")
|
|
||||||
}
|
|
||||||
|
|
||||||
gopath := gps[0]
|
|
||||||
|
|
||||||
logger.Debugf("GOPATH: %s", __FILE__(), __LINE__(), gopath)
|
|
||||||
|
|
||||||
gcmd := args[0]
|
|
||||||
switch gcmd {
|
|
||||||
case "scaffold":
|
|
||||||
if len(args) < 2 {
|
|
||||||
logger.Fatal("Wrong number of arguments. Run: bee help generate")
|
|
||||||
}
|
|
||||||
// Load the configuration
|
|
||||||
err := loadConfig()
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("Failed to load configuration: %s", err)
|
|
||||||
}
|
|
||||||
cmd.Flag.Parse(args[2:])
|
|
||||||
if driver == "" {
|
|
||||||
driver = docValue(conf.Database.Driver)
|
|
||||||
if driver == "" {
|
|
||||||
driver = "mysql"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if conn == "" {
|
|
||||||
conn = docValue(conf.Database.Conn)
|
|
||||||
if conn == "" {
|
|
||||||
conn = "root:@tcp(127.0.0.1:3306)/test"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if fields == "" {
|
|
||||||
logger.Hint("fields option should not be empty, i.e. -fields=\"title:string,body:text\"")
|
|
||||||
logger.Fatal("Wrong number of arguments. Run: bee help generate")
|
|
||||||
}
|
|
||||||
sname := args[1]
|
|
||||||
generateScaffold(sname, fields.String(), currpath, driver.String(), conn.String())
|
|
||||||
case "docs":
|
|
||||||
generateDocs(currpath)
|
|
||||||
case "appcode":
|
|
||||||
// Load the configuration
|
|
||||||
err := loadConfig()
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("Failed to load configuration: %s", err)
|
|
||||||
}
|
|
||||||
cmd.Flag.Parse(args[1:])
|
|
||||||
if driver == "" {
|
|
||||||
driver = docValue(conf.Database.Driver)
|
|
||||||
if driver == "" {
|
|
||||||
driver = "mysql"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if conn == "" {
|
|
||||||
conn = docValue(conf.Database.Conn)
|
|
||||||
if conn == "" {
|
|
||||||
if driver == "mysql" {
|
|
||||||
conn = "root:@tcp(127.0.0.1:3306)/test"
|
|
||||||
} else if driver == "postgres" {
|
|
||||||
conn = "postgres://postgres:postgres@127.0.0.1:5432/postgres"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if level == "" {
|
|
||||||
level = "3"
|
|
||||||
}
|
|
||||||
logger.Infof("Using '%s' as 'driver'", driver)
|
|
||||||
logger.Infof("Using '%s' as 'conn'", conn)
|
|
||||||
logger.Infof("Using '%s' as 'tables'", tables)
|
|
||||||
logger.Infof("Using '%s' as 'level'", level)
|
|
||||||
generateAppcode(driver.String(), conn.String(), level.String(), tables.String(), currpath)
|
|
||||||
case "migration":
|
|
||||||
if len(args) < 2 {
|
|
||||||
logger.Fatal("Wrong number of arguments. Run: bee help generate")
|
|
||||||
}
|
|
||||||
cmd.Flag.Parse(args[2:])
|
|
||||||
mname := args[1]
|
|
||||||
|
|
||||||
logger.Infof("Using '%s' as migration name", mname)
|
|
||||||
|
|
||||||
upsql := ""
|
|
||||||
downsql := ""
|
|
||||||
if fields != "" {
|
|
||||||
dbMigrator := newDBDriver()
|
|
||||||
upsql = dbMigrator.generateCreateUp(mname)
|
|
||||||
downsql = dbMigrator.generateCreateDown(mname)
|
|
||||||
}
|
|
||||||
generateMigration(mname, upsql, downsql, currpath)
|
|
||||||
case "controller":
|
|
||||||
if len(args) == 2 {
|
|
||||||
cname := args[1]
|
|
||||||
generateController(cname, currpath)
|
|
||||||
} else {
|
|
||||||
logger.Fatal("Wrong number of arguments. Run: bee help generate")
|
|
||||||
}
|
|
||||||
case "model":
|
|
||||||
if len(args) < 2 {
|
|
||||||
logger.Fatal("Wrong number of arguments. Run: bee help generate")
|
|
||||||
}
|
|
||||||
cmd.Flag.Parse(args[2:])
|
|
||||||
if fields == "" {
|
|
||||||
logger.Hint("fields option should not be empty, i.e. -fields=\"title:string,body:text\"")
|
|
||||||
logger.Fatal("Wrong number of arguments. Run: bee help generate")
|
|
||||||
}
|
|
||||||
sname := args[1]
|
|
||||||
generateModel(sname, fields.String(), currpath)
|
|
||||||
case "view":
|
|
||||||
if len(args) == 2 {
|
|
||||||
cname := args[1]
|
|
||||||
generateView(cname, currpath)
|
|
||||||
} else {
|
|
||||||
logger.Fatal("Wrong number of arguments. Run: bee help generate")
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
logger.Fatal("Command is missing")
|
|
||||||
}
|
|
||||||
logger.Successf("%s successfully generated!", strings.Title(gcmd))
|
|
||||||
return 0
|
|
||||||
}
|
|
950
g_docs.go
950
g_docs.go
|
@ -1,950 +0,0 @@
|
||||||
// Copyright 2013 bee authors
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"go/ast"
|
|
||||||
"go/parser"
|
|
||||||
"go/token"
|
|
||||||
"os"
|
|
||||||
"path"
|
|
||||||
"path/filepath"
|
|
||||||
"reflect"
|
|
||||||
"regexp"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"unicode"
|
|
||||||
|
|
||||||
"gopkg.in/yaml.v2"
|
|
||||||
|
|
||||||
"github.com/astaxie/beego/swagger"
|
|
||||||
"github.com/astaxie/beego/utils"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
ajson = "application/json"
|
|
||||||
axml = "application/xml"
|
|
||||||
aplain = "text/plain"
|
|
||||||
ahtml = "text/html"
|
|
||||||
)
|
|
||||||
|
|
||||||
var pkgCache map[string]struct{} //pkg:controller:function:comments comments: key:value
|
|
||||||
var controllerComments map[string]string
|
|
||||||
var importlist map[string]string
|
|
||||||
var controllerList map[string]map[string]*swagger.Item //controllername Paths items
|
|
||||||
var modelsList map[string]map[string]swagger.Schema
|
|
||||||
var rootapi swagger.Swagger
|
|
||||||
var astPkgs map[string]*ast.Package
|
|
||||||
|
|
||||||
// refer to builtin.go
|
|
||||||
var basicTypes = map[string]string{
|
|
||||||
"bool": "boolean:",
|
|
||||||
"uint": "integer:int32",
|
|
||||||
"uint8": "integer:int32",
|
|
||||||
"uint16": "integer:int32",
|
|
||||||
"uint32": "integer:int32",
|
|
||||||
"uint64": "integer:int64",
|
|
||||||
"int": "integer:int64",
|
|
||||||
"int8": "integer:int32",
|
|
||||||
"int16:int32": "integer:int32",
|
|
||||||
"int32": "integer:int32",
|
|
||||||
"int64": "integer:int64",
|
|
||||||
"uintptr": "integer:int64",
|
|
||||||
"float32": "number:float",
|
|
||||||
"float64": "number:double",
|
|
||||||
"string": "string:",
|
|
||||||
"complex64": "number:float",
|
|
||||||
"complex128": "number:double",
|
|
||||||
"byte": "string:byte",
|
|
||||||
"rune": "string:byte",
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
pkgCache = make(map[string]struct{})
|
|
||||||
controllerComments = make(map[string]string)
|
|
||||||
importlist = make(map[string]string)
|
|
||||||
controllerList = make(map[string]map[string]*swagger.Item)
|
|
||||||
modelsList = make(map[string]map[string]swagger.Schema)
|
|
||||||
astPkgs = map[string]*ast.Package{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func parsePackagesFromDir(dirpath string) {
|
|
||||||
c := make(chan error)
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
filepath.Walk(dirpath, func(fpath string, fileInfo os.FileInfo, err error) error {
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if !fileInfo.IsDir() {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
if !strings.Contains(fpath, "vendor") && !strings.Contains(fpath, "tests") {
|
|
||||||
err = parsePackageFromDir(fpath)
|
|
||||||
if err != nil {
|
|
||||||
// Send the error to through the channel and continue walking
|
|
||||||
c <- fmt.Errorf("Error while parsing directory: %s", err.Error())
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
close(c)
|
|
||||||
}()
|
|
||||||
|
|
||||||
for err := range c {
|
|
||||||
logger.Warnf("%s", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func parsePackageFromDir(path string) error {
|
|
||||||
fileSet := token.NewFileSet()
|
|
||||||
folderPkgs, err := parser.ParseDir(fileSet, path, func(info os.FileInfo) bool {
|
|
||||||
name := info.Name()
|
|
||||||
return !info.IsDir() && !strings.HasPrefix(name, ".") && strings.HasSuffix(name, ".go")
|
|
||||||
}, parser.ParseComments)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
for k, v := range folderPkgs {
|
|
||||||
astPkgs[k] = v
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func generateDocs(curpath string) {
|
|
||||||
fset := token.NewFileSet()
|
|
||||||
|
|
||||||
f, err := parser.ParseFile(fset, path.Join(curpath, "routers", "router.go"), nil, parser.ParseComments)
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("Error while parsing router.go: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
rootapi.Infos = swagger.Information{}
|
|
||||||
rootapi.SwaggerVersion = "2.0"
|
|
||||||
|
|
||||||
// Analyse API comments
|
|
||||||
if f.Comments != nil {
|
|
||||||
for _, c := range f.Comments {
|
|
||||||
for _, s := range strings.Split(c.Text(), "\n") {
|
|
||||||
if strings.HasPrefix(s, "@APIVersion") {
|
|
||||||
rootapi.Infos.Version = strings.TrimSpace(s[len("@APIVersion"):])
|
|
||||||
} else if strings.HasPrefix(s, "@Title") {
|
|
||||||
rootapi.Infos.Title = strings.TrimSpace(s[len("@Title"):])
|
|
||||||
} else if strings.HasPrefix(s, "@Description") {
|
|
||||||
rootapi.Infos.Description = strings.TrimSpace(s[len("@Description"):])
|
|
||||||
} else if strings.HasPrefix(s, "@TermsOfServiceUrl") {
|
|
||||||
rootapi.Infos.TermsOfService = strings.TrimSpace(s[len("@TermsOfServiceUrl"):])
|
|
||||||
} else if strings.HasPrefix(s, "@Contact") {
|
|
||||||
rootapi.Infos.Contact.EMail = strings.TrimSpace(s[len("@Contact"):])
|
|
||||||
} else if strings.HasPrefix(s, "@Name") {
|
|
||||||
rootapi.Infos.Contact.Name = strings.TrimSpace(s[len("@Name"):])
|
|
||||||
} else if strings.HasPrefix(s, "@URL") {
|
|
||||||
rootapi.Infos.Contact.URL = strings.TrimSpace(s[len("@URL"):])
|
|
||||||
} else if strings.HasPrefix(s, "@LicenseUrl") {
|
|
||||||
if rootapi.Infos.License == nil {
|
|
||||||
rootapi.Infos.License = &swagger.License{URL: strings.TrimSpace(s[len("@LicenseUrl"):])}
|
|
||||||
} else {
|
|
||||||
rootapi.Infos.License.URL = strings.TrimSpace(s[len("@LicenseUrl"):])
|
|
||||||
}
|
|
||||||
} else if strings.HasPrefix(s, "@License") {
|
|
||||||
if rootapi.Infos.License == nil {
|
|
||||||
rootapi.Infos.License = &swagger.License{Name: strings.TrimSpace(s[len("@License"):])}
|
|
||||||
} else {
|
|
||||||
rootapi.Infos.License.Name = strings.TrimSpace(s[len("@License"):])
|
|
||||||
}
|
|
||||||
} else if strings.HasPrefix(s, "@Schemes") {
|
|
||||||
rootapi.Schemes = strings.Split(strings.TrimSpace(s[len("@Schemes"):]), ",")
|
|
||||||
} else if strings.HasPrefix(s, "@Host") {
|
|
||||||
rootapi.Host = strings.TrimSpace(s[len("@Host"):])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Analyse controller package
|
|
||||||
for _, im := range f.Imports {
|
|
||||||
localName := ""
|
|
||||||
if im.Name != nil {
|
|
||||||
localName = im.Name.Name
|
|
||||||
}
|
|
||||||
analyseControllerPkg(localName, im.Path.Value)
|
|
||||||
}
|
|
||||||
for _, d := range f.Decls {
|
|
||||||
switch specDecl := d.(type) {
|
|
||||||
case *ast.FuncDecl:
|
|
||||||
for _, l := range specDecl.Body.List {
|
|
||||||
switch stmt := l.(type) {
|
|
||||||
case *ast.AssignStmt:
|
|
||||||
for _, l := range stmt.Rhs {
|
|
||||||
if v, ok := l.(*ast.CallExpr); ok {
|
|
||||||
// Analyse NewNamespace, it will return version and the subfunction
|
|
||||||
if selName := v.Fun.(*ast.SelectorExpr).Sel.String(); selName != "NewNamespace" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
version, params := analyseNewNamespace(v)
|
|
||||||
if rootapi.BasePath == "" && version != "" {
|
|
||||||
rootapi.BasePath = version
|
|
||||||
}
|
|
||||||
for _, p := range params {
|
|
||||||
switch pp := p.(type) {
|
|
||||||
case *ast.CallExpr:
|
|
||||||
controllerName := ""
|
|
||||||
if selname := pp.Fun.(*ast.SelectorExpr).Sel.String(); selname == "NSNamespace" {
|
|
||||||
s, params := analyseNewNamespace(pp)
|
|
||||||
for _, sp := range params {
|
|
||||||
switch pp := sp.(type) {
|
|
||||||
case *ast.CallExpr:
|
|
||||||
if pp.Fun.(*ast.SelectorExpr).Sel.String() == "NSInclude" {
|
|
||||||
controllerName = analyseNSInclude(s, pp)
|
|
||||||
if v, ok := controllerComments[controllerName]; ok {
|
|
||||||
rootapi.Tags = append(rootapi.Tags, swagger.Tag{
|
|
||||||
Name: strings.Trim(s, "/"),
|
|
||||||
Description: v,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if selname == "NSInclude" {
|
|
||||||
controllerName = analyseNSInclude("", pp)
|
|
||||||
if v, ok := controllerComments[controllerName]; ok {
|
|
||||||
rootapi.Tags = append(rootapi.Tags, swagger.Tag{
|
|
||||||
Name: controllerName, // if the NSInclude has no prefix, we use the controllername as the tag
|
|
||||||
Description: v,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
os.Mkdir(path.Join(curpath, "swagger"), 0755)
|
|
||||||
fd, err := os.Create(path.Join(curpath, "swagger", "swagger.json"))
|
|
||||||
fdyml, err := os.Create(path.Join(curpath, "swagger", "swagger.yml"))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
defer fdyml.Close()
|
|
||||||
defer fd.Close()
|
|
||||||
dt, err := json.MarshalIndent(rootapi, "", " ")
|
|
||||||
dtyml, erryml := yaml.Marshal(rootapi)
|
|
||||||
if err != nil || erryml != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
_, err = fd.Write(dt)
|
|
||||||
_, erryml = fdyml.Write(dtyml)
|
|
||||||
if err != nil || erryml != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// analyseNewNamespace returns version and the others params
|
|
||||||
func analyseNewNamespace(ce *ast.CallExpr) (first string, others []ast.Expr) {
|
|
||||||
for i, p := range ce.Args {
|
|
||||||
if i == 0 {
|
|
||||||
switch pp := p.(type) {
|
|
||||||
case *ast.BasicLit:
|
|
||||||
first = strings.Trim(pp.Value, `"`)
|
|
||||||
}
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
others = append(others, p)
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func analyseNSInclude(baseurl string, ce *ast.CallExpr) string {
|
|
||||||
cname := ""
|
|
||||||
for _, p := range ce.Args {
|
|
||||||
x := p.(*ast.UnaryExpr).X.(*ast.CompositeLit).Type.(*ast.SelectorExpr)
|
|
||||||
if v, ok := importlist[fmt.Sprint(x.X)]; ok {
|
|
||||||
cname = v + x.Sel.Name
|
|
||||||
}
|
|
||||||
if apis, ok := controllerList[cname]; ok {
|
|
||||||
for rt, item := range apis {
|
|
||||||
tag := ""
|
|
||||||
if baseurl != "" {
|
|
||||||
rt = baseurl + rt
|
|
||||||
tag = strings.Trim(baseurl, "/")
|
|
||||||
} else {
|
|
||||||
tag = cname
|
|
||||||
}
|
|
||||||
if item.Get != nil {
|
|
||||||
item.Get.Tags = []string{tag}
|
|
||||||
}
|
|
||||||
if item.Post != nil {
|
|
||||||
item.Post.Tags = []string{tag}
|
|
||||||
}
|
|
||||||
if item.Put != nil {
|
|
||||||
item.Put.Tags = []string{tag}
|
|
||||||
}
|
|
||||||
if item.Patch != nil {
|
|
||||||
item.Patch.Tags = []string{tag}
|
|
||||||
}
|
|
||||||
if item.Head != nil {
|
|
||||||
item.Head.Tags = []string{tag}
|
|
||||||
}
|
|
||||||
if item.Delete != nil {
|
|
||||||
item.Delete.Tags = []string{tag}
|
|
||||||
}
|
|
||||||
if item.Options != nil {
|
|
||||||
item.Options.Tags = []string{tag}
|
|
||||||
}
|
|
||||||
if len(rootapi.Paths) == 0 {
|
|
||||||
rootapi.Paths = make(map[string]*swagger.Item)
|
|
||||||
}
|
|
||||||
rt = urlReplace(rt)
|
|
||||||
rootapi.Paths[rt] = item
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return cname
|
|
||||||
}
|
|
||||||
|
|
||||||
func analyseControllerPkg(localName, pkgpath string) {
|
|
||||||
pkgpath = strings.Trim(pkgpath, "\"")
|
|
||||||
if isSystemPackage(pkgpath) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if pkgpath == "github.com/astaxie/beego" {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if localName != "" {
|
|
||||||
importlist[localName] = pkgpath
|
|
||||||
} else {
|
|
||||||
pps := strings.Split(pkgpath, "/")
|
|
||||||
importlist[pps[len(pps)-1]] = pkgpath
|
|
||||||
}
|
|
||||||
gopath := os.Getenv("GOPATH")
|
|
||||||
if gopath == "" {
|
|
||||||
logger.Fatal("GOPATH environment variable is not set or empty")
|
|
||||||
}
|
|
||||||
pkgRealpath := ""
|
|
||||||
|
|
||||||
wgopath := filepath.SplitList(gopath)
|
|
||||||
for _, wg := range wgopath {
|
|
||||||
wg, _ = filepath.EvalSymlinks(filepath.Join(wg, "src", pkgpath))
|
|
||||||
if utils.FileExists(wg) {
|
|
||||||
pkgRealpath = wg
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if pkgRealpath != "" {
|
|
||||||
if _, ok := pkgCache[pkgpath]; ok {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
pkgCache[pkgpath] = struct{}{}
|
|
||||||
} else {
|
|
||||||
logger.Fatalf("Package '%s' does not exist in the GOPATH", pkgpath)
|
|
||||||
}
|
|
||||||
|
|
||||||
fileSet := token.NewFileSet()
|
|
||||||
astPkgs, err := parser.ParseDir(fileSet, pkgRealpath, func(info os.FileInfo) bool {
|
|
||||||
name := info.Name()
|
|
||||||
return !info.IsDir() && !strings.HasPrefix(name, ".") && strings.HasSuffix(name, ".go")
|
|
||||||
}, parser.ParseComments)
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("Error while parsing dir at '%s': %s", pkgpath, err)
|
|
||||||
}
|
|
||||||
for _, pkg := range astPkgs {
|
|
||||||
for _, fl := range pkg.Files {
|
|
||||||
for _, d := range fl.Decls {
|
|
||||||
switch specDecl := d.(type) {
|
|
||||||
case *ast.FuncDecl:
|
|
||||||
if specDecl.Recv != nil && len(specDecl.Recv.List) > 0 {
|
|
||||||
if t, ok := specDecl.Recv.List[0].Type.(*ast.StarExpr); ok {
|
|
||||||
// Parse controller method
|
|
||||||
parserComments(specDecl.Doc, specDecl.Name.String(), fmt.Sprint(t.X), pkgpath)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
case *ast.GenDecl:
|
|
||||||
if specDecl.Tok == token.TYPE {
|
|
||||||
for _, s := range specDecl.Specs {
|
|
||||||
switch tp := s.(*ast.TypeSpec).Type.(type) {
|
|
||||||
case *ast.StructType:
|
|
||||||
_ = tp.Struct
|
|
||||||
// Parse controller definition comments
|
|
||||||
if strings.TrimSpace(specDecl.Doc.Text()) != "" {
|
|
||||||
controllerComments[pkgpath+s.(*ast.TypeSpec).Name.String()] = specDecl.Doc.Text()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func isSystemPackage(pkgpath string) bool {
|
|
||||||
goroot := os.Getenv("GOROOT")
|
|
||||||
if goroot == "" {
|
|
||||||
logger.Fatalf("GOROOT environment variable is not set or empty")
|
|
||||||
}
|
|
||||||
|
|
||||||
wg, _ := filepath.EvalSymlinks(filepath.Join(goroot, "src", "pkg", pkgpath))
|
|
||||||
if utils.FileExists(wg) {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
//TODO(zh):support go1.4
|
|
||||||
wg, _ = filepath.EvalSymlinks(filepath.Join(goroot, "src", pkgpath))
|
|
||||||
if utils.FileExists(wg) {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func peekNextSplitString(ss string) (s string, spacePos int) {
|
|
||||||
spacePos = strings.IndexFunc(ss, unicode.IsSpace)
|
|
||||||
if spacePos < 0 {
|
|
||||||
s = ss
|
|
||||||
spacePos = len(ss)
|
|
||||||
} else {
|
|
||||||
s = strings.TrimSpace(ss[:spacePos])
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// parse the func comments
|
|
||||||
func parserComments(comments *ast.CommentGroup, funcName, controllerName, pkgpath string) error {
|
|
||||||
var routerPath string
|
|
||||||
var HTTPMethod string
|
|
||||||
opts := swagger.Operation{
|
|
||||||
Responses: make(map[string]swagger.Response),
|
|
||||||
}
|
|
||||||
if comments != nil && comments.List != nil {
|
|
||||||
for _, c := range comments.List {
|
|
||||||
t := strings.TrimSpace(strings.TrimLeft(c.Text, "//"))
|
|
||||||
if strings.HasPrefix(t, "@router") {
|
|
||||||
elements := strings.TrimSpace(t[len("@router"):])
|
|
||||||
e1 := strings.SplitN(elements, " ", 2)
|
|
||||||
if len(e1) < 1 {
|
|
||||||
return errors.New("you should has router infomation")
|
|
||||||
}
|
|
||||||
routerPath = e1[0]
|
|
||||||
if len(e1) == 2 && e1[1] != "" {
|
|
||||||
e1 = strings.SplitN(e1[1], " ", 2)
|
|
||||||
HTTPMethod = strings.ToUpper(strings.Trim(e1[0], "[]"))
|
|
||||||
} else {
|
|
||||||
HTTPMethod = "GET"
|
|
||||||
}
|
|
||||||
} else if strings.HasPrefix(t, "@Title") {
|
|
||||||
opts.OperationID = controllerName + "." + strings.TrimSpace(t[len("@Title"):])
|
|
||||||
} else if strings.HasPrefix(t, "@Description") {
|
|
||||||
opts.Description = strings.TrimSpace(t[len("@Description"):])
|
|
||||||
} else if strings.HasPrefix(t, "@Summary") {
|
|
||||||
opts.Summary = strings.TrimSpace(t[len("@Summary"):])
|
|
||||||
} else if strings.HasPrefix(t, "@Success") {
|
|
||||||
ss := strings.TrimSpace(t[len("@Success"):])
|
|
||||||
rs := swagger.Response{}
|
|
||||||
respCode, pos := peekNextSplitString(ss)
|
|
||||||
ss = strings.TrimSpace(ss[pos:])
|
|
||||||
respType, pos := peekNextSplitString(ss)
|
|
||||||
if respType == "{object}" || respType == "{array}" {
|
|
||||||
isArray := respType == "{array}"
|
|
||||||
ss = strings.TrimSpace(ss[pos:])
|
|
||||||
schemaName, pos := peekNextSplitString(ss)
|
|
||||||
if schemaName == "" {
|
|
||||||
logger.Fatalf("[%s.%s] Schema must follow {object} or {array}", controllerName, funcName)
|
|
||||||
}
|
|
||||||
if strings.HasPrefix(schemaName, "[]") {
|
|
||||||
schemaName = schemaName[2:]
|
|
||||||
isArray = true
|
|
||||||
}
|
|
||||||
schema := swagger.Schema{}
|
|
||||||
if sType, ok := basicTypes[schemaName]; ok {
|
|
||||||
typeFormat := strings.Split(sType, ":")
|
|
||||||
schema.Type = typeFormat[0]
|
|
||||||
schema.Format = typeFormat[1]
|
|
||||||
} else {
|
|
||||||
m, mod, realTypes := getModel(schemaName)
|
|
||||||
schema.Ref = "#/definitions/" + m
|
|
||||||
if _, ok := modelsList[pkgpath+controllerName]; !ok {
|
|
||||||
modelsList[pkgpath+controllerName] = make(map[string]swagger.Schema, 0)
|
|
||||||
}
|
|
||||||
modelsList[pkgpath+controllerName][schemaName] = mod
|
|
||||||
appendModels(pkgpath, controllerName, realTypes)
|
|
||||||
}
|
|
||||||
if isArray {
|
|
||||||
rs.Schema = &swagger.Schema{
|
|
||||||
Type: "array",
|
|
||||||
Items: &schema,
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
rs.Schema = &schema
|
|
||||||
}
|
|
||||||
rs.Description = strings.TrimSpace(ss[pos:])
|
|
||||||
} else {
|
|
||||||
rs.Description = strings.TrimSpace(ss)
|
|
||||||
}
|
|
||||||
opts.Responses[respCode] = rs
|
|
||||||
} else if strings.HasPrefix(t, "@Param") {
|
|
||||||
para := swagger.Parameter{}
|
|
||||||
p := getparams(strings.TrimSpace(t[len("@Param "):]))
|
|
||||||
if len(p) < 4 {
|
|
||||||
logger.Fatal(controllerName + "_" + funcName + "'s comments @Param should have at least 4 params")
|
|
||||||
}
|
|
||||||
para.Name = p[0]
|
|
||||||
switch p[1] {
|
|
||||||
case "query":
|
|
||||||
fallthrough
|
|
||||||
case "header":
|
|
||||||
fallthrough
|
|
||||||
case "path":
|
|
||||||
fallthrough
|
|
||||||
case "formData":
|
|
||||||
fallthrough
|
|
||||||
case "body":
|
|
||||||
break
|
|
||||||
default:
|
|
||||||
logger.Warnf("[%s.%s] Unknown param location: %s. Possible values are `query`, `header`, `path`, `formData` or `body`.\n", controllerName, funcName, p[1])
|
|
||||||
}
|
|
||||||
para.In = p[1]
|
|
||||||
pp := strings.Split(p[2], ".")
|
|
||||||
typ := pp[len(pp)-1]
|
|
||||||
if len(pp) >= 2 {
|
|
||||||
m, mod, realTypes := getModel(p[2])
|
|
||||||
para.Schema = &swagger.Schema{
|
|
||||||
Ref: "#/definitions/" + m,
|
|
||||||
}
|
|
||||||
if _, ok := modelsList[pkgpath+controllerName]; !ok {
|
|
||||||
modelsList[pkgpath+controllerName] = make(map[string]swagger.Schema, 0)
|
|
||||||
}
|
|
||||||
modelsList[pkgpath+controllerName][typ] = mod
|
|
||||||
appendModels(pkgpath, controllerName, realTypes)
|
|
||||||
} else {
|
|
||||||
isArray := false
|
|
||||||
paraType := ""
|
|
||||||
paraFormat := ""
|
|
||||||
if strings.HasPrefix(typ, "[]") {
|
|
||||||
typ = typ[2:]
|
|
||||||
isArray = true
|
|
||||||
}
|
|
||||||
if typ == "string" || typ == "number" || typ == "integer" || typ == "boolean" ||
|
|
||||||
typ == "array" || typ == "file" {
|
|
||||||
paraType = typ
|
|
||||||
} else if sType, ok := basicTypes[typ]; ok {
|
|
||||||
typeFormat := strings.Split(sType, ":")
|
|
||||||
paraType = typeFormat[0]
|
|
||||||
paraFormat = typeFormat[1]
|
|
||||||
} else {
|
|
||||||
logger.Warnf("[%s.%s] Unknown param type: %s\n", controllerName, funcName, typ)
|
|
||||||
}
|
|
||||||
if isArray {
|
|
||||||
para.Type = "array"
|
|
||||||
para.Items = &swagger.ParameterItems{
|
|
||||||
Type: paraType,
|
|
||||||
Format: paraFormat,
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
para.Type = paraType
|
|
||||||
para.Format = paraFormat
|
|
||||||
}
|
|
||||||
}
|
|
||||||
switch len(p) {
|
|
||||||
case 5:
|
|
||||||
para.Required, _ = strconv.ParseBool(p[3])
|
|
||||||
para.Description = strings.Trim(p[4], `" `)
|
|
||||||
case 6:
|
|
||||||
para.Default = str2RealType(p[3], para.Type)
|
|
||||||
para.Required, _ = strconv.ParseBool(p[4])
|
|
||||||
para.Description = strings.Trim(p[5], `" `)
|
|
||||||
default:
|
|
||||||
para.Description = strings.Trim(p[3], `" `)
|
|
||||||
}
|
|
||||||
opts.Parameters = append(opts.Parameters, para)
|
|
||||||
} else if strings.HasPrefix(t, "@Failure") {
|
|
||||||
rs := swagger.Response{}
|
|
||||||
st := strings.TrimSpace(t[len("@Failure"):])
|
|
||||||
var cd []rune
|
|
||||||
var start bool
|
|
||||||
for i, s := range st {
|
|
||||||
if unicode.IsSpace(s) {
|
|
||||||
if start {
|
|
||||||
rs.Description = strings.TrimSpace(st[i+1:])
|
|
||||||
break
|
|
||||||
} else {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
start = true
|
|
||||||
cd = append(cd, s)
|
|
||||||
}
|
|
||||||
opts.Responses[string(cd)] = rs
|
|
||||||
} else if strings.HasPrefix(t, "@Deprecated") {
|
|
||||||
opts.Deprecated, _ = strconv.ParseBool(strings.TrimSpace(t[len("@Deprecated"):]))
|
|
||||||
} else if strings.HasPrefix(t, "@Accept") {
|
|
||||||
accepts := strings.Split(strings.TrimSpace(strings.TrimSpace(t[len("@Accept"):])), ",")
|
|
||||||
for _, a := range accepts {
|
|
||||||
switch a {
|
|
||||||
case "json":
|
|
||||||
opts.Consumes = append(opts.Consumes, ajson)
|
|
||||||
opts.Produces = append(opts.Produces, ajson)
|
|
||||||
case "xml":
|
|
||||||
opts.Consumes = append(opts.Consumes, axml)
|
|
||||||
opts.Produces = append(opts.Produces, axml)
|
|
||||||
case "plain":
|
|
||||||
opts.Consumes = append(opts.Consumes, aplain)
|
|
||||||
opts.Produces = append(opts.Produces, aplain)
|
|
||||||
case "html":
|
|
||||||
opts.Consumes = append(opts.Consumes, ahtml)
|
|
||||||
opts.Produces = append(opts.Produces, ahtml)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if routerPath != "" {
|
|
||||||
var item *swagger.Item
|
|
||||||
if itemList, ok := controllerList[pkgpath+controllerName]; ok {
|
|
||||||
if it, ok := itemList[routerPath]; !ok {
|
|
||||||
item = &swagger.Item{}
|
|
||||||
} else {
|
|
||||||
item = it
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
controllerList[pkgpath+controllerName] = make(map[string]*swagger.Item)
|
|
||||||
item = &swagger.Item{}
|
|
||||||
}
|
|
||||||
switch HTTPMethod {
|
|
||||||
case "GET":
|
|
||||||
item.Get = &opts
|
|
||||||
case "POST":
|
|
||||||
item.Post = &opts
|
|
||||||
case "PUT":
|
|
||||||
item.Put = &opts
|
|
||||||
case "PATCH":
|
|
||||||
item.Patch = &opts
|
|
||||||
case "DELETE":
|
|
||||||
item.Delete = &opts
|
|
||||||
case "HEAD":
|
|
||||||
item.Head = &opts
|
|
||||||
case "OPTIONS":
|
|
||||||
item.Options = &opts
|
|
||||||
}
|
|
||||||
controllerList[pkgpath+controllerName][routerPath] = item
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// analisys params return []string
|
|
||||||
// @Param query form string true "The email for login"
|
|
||||||
// [query form string true "The email for login"]
|
|
||||||
func getparams(str string) []string {
|
|
||||||
var s []rune
|
|
||||||
var j int
|
|
||||||
var start bool
|
|
||||||
var r []string
|
|
||||||
var quoted int8
|
|
||||||
for _, c := range []rune(str) {
|
|
||||||
if unicode.IsSpace(c) && quoted == 0 {
|
|
||||||
if !start {
|
|
||||||
continue
|
|
||||||
} else {
|
|
||||||
start = false
|
|
||||||
j++
|
|
||||||
r = append(r, string(s))
|
|
||||||
s = make([]rune, 0)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
start = true
|
|
||||||
if c == '"' {
|
|
||||||
quoted ^= 1
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
s = append(s, c)
|
|
||||||
}
|
|
||||||
if len(s) > 0 {
|
|
||||||
r = append(r, string(s))
|
|
||||||
}
|
|
||||||
return r
|
|
||||||
}
|
|
||||||
|
|
||||||
func getModel(str string) (objectname string, m swagger.Schema, realTypes []string) {
|
|
||||||
strs := strings.Split(str, ".")
|
|
||||||
objectname = strs[len(strs)-1]
|
|
||||||
packageName := ""
|
|
||||||
m.Type = "object"
|
|
||||||
for _, pkg := range astPkgs {
|
|
||||||
for _, fl := range pkg.Files {
|
|
||||||
for k, d := range fl.Scope.Objects {
|
|
||||||
if d.Kind == ast.Typ {
|
|
||||||
if k != objectname {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
packageName = pkg.Name
|
|
||||||
parseObject(d, k, &m, &realTypes, astPkgs, pkg.Name)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if m.Title == "" {
|
|
||||||
logger.Warnf("Cannot find the object: %s", str)
|
|
||||||
// TODO remove when all type have been supported
|
|
||||||
//os.Exit(1)
|
|
||||||
}
|
|
||||||
if len(rootapi.Definitions) == 0 {
|
|
||||||
rootapi.Definitions = make(map[string]swagger.Schema)
|
|
||||||
}
|
|
||||||
objectname = packageName + "." + objectname
|
|
||||||
rootapi.Definitions[objectname] = m
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseObject(d *ast.Object, k string, m *swagger.Schema, realTypes *[]string, astPkgs map[string]*ast.Package, packageName string) {
|
|
||||||
ts, ok := d.Decl.(*ast.TypeSpec)
|
|
||||||
if !ok {
|
|
||||||
logger.Fatalf("Unknown type without TypeSec: %v\n", d)
|
|
||||||
}
|
|
||||||
// TODO support other types, such as `ArrayType`, `MapType`, `InterfaceType` etc...
|
|
||||||
st, ok := ts.Type.(*ast.StructType)
|
|
||||||
if !ok {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
m.Title = k
|
|
||||||
if st.Fields.List != nil {
|
|
||||||
m.Properties = make(map[string]swagger.Propertie)
|
|
||||||
for _, field := range st.Fields.List {
|
|
||||||
realType := ""
|
|
||||||
isSlice, realType, sType := typeAnalyser(field)
|
|
||||||
if (isSlice && isBasicType(realType)) || sType == "object" {
|
|
||||||
if len(strings.Split(realType, " ")) > 1 {
|
|
||||||
realType = strings.Replace(realType, " ", ".", -1)
|
|
||||||
realType = strings.Replace(realType, "&", "", -1)
|
|
||||||
realType = strings.Replace(realType, "{", "", -1)
|
|
||||||
realType = strings.Replace(realType, "}", "", -1)
|
|
||||||
} else {
|
|
||||||
realType = packageName + "." + realType
|
|
||||||
}
|
|
||||||
}
|
|
||||||
*realTypes = append(*realTypes, realType)
|
|
||||||
mp := swagger.Propertie{}
|
|
||||||
if isSlice {
|
|
||||||
mp.Type = "array"
|
|
||||||
if isBasicType(realType) {
|
|
||||||
typeFormat := strings.Split(sType, ":")
|
|
||||||
mp.Items = &swagger.Propertie{
|
|
||||||
Type: typeFormat[0],
|
|
||||||
Format: typeFormat[1],
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
mp.Items = &swagger.Propertie{
|
|
||||||
Ref: "#/definitions/" + realType,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if sType == "object" {
|
|
||||||
mp.Ref = "#/definitions/" + realType
|
|
||||||
} else if isBasicType(realType) {
|
|
||||||
typeFormat := strings.Split(sType, ":")
|
|
||||||
mp.Type = typeFormat[0]
|
|
||||||
mp.Format = typeFormat[1]
|
|
||||||
} else if realType == "map" {
|
|
||||||
typeFormat := strings.Split(sType, ":")
|
|
||||||
mp.AdditionalProperties = &swagger.Propertie{
|
|
||||||
Type: typeFormat[0],
|
|
||||||
Format: typeFormat[1],
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if field.Names != nil {
|
|
||||||
|
|
||||||
// set property name as field name
|
|
||||||
var name = field.Names[0].Name
|
|
||||||
|
|
||||||
// if no tag skip tag processing
|
|
||||||
if field.Tag == nil {
|
|
||||||
m.Properties[name] = mp
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
var tagValues []string
|
|
||||||
|
|
||||||
stag := reflect.StructTag(strings.Trim(field.Tag.Value, "`"))
|
|
||||||
|
|
||||||
defaultValue := stag.Get("doc")
|
|
||||||
if defaultValue != "" {
|
|
||||||
r, _ := regexp.Compile(`default\((.*)\)`)
|
|
||||||
if r.MatchString(defaultValue) {
|
|
||||||
res := r.FindStringSubmatch(defaultValue)
|
|
||||||
mp.Default = str2RealType(res[1], realType)
|
|
||||||
|
|
||||||
} else {
|
|
||||||
logger.Warnf("Invalid default value: %s", defaultValue)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
tag := stag.Get("json")
|
|
||||||
|
|
||||||
if tag != "" {
|
|
||||||
tagValues = strings.Split(tag, ",")
|
|
||||||
}
|
|
||||||
|
|
||||||
// dont add property if json tag first value is "-"
|
|
||||||
if len(tagValues) == 0 || tagValues[0] != "-" {
|
|
||||||
|
|
||||||
// set property name to the left most json tag value only if is not omitempty
|
|
||||||
if len(tagValues) > 0 && tagValues[0] != "omitempty" {
|
|
||||||
name = tagValues[0]
|
|
||||||
}
|
|
||||||
|
|
||||||
if thrifttag := stag.Get("thrift"); thrifttag != "" {
|
|
||||||
ts := strings.Split(thrifttag, ",")
|
|
||||||
if ts[0] != "" {
|
|
||||||
name = ts[0]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if required := stag.Get("required"); required != "" {
|
|
||||||
m.Required = append(m.Required, name)
|
|
||||||
}
|
|
||||||
if desc := stag.Get("description"); desc != "" {
|
|
||||||
mp.Description = desc
|
|
||||||
}
|
|
||||||
|
|
||||||
m.Properties[name] = mp
|
|
||||||
}
|
|
||||||
if ignore := stag.Get("ignore"); ignore != "" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
for _, pkg := range astPkgs {
|
|
||||||
for _, fl := range pkg.Files {
|
|
||||||
for nameOfObj, obj := range fl.Scope.Objects {
|
|
||||||
if obj.Name == fmt.Sprint(field.Type) {
|
|
||||||
parseObject(obj, nameOfObj, m, realTypes, astPkgs, pkg.Name)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func typeAnalyser(f *ast.Field) (isSlice bool, realType, swaggerType string) {
|
|
||||||
if arr, ok := f.Type.(*ast.ArrayType); ok {
|
|
||||||
if isBasicType(fmt.Sprint(arr.Elt)) {
|
|
||||||
return false, fmt.Sprintf("[]%v", arr.Elt), basicTypes[fmt.Sprint(arr.Elt)]
|
|
||||||
}
|
|
||||||
if mp, ok := arr.Elt.(*ast.MapType); ok {
|
|
||||||
return false, fmt.Sprintf("map[%v][%v]", mp.Key, mp.Value), "object"
|
|
||||||
}
|
|
||||||
if star, ok := arr.Elt.(*ast.StarExpr); ok {
|
|
||||||
return true, fmt.Sprint(star.X), "object"
|
|
||||||
}
|
|
||||||
return true, fmt.Sprint(arr.Elt), "object"
|
|
||||||
}
|
|
||||||
switch t := f.Type.(type) {
|
|
||||||
case *ast.StarExpr:
|
|
||||||
return false, fmt.Sprint(t.X), "object"
|
|
||||||
case *ast.MapType:
|
|
||||||
val := fmt.Sprintf("%v", t.Value)
|
|
||||||
if isBasicType(val) {
|
|
||||||
return false, "map", basicTypes[val]
|
|
||||||
}
|
|
||||||
return false, val, "object"
|
|
||||||
}
|
|
||||||
if k, ok := basicTypes[fmt.Sprint(f.Type)]; ok {
|
|
||||||
return false, fmt.Sprint(f.Type), k
|
|
||||||
}
|
|
||||||
return false, fmt.Sprint(f.Type), "object"
|
|
||||||
}
|
|
||||||
|
|
||||||
func isBasicType(Type string) bool {
|
|
||||||
if _, ok := basicTypes[Type]; ok {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
// regexp get json tag
|
|
||||||
func grepJSONTag(tag string) string {
|
|
||||||
r, _ := regexp.Compile(`json:"([^"]*)"`)
|
|
||||||
matches := r.FindAllStringSubmatch(tag, -1)
|
|
||||||
if len(matches) > 0 {
|
|
||||||
return matches[0][1]
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
// append models
|
|
||||||
func appendModels(pkgpath, controllerName string, realTypes []string) {
|
|
||||||
for _, realType := range realTypes {
|
|
||||||
if realType != "" && !isBasicType(strings.TrimLeft(realType, "[]")) &&
|
|
||||||
!strings.HasPrefix(realType, "map") && !strings.HasPrefix(realType, "&") {
|
|
||||||
if _, ok := modelsList[pkgpath+controllerName][realType]; ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
_, mod, newRealTypes := getModel(realType)
|
|
||||||
modelsList[pkgpath+controllerName][realType] = mod
|
|
||||||
appendModels(pkgpath, controllerName, newRealTypes)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func urlReplace(src string) string {
|
|
||||||
pt := strings.Split(src, "/")
|
|
||||||
for i, p := range pt {
|
|
||||||
if len(p) > 0 {
|
|
||||||
if p[0] == ':' {
|
|
||||||
pt[i] = "{" + p[1:] + "}"
|
|
||||||
} else if p[0] == '?' && p[1] == ':' {
|
|
||||||
pt[i] = "{" + p[2:] + "}"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return strings.Join(pt, "/")
|
|
||||||
}
|
|
||||||
|
|
||||||
func str2RealType(s string, typ string) interface{} {
|
|
||||||
var err error
|
|
||||||
var ret interface{}
|
|
||||||
|
|
||||||
switch typ {
|
|
||||||
case "int", "int64", "int32", "int16", "int8":
|
|
||||||
ret, err = strconv.Atoi(s)
|
|
||||||
case "bool":
|
|
||||||
ret, err = strconv.ParseBool(s)
|
|
||||||
case "float64":
|
|
||||||
ret, err = strconv.ParseFloat(s, 64)
|
|
||||||
case "float32":
|
|
||||||
ret, err = strconv.ParseFloat(s, 32)
|
|
||||||
default:
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
logger.Warnf("Invalid default value type '%s': %s", typ, s)
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret
|
|
||||||
}
|
|
|
@ -1,48 +0,0 @@
|
||||||
package main
|
|
||||||
|
|
||||||
import "strings"
|
|
||||||
|
|
||||||
func generateScaffold(sname, fields, currpath, driver, conn string) {
|
|
||||||
logger.Infof("Do you want to create a '%s' model? [Yes|No] ", sname)
|
|
||||||
|
|
||||||
// Generate the model
|
|
||||||
if askForConfirmation() {
|
|
||||||
generateModel(sname, fields, currpath)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generate the controller
|
|
||||||
logger.Infof("Do you want to create a '%s' controller? [Yes|No] ", sname)
|
|
||||||
if askForConfirmation() {
|
|
||||||
generateController(sname, currpath)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generate the views
|
|
||||||
logger.Infof("Do you want to create views for this '%s' resource? [Yes|No] ", sname)
|
|
||||||
if askForConfirmation() {
|
|
||||||
generateView(sname, currpath)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generate a migration
|
|
||||||
logger.Infof("Do you want to create a '%s' migration and schema for this resource? [Yes|No] ", sname)
|
|
||||||
if askForConfirmation() {
|
|
||||||
upsql := ""
|
|
||||||
downsql := ""
|
|
||||||
if fields != "" {
|
|
||||||
dbMigrator := newDBDriver()
|
|
||||||
upsql = dbMigrator.generateCreateUp(sname)
|
|
||||||
downsql = dbMigrator.generateCreateDown(sname)
|
|
||||||
//todo remove
|
|
||||||
//if driver == "" {
|
|
||||||
// downsql = strings.Replace(downsql, "`", "", -1)
|
|
||||||
//}
|
|
||||||
}
|
|
||||||
generateMigration(sname, upsql, downsql, currpath)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Run the migration
|
|
||||||
logger.Infof("Do you want to migrate the database? [Yes|No] ")
|
|
||||||
if askForConfirmation() {
|
|
||||||
migrateUpdate(currpath, driver, conn)
|
|
||||||
}
|
|
||||||
logger.Successf("All done! Don't forget to add beego.Router(\"/%s\" ,&controllers.%sController{}) to routers/route.go\n", sname, strings.Title(sname))
|
|
||||||
}
|
|
30
generate/g.go
Normal file
30
generate/g.go
Normal file
|
@ -0,0 +1,30 @@
|
||||||
|
// Copyright 2013 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
|
||||||
|
package generate
|
||||||
|
|
||||||
|
import "github.com/beego/bee/v2/utils"
|
||||||
|
|
||||||
|
var SQLDriver utils.DocValue
|
||||||
|
var SQLConn utils.DocValue
|
||||||
|
var Level utils.DocValue
|
||||||
|
var Tables utils.DocValue
|
||||||
|
var Fields utils.DocValue
|
||||||
|
var DDL utils.DocValue
|
||||||
|
|
||||||
|
|
||||||
|
// bee generate routers
|
||||||
|
var ControllerDirectory utils.DocValue
|
||||||
|
var RoutersFile utils.DocValue
|
||||||
|
var RouterPkg utils.DocValue
|
|
@ -12,17 +12,22 @@
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
|
|
||||||
package main
|
package generate
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bufio"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
_ "github.com/go-sql-driver/mysql"
|
_ "github.com/go-sql-driver/mysql"
|
||||||
_ "github.com/lib/pq"
|
_ "github.com/lib/pq"
|
||||||
)
|
)
|
||||||
|
@ -98,6 +103,7 @@ var typeMappingMysql = map[string]string{
|
||||||
"decimal": "float64",
|
"decimal": "float64",
|
||||||
"binary": "string", // binary
|
"binary": "string", // binary
|
||||||
"varbinary": "string",
|
"varbinary": "string",
|
||||||
|
"year": "int16",
|
||||||
}
|
}
|
||||||
|
|
||||||
// typeMappingPostgres maps SQL data type to corresponding Go data type
|
// typeMappingPostgres maps SQL data type to corresponding Go data type
|
||||||
|
@ -179,11 +185,12 @@ type OrmTag struct {
|
||||||
RelFk bool
|
RelFk bool
|
||||||
ReverseMany bool
|
ReverseMany bool
|
||||||
RelM2M bool
|
RelM2M bool
|
||||||
|
Comment string //column comment
|
||||||
}
|
}
|
||||||
|
|
||||||
// String returns the source code string for the Table struct
|
// String returns the source code string for the Table struct
|
||||||
func (tb *Table) String() string {
|
func (tb *Table) String() string {
|
||||||
rv := fmt.Sprintf("type %s struct {\n", camelCase(tb.Name))
|
rv := fmt.Sprintf("type %s struct {\n", utils.CamelCase(tb.Name))
|
||||||
for _, v := range tb.Columns {
|
for _, v := range tb.Columns {
|
||||||
rv += v.String() + "\n"
|
rv += v.String() + "\n"
|
||||||
}
|
}
|
||||||
|
@ -252,10 +259,13 @@ func (tag *OrmTag) String() string {
|
||||||
if len(ormOptions) == 0 {
|
if len(ormOptions) == 0 {
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
|
if tag.Comment != "" {
|
||||||
|
return fmt.Sprintf("`orm:\"%s\" description:\"%s\"`", strings.Join(ormOptions, ";"), tag.Comment)
|
||||||
|
}
|
||||||
return fmt.Sprintf("`orm:\"%s\"`", strings.Join(ormOptions, ";"))
|
return fmt.Sprintf("`orm:\"%s\"`", strings.Join(ormOptions, ";"))
|
||||||
}
|
}
|
||||||
|
|
||||||
func generateAppcode(driver, connStr, level, tables, currpath string) {
|
func GenerateAppcode(driver, connStr, level, tables, currpath string) {
|
||||||
var mode byte
|
var mode byte
|
||||||
switch level {
|
switch level {
|
||||||
case "1":
|
case "1":
|
||||||
|
@ -265,7 +275,7 @@ func generateAppcode(driver, connStr, level, tables, currpath string) {
|
||||||
case "3":
|
case "3":
|
||||||
mode = OModel | OController | ORouter
|
mode = OModel | OController | ORouter
|
||||||
default:
|
default:
|
||||||
logger.Fatal("Invalid level value. Must be either \"1\", \"2\", or \"3\"")
|
beeLogger.Log.Fatal("Invalid level value. Must be either \"1\", \"2\", or \"3\"")
|
||||||
}
|
}
|
||||||
var selectedTables map[string]bool
|
var selectedTables map[string]bool
|
||||||
if tables != "" {
|
if tables != "" {
|
||||||
|
@ -278,9 +288,9 @@ func generateAppcode(driver, connStr, level, tables, currpath string) {
|
||||||
case "mysql":
|
case "mysql":
|
||||||
case "postgres":
|
case "postgres":
|
||||||
case "sqlite":
|
case "sqlite":
|
||||||
logger.Fatal("Generating app code from SQLite database is not supported yet.")
|
beeLogger.Log.Fatal("Generating app code from SQLite database is not supported yet.")
|
||||||
default:
|
default:
|
||||||
logger.Fatal("Unknown database driver. Must be either \"mysql\", \"postgres\" or \"sqlite\"")
|
beeLogger.Log.Fatal("Unknown database driver. Must be either \"mysql\", \"postgres\" or \"sqlite\"")
|
||||||
}
|
}
|
||||||
gen(driver, connStr, mode, selectedTables, currpath)
|
gen(driver, connStr, mode, selectedTables, currpath)
|
||||||
}
|
}
|
||||||
|
@ -290,12 +300,19 @@ func generateAppcode(driver, connStr, level, tables, currpath string) {
|
||||||
func gen(dbms, connStr string, mode byte, selectedTableNames map[string]bool, apppath string) {
|
func gen(dbms, connStr string, mode byte, selectedTableNames map[string]bool, apppath string) {
|
||||||
db, err := sql.Open(dbms, connStr)
|
db, err := sql.Open(dbms, connStr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Could not connect to '%s' database using '%s': %s", dbms, connStr, err)
|
beeLogger.Log.Fatalf("Could not connect to '%s' database using '%s': %s", dbms, connStr, err)
|
||||||
}
|
}
|
||||||
defer db.Close()
|
defer db.Close()
|
||||||
if trans, ok := dbDriver[dbms]; ok {
|
if trans, ok := dbDriver[dbms]; ok {
|
||||||
logger.Info("Analyzing database tables...")
|
beeLogger.Log.Info("Analyzing database tables...")
|
||||||
tableNames := trans.GetTableNames(db)
|
var tableNames []string
|
||||||
|
if len(selectedTableNames) != 0 {
|
||||||
|
for tableName := range selectedTableNames {
|
||||||
|
tableNames = append(tableNames, tableName)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
tableNames = trans.GetTableNames(db)
|
||||||
|
}
|
||||||
tables := getTableObjects(tableNames, db, trans)
|
tables := getTableObjects(tableNames, db, trans)
|
||||||
mvcPath := new(MvcPath)
|
mvcPath := new(MvcPath)
|
||||||
mvcPath.ModelPath = path.Join(apppath, "models")
|
mvcPath.ModelPath = path.Join(apppath, "models")
|
||||||
|
@ -303,9 +320,9 @@ func gen(dbms, connStr string, mode byte, selectedTableNames map[string]bool, ap
|
||||||
mvcPath.RouterPath = path.Join(apppath, "routers")
|
mvcPath.RouterPath = path.Join(apppath, "routers")
|
||||||
createPaths(mode, mvcPath)
|
createPaths(mode, mvcPath)
|
||||||
pkgPath := getPackagePath(apppath)
|
pkgPath := getPackagePath(apppath)
|
||||||
writeSourceFiles(pkgPath, tables, mode, mvcPath, selectedTableNames)
|
writeSourceFiles(pkgPath, tables, mode, mvcPath)
|
||||||
} else {
|
} else {
|
||||||
logger.Fatalf("Generating app code from '%s' database is not supported yet.", dbms)
|
beeLogger.Log.Fatalf("Generating app code from '%s' database is not supported yet.", dbms)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -313,13 +330,13 @@ func gen(dbms, connStr string, mode byte, selectedTableNames map[string]bool, ap
|
||||||
func (*MysqlDB) GetTableNames(db *sql.DB) (tables []string) {
|
func (*MysqlDB) GetTableNames(db *sql.DB) (tables []string) {
|
||||||
rows, err := db.Query("SHOW TABLES")
|
rows, err := db.Query("SHOW TABLES")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Could not show tables: %s", err)
|
beeLogger.Log.Fatalf("Could not show tables: %s", err)
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
defer rows.Close()
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
var name string
|
var name string
|
||||||
if err := rows.Scan(&name); err != nil {
|
if err := rows.Scan(&name); err != nil {
|
||||||
logger.Fatalf("Could not show tables: %s", err)
|
beeLogger.Log.Fatalf("Could not show tables: %s", err)
|
||||||
}
|
}
|
||||||
tables = append(tables, name)
|
tables = append(tables, name)
|
||||||
}
|
}
|
||||||
|
@ -362,12 +379,12 @@ func (*MysqlDB) GetConstraints(db *sql.DB, table *Table, blackList map[string]bo
|
||||||
c.table_schema = database() AND c.table_name = ? AND u.table_schema = database() AND u.table_name = ?`,
|
c.table_schema = database() AND c.table_name = ? AND u.table_schema = database() AND u.table_name = ?`,
|
||||||
table.Name, table.Name) // u.position_in_unique_constraint,
|
table.Name, table.Name) // u.position_in_unique_constraint,
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatal("Could not query INFORMATION_SCHEMA for PK/UK/FK information")
|
beeLogger.Log.Fatal("Could not query INFORMATION_SCHEMA for PK/UK/FK information")
|
||||||
}
|
}
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
var constraintTypeBytes, columnNameBytes, refTableSchemaBytes, refTableNameBytes, refColumnNameBytes, refOrdinalPosBytes []byte
|
var constraintTypeBytes, columnNameBytes, refTableSchemaBytes, refTableNameBytes, refColumnNameBytes, refOrdinalPosBytes []byte
|
||||||
if err := rows.Scan(&constraintTypeBytes, &columnNameBytes, &refTableSchemaBytes, &refTableNameBytes, &refColumnNameBytes, &refOrdinalPosBytes); err != nil {
|
if err := rows.Scan(&constraintTypeBytes, &columnNameBytes, &refTableSchemaBytes, &refTableNameBytes, &refColumnNameBytes, &refOrdinalPosBytes); err != nil {
|
||||||
logger.Fatal("Could not read INFORMATION_SCHEMA for PK/UK/FK information")
|
beeLogger.Log.Fatal("Could not read INFORMATION_SCHEMA for PK/UK/FK information")
|
||||||
}
|
}
|
||||||
constraintType, columnName, refTableSchema, refTableName, refColumnName, refOrdinalPos :=
|
constraintType, columnName, refTableSchema, refTableName, refColumnName, refOrdinalPos :=
|
||||||
string(constraintTypeBytes), string(columnNameBytes), string(refTableSchemaBytes),
|
string(constraintTypeBytes), string(columnNameBytes), string(refTableSchemaBytes),
|
||||||
|
@ -400,37 +417,38 @@ func (mysqlDB *MysqlDB) GetColumns(db *sql.DB, table *Table, blackList map[strin
|
||||||
// retrieve columns
|
// retrieve columns
|
||||||
colDefRows, err := db.Query(
|
colDefRows, err := db.Query(
|
||||||
`SELECT
|
`SELECT
|
||||||
column_name, data_type, column_type, is_nullable, column_default, extra
|
column_name, data_type, column_type, is_nullable, column_default, extra, column_comment
|
||||||
FROM
|
FROM
|
||||||
information_schema.columns
|
information_schema.columns
|
||||||
WHERE
|
WHERE
|
||||||
table_schema = database() AND table_name = ?`,
|
table_schema = database() AND table_name = ?`,
|
||||||
table.Name)
|
table.Name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Could not query the database: %s", err)
|
beeLogger.Log.Fatalf("Could not query the database: %s", err)
|
||||||
}
|
}
|
||||||
defer colDefRows.Close()
|
defer colDefRows.Close()
|
||||||
|
|
||||||
for colDefRows.Next() {
|
for colDefRows.Next() {
|
||||||
// datatype as bytes so that SQL <null> values can be retrieved
|
// datatype as bytes so that SQL <null> values can be retrieved
|
||||||
var colNameBytes, dataTypeBytes, columnTypeBytes, isNullableBytes, columnDefaultBytes, extraBytes []byte
|
var colNameBytes, dataTypeBytes, columnTypeBytes, isNullableBytes, columnDefaultBytes, extraBytes, columnCommentBytes []byte
|
||||||
if err := colDefRows.Scan(&colNameBytes, &dataTypeBytes, &columnTypeBytes, &isNullableBytes, &columnDefaultBytes, &extraBytes); err != nil {
|
if err := colDefRows.Scan(&colNameBytes, &dataTypeBytes, &columnTypeBytes, &isNullableBytes, &columnDefaultBytes, &extraBytes, &columnCommentBytes); err != nil {
|
||||||
logger.Fatal("Could not query INFORMATION_SCHEMA for column information")
|
beeLogger.Log.Fatal("Could not query INFORMATION_SCHEMA for column information")
|
||||||
}
|
}
|
||||||
colName, dataType, columnType, isNullable, columnDefault, extra :=
|
colName, dataType, columnType, isNullable, columnDefault, extra, columnComment :=
|
||||||
string(colNameBytes), string(dataTypeBytes), string(columnTypeBytes), string(isNullableBytes), string(columnDefaultBytes), string(extraBytes)
|
string(colNameBytes), string(dataTypeBytes), string(columnTypeBytes), string(isNullableBytes), string(columnDefaultBytes), string(extraBytes), string(columnCommentBytes)
|
||||||
|
|
||||||
// create a column
|
// create a column
|
||||||
col := new(Column)
|
col := new(Column)
|
||||||
col.Name = camelCase(colName)
|
col.Name = utils.CamelCase(colName)
|
||||||
col.Type, err = mysqlDB.GetGoDataType(dataType)
|
col.Type, err = mysqlDB.GetGoDataType(dataType)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("%s", err)
|
beeLogger.Log.Fatalf("%s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Tag info
|
// Tag info
|
||||||
tag := new(OrmTag)
|
tag := new(OrmTag)
|
||||||
tag.Column = colName
|
tag.Column = colName
|
||||||
|
tag.Comment = columnComment
|
||||||
if table.Pk == colName {
|
if table.Pk == colName {
|
||||||
col.Name = "Id"
|
col.Name = "Id"
|
||||||
col.Type = "int"
|
col.Type = "int"
|
||||||
|
@ -449,8 +467,8 @@ func (mysqlDB *MysqlDB) GetColumns(db *sql.DB, table *Table, blackList map[strin
|
||||||
if isFk && !isBl {
|
if isFk && !isBl {
|
||||||
tag.RelFk = true
|
tag.RelFk = true
|
||||||
refStructName := fkCol.RefTable
|
refStructName := fkCol.RefTable
|
||||||
col.Name = camelCase(colName)
|
col.Name = utils.CamelCase(colName)
|
||||||
col.Type = "*" + camelCase(refStructName)
|
col.Type = "*" + utils.CamelCase(refStructName)
|
||||||
} else {
|
} else {
|
||||||
// if the name of column is Id, and it's not primary key
|
// if the name of column is Id, and it's not primary key
|
||||||
if colName == "id" {
|
if colName == "id" {
|
||||||
|
@ -464,7 +482,7 @@ func (mysqlDB *MysqlDB) GetColumns(db *sql.DB, table *Table, blackList map[strin
|
||||||
if sign == "unsigned" && extra != "auto_increment" {
|
if sign == "unsigned" && extra != "auto_increment" {
|
||||||
col.Type, err = mysqlDB.GetGoDataType(dataType + " " + sign)
|
col.Type, err = mysqlDB.GetGoDataType(dataType + " " + sign)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("%s", err)
|
beeLogger.Log.Fatalf("%s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -500,9 +518,7 @@ func (mysqlDB *MysqlDB) GetColumns(db *sql.DB, table *Table, blackList map[strin
|
||||||
|
|
||||||
// GetGoDataType maps an SQL data type to Golang data type
|
// GetGoDataType maps an SQL data type to Golang data type
|
||||||
func (*MysqlDB) GetGoDataType(sqlType string) (string, error) {
|
func (*MysqlDB) GetGoDataType(sqlType string) (string, error) {
|
||||||
var typeMapping = map[string]string{}
|
if v, ok := typeMappingMysql[sqlType]; ok {
|
||||||
typeMapping = typeMappingMysql
|
|
||||||
if v, ok := typeMapping[sqlType]; ok {
|
|
||||||
return v, nil
|
return v, nil
|
||||||
}
|
}
|
||||||
return "", fmt.Errorf("data type '%s' not found", sqlType)
|
return "", fmt.Errorf("data type '%s' not found", sqlType)
|
||||||
|
@ -516,14 +532,14 @@ func (*PostgresDB) GetTableNames(db *sql.DB) (tables []string) {
|
||||||
table_type = 'BASE TABLE' AND
|
table_type = 'BASE TABLE' AND
|
||||||
table_schema NOT IN ('pg_catalog', 'information_schema')`)
|
table_schema NOT IN ('pg_catalog', 'information_schema')`)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Could not show tables: %s", err)
|
beeLogger.Log.Fatalf("Could not show tables: %s", err)
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
defer rows.Close()
|
||||||
|
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
var name string
|
var name string
|
||||||
if err := rows.Scan(&name); err != nil {
|
if err := rows.Scan(&name); err != nil {
|
||||||
logger.Fatalf("Could not show tables: %s", err)
|
beeLogger.Log.Fatalf("Could not show tables: %s", err)
|
||||||
}
|
}
|
||||||
tables = append(tables, name)
|
tables = append(tables, name)
|
||||||
}
|
}
|
||||||
|
@ -553,13 +569,13 @@ func (*PostgresDB) GetConstraints(db *sql.DB, table *Table, blackList map[string
|
||||||
AND u.table_name = $2`,
|
AND u.table_name = $2`,
|
||||||
table.Name, table.Name) // u.position_in_unique_constraint,
|
table.Name, table.Name) // u.position_in_unique_constraint,
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Could not query INFORMATION_SCHEMA for PK/UK/FK information: %s", err)
|
beeLogger.Log.Fatalf("Could not query INFORMATION_SCHEMA for PK/UK/FK information: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
var constraintTypeBytes, columnNameBytes, refTableSchemaBytes, refTableNameBytes, refColumnNameBytes, refOrdinalPosBytes []byte
|
var constraintTypeBytes, columnNameBytes, refTableSchemaBytes, refTableNameBytes, refColumnNameBytes, refOrdinalPosBytes []byte
|
||||||
if err := rows.Scan(&constraintTypeBytes, &columnNameBytes, &refTableSchemaBytes, &refTableNameBytes, &refColumnNameBytes, &refOrdinalPosBytes); err != nil {
|
if err := rows.Scan(&constraintTypeBytes, &columnNameBytes, &refTableSchemaBytes, &refTableNameBytes, &refColumnNameBytes, &refOrdinalPosBytes); err != nil {
|
||||||
logger.Fatalf("Could not read INFORMATION_SCHEMA for PK/UK/FK information: %s", err)
|
beeLogger.Log.Fatalf("Could not read INFORMATION_SCHEMA for PK/UK/FK information: %s", err)
|
||||||
}
|
}
|
||||||
constraintType, columnName, refTableSchema, refTableName, refColumnName, refOrdinalPos :=
|
constraintType, columnName, refTableSchema, refTableName, refColumnName, refOrdinalPos :=
|
||||||
string(constraintTypeBytes), string(columnNameBytes), string(refTableSchemaBytes),
|
string(constraintTypeBytes), string(columnNameBytes), string(refTableSchemaBytes),
|
||||||
|
@ -609,7 +625,7 @@ func (postgresDB *PostgresDB) GetColumns(db *sql.DB, table *Table, blackList map
|
||||||
AND table_name = $1`,
|
AND table_name = $1`,
|
||||||
table.Name)
|
table.Name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Could not query INFORMATION_SCHEMA for column information: %s", err)
|
beeLogger.Log.Fatalf("Could not query INFORMATION_SCHEMA for column information: %s", err)
|
||||||
}
|
}
|
||||||
defer colDefRows.Close()
|
defer colDefRows.Close()
|
||||||
|
|
||||||
|
@ -617,16 +633,16 @@ func (postgresDB *PostgresDB) GetColumns(db *sql.DB, table *Table, blackList map
|
||||||
// datatype as bytes so that SQL <null> values can be retrieved
|
// datatype as bytes so that SQL <null> values can be retrieved
|
||||||
var colNameBytes, dataTypeBytes, columnTypeBytes, isNullableBytes, columnDefaultBytes, extraBytes []byte
|
var colNameBytes, dataTypeBytes, columnTypeBytes, isNullableBytes, columnDefaultBytes, extraBytes []byte
|
||||||
if err := colDefRows.Scan(&colNameBytes, &dataTypeBytes, &columnTypeBytes, &isNullableBytes, &columnDefaultBytes, &extraBytes); err != nil {
|
if err := colDefRows.Scan(&colNameBytes, &dataTypeBytes, &columnTypeBytes, &isNullableBytes, &columnDefaultBytes, &extraBytes); err != nil {
|
||||||
logger.Fatalf("Could not query INFORMATION_SCHEMA for column information: %s", err)
|
beeLogger.Log.Fatalf("Could not query INFORMATION_SCHEMA for column information: %s", err)
|
||||||
}
|
}
|
||||||
colName, dataType, columnType, isNullable, columnDefault, extra :=
|
colName, dataType, columnType, isNullable, columnDefault, extra :=
|
||||||
string(colNameBytes), string(dataTypeBytes), string(columnTypeBytes), string(isNullableBytes), string(columnDefaultBytes), string(extraBytes)
|
string(colNameBytes), string(dataTypeBytes), string(columnTypeBytes), string(isNullableBytes), string(columnDefaultBytes), string(extraBytes)
|
||||||
// Create a column
|
// Create a column
|
||||||
col := new(Column)
|
col := new(Column)
|
||||||
col.Name = camelCase(colName)
|
col.Name = utils.CamelCase(colName)
|
||||||
col.Type, err = postgresDB.GetGoDataType(dataType)
|
col.Type, err = postgresDB.GetGoDataType(dataType)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("%s", err)
|
beeLogger.Log.Fatalf("%s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Tag info
|
// Tag info
|
||||||
|
@ -650,8 +666,8 @@ func (postgresDB *PostgresDB) GetColumns(db *sql.DB, table *Table, blackList map
|
||||||
if isFk && !isBl {
|
if isFk && !isBl {
|
||||||
tag.RelFk = true
|
tag.RelFk = true
|
||||||
refStructName := fkCol.RefTable
|
refStructName := fkCol.RefTable
|
||||||
col.Name = camelCase(colName)
|
col.Name = utils.CamelCase(colName)
|
||||||
col.Type = "*" + camelCase(refStructName)
|
col.Type = "*" + utils.CamelCase(refStructName)
|
||||||
} else {
|
} else {
|
||||||
// if the name of column is Id, and it's not primary key
|
// if the name of column is Id, and it's not primary key
|
||||||
if colName == "id" {
|
if colName == "id" {
|
||||||
|
@ -714,63 +730,57 @@ func createPaths(mode byte, paths *MvcPath) {
|
||||||
// writeSourceFiles generates source files for model/controller/router
|
// writeSourceFiles generates source files for model/controller/router
|
||||||
// It will wipe the following directories and recreate them:./models, ./controllers, ./routers
|
// It will wipe the following directories and recreate them:./models, ./controllers, ./routers
|
||||||
// Newly geneated files will be inside these folders.
|
// Newly geneated files will be inside these folders.
|
||||||
func writeSourceFiles(pkgPath string, tables []*Table, mode byte, paths *MvcPath, selectedTables map[string]bool) {
|
func writeSourceFiles(pkgPath string, tables []*Table, mode byte, paths *MvcPath) {
|
||||||
if (OModel & mode) == OModel {
|
if (OModel & mode) == OModel {
|
||||||
logger.Info("Creating model files...")
|
beeLogger.Log.Info("Creating model files...")
|
||||||
writeModelFiles(tables, paths.ModelPath, selectedTables)
|
writeModelFiles(tables, paths.ModelPath)
|
||||||
}
|
}
|
||||||
if (OController & mode) == OController {
|
if (OController & mode) == OController {
|
||||||
logger.Info("Creating controller files...")
|
beeLogger.Log.Info("Creating controller files...")
|
||||||
writeControllerFiles(tables, paths.ControllerPath, selectedTables, pkgPath)
|
writeControllerFiles(tables, paths.ControllerPath, pkgPath)
|
||||||
}
|
}
|
||||||
if (ORouter & mode) == ORouter {
|
if (ORouter & mode) == ORouter {
|
||||||
logger.Info("Creating router files...")
|
beeLogger.Log.Info("Creating router files...")
|
||||||
writeRouterFile(tables, paths.RouterPath, selectedTables, pkgPath)
|
writeRouterFile(tables, paths.RouterPath, pkgPath)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// writeModelFiles generates model files
|
// writeModelFiles generates model files
|
||||||
func writeModelFiles(tables []*Table, mPath string, selectedTables map[string]bool) {
|
func writeModelFiles(tables []*Table, mPath string) {
|
||||||
w := NewColorWriter(os.Stdout)
|
w := colors.NewColorWriter(os.Stdout)
|
||||||
|
|
||||||
for _, tb := range tables {
|
for _, tb := range tables {
|
||||||
// if selectedTables map is not nil and this table is not selected, ignore it
|
|
||||||
if selectedTables != nil {
|
|
||||||
if _, selected := selectedTables[tb.Name]; !selected {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
filename := getFileName(tb.Name)
|
filename := getFileName(tb.Name)
|
||||||
fpath := path.Join(mPath, filename+".go")
|
fpath := path.Join(mPath, filename+".go")
|
||||||
var f *os.File
|
var f *os.File
|
||||||
var err error
|
var err error
|
||||||
if isExist(fpath) {
|
if utils.IsExist(fpath) {
|
||||||
logger.Warnf("'%s' already exists. Do you want to overwrite it? [Yes|No] ", fpath)
|
beeLogger.Log.Warnf("'%s' already exists. Do you want to overwrite it? [Yes|No] ", fpath)
|
||||||
if askForConfirmation() {
|
if utils.AskForConfirmation() {
|
||||||
f, err = os.OpenFile(fpath, os.O_RDWR|os.O_TRUNC, 0666)
|
f, err = os.OpenFile(fpath, os.O_RDWR|os.O_TRUNC, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Warnf("%s", err)
|
beeLogger.Log.Warnf("%s", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
logger.Warnf("Skipped create file '%s'", fpath)
|
beeLogger.Log.Warnf("Skipped create file '%s'", fpath)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
f, err = os.OpenFile(fpath, os.O_CREATE|os.O_RDWR, 0666)
|
f, err = os.OpenFile(fpath, os.O_CREATE|os.O_RDWR, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Warnf("%s", err)
|
beeLogger.Log.Warnf("%s", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
template := ""
|
var template string
|
||||||
if tb.Pk == "" {
|
if tb.Pk == "" {
|
||||||
template = StructModelTPL
|
template = StructModelTPL
|
||||||
} else {
|
} else {
|
||||||
template = ModelTPL
|
template = ModelTPL
|
||||||
}
|
}
|
||||||
fileStr := strings.Replace(template, "{{modelStruct}}", tb.String(), 1)
|
fileStr := strings.Replace(template, "{{modelStruct}}", tb.String(), 1)
|
||||||
fileStr = strings.Replace(fileStr, "{{modelName}}", camelCase(tb.Name), -1)
|
fileStr = strings.Replace(fileStr, "{{modelName}}", utils.CamelCase(tb.Name), -1)
|
||||||
fileStr = strings.Replace(fileStr, "{{tableName}}", tb.Name, -1)
|
fileStr = strings.Replace(fileStr, "{{tableName}}", tb.Name, -1)
|
||||||
|
|
||||||
// If table contains time field, import time.Time package
|
// If table contains time field, import time.Time package
|
||||||
|
@ -783,25 +793,19 @@ func writeModelFiles(tables []*Table, mPath string, selectedTables map[string]bo
|
||||||
fileStr = strings.Replace(fileStr, "{{timePkg}}", timePkg, -1)
|
fileStr = strings.Replace(fileStr, "{{timePkg}}", timePkg, -1)
|
||||||
fileStr = strings.Replace(fileStr, "{{importTimePkg}}", importTimePkg, -1)
|
fileStr = strings.Replace(fileStr, "{{importTimePkg}}", importTimePkg, -1)
|
||||||
if _, err := f.WriteString(fileStr); err != nil {
|
if _, err := f.WriteString(fileStr); err != nil {
|
||||||
logger.Fatalf("Could not write model file to '%s': %s", fpath, err)
|
beeLogger.Log.Fatalf("Could not write model file to '%s': %s", fpath, err)
|
||||||
}
|
}
|
||||||
CloseFile(f)
|
utils.CloseFile(f)
|
||||||
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
||||||
formatSourceCode(fpath)
|
utils.FormatSourceCode(fpath)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// writeControllerFiles generates controller files
|
// writeControllerFiles generates controller files
|
||||||
func writeControllerFiles(tables []*Table, cPath string, selectedTables map[string]bool, pkgPath string) {
|
func writeControllerFiles(tables []*Table, cPath string, pkgPath string) {
|
||||||
w := NewColorWriter(os.Stdout)
|
w := colors.NewColorWriter(os.Stdout)
|
||||||
|
|
||||||
for _, tb := range tables {
|
for _, tb := range tables {
|
||||||
// If selectedTables map is not nil and this table is not selected, ignore it
|
|
||||||
if selectedTables != nil {
|
|
||||||
if _, selected := selectedTables[tb.Name]; !selected {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if tb.Pk == "" {
|
if tb.Pk == "" {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
@ -809,87 +813,81 @@ func writeControllerFiles(tables []*Table, cPath string, selectedTables map[stri
|
||||||
fpath := path.Join(cPath, filename+".go")
|
fpath := path.Join(cPath, filename+".go")
|
||||||
var f *os.File
|
var f *os.File
|
||||||
var err error
|
var err error
|
||||||
if isExist(fpath) {
|
if utils.IsExist(fpath) {
|
||||||
logger.Warnf("'%s' already exists. Do you want to overwrite it? [Yes|No] ", fpath)
|
beeLogger.Log.Warnf("'%s' already exists. Do you want to overwrite it? [Yes|No] ", fpath)
|
||||||
if askForConfirmation() {
|
if utils.AskForConfirmation() {
|
||||||
f, err = os.OpenFile(fpath, os.O_RDWR|os.O_TRUNC, 0666)
|
f, err = os.OpenFile(fpath, os.O_RDWR|os.O_TRUNC, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Warnf("%s", err)
|
beeLogger.Log.Warnf("%s", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
logger.Warnf("Skipped create file '%s'", fpath)
|
beeLogger.Log.Warnf("Skipped create file '%s'", fpath)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
f, err = os.OpenFile(fpath, os.O_CREATE|os.O_RDWR, 0666)
|
f, err = os.OpenFile(fpath, os.O_CREATE|os.O_RDWR, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Warnf("%s", err)
|
beeLogger.Log.Warnf("%s", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
fileStr := strings.Replace(CtrlTPL, "{{ctrlName}}", camelCase(tb.Name), -1)
|
fileStr := strings.Replace(CtrlTPL, "{{ctrlName}}", utils.CamelCase(tb.Name), -1)
|
||||||
fileStr = strings.Replace(fileStr, "{{pkgPath}}", pkgPath, -1)
|
fileStr = strings.Replace(fileStr, "{{pkgPath}}", pkgPath, -1)
|
||||||
if _, err := f.WriteString(fileStr); err != nil {
|
if _, err := f.WriteString(fileStr); err != nil {
|
||||||
logger.Fatalf("Could not write controller file to '%s': %s", fpath, err)
|
beeLogger.Log.Fatalf("Could not write controller file to '%s': %s", fpath, err)
|
||||||
}
|
}
|
||||||
CloseFile(f)
|
utils.CloseFile(f)
|
||||||
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
||||||
formatSourceCode(fpath)
|
utils.FormatSourceCode(fpath)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// writeRouterFile generates router file
|
// writeRouterFile generates router file
|
||||||
func writeRouterFile(tables []*Table, rPath string, selectedTables map[string]bool, pkgPath string) {
|
func writeRouterFile(tables []*Table, rPath string, pkgPath string) {
|
||||||
w := NewColorWriter(os.Stdout)
|
w := colors.NewColorWriter(os.Stdout)
|
||||||
|
|
||||||
var nameSpaces []string
|
var nameSpaces []string
|
||||||
for _, tb := range tables {
|
for _, tb := range tables {
|
||||||
// If selectedTables map is not nil and this table is not selected, ignore it
|
|
||||||
if selectedTables != nil {
|
|
||||||
if _, selected := selectedTables[tb.Name]; !selected {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if tb.Pk == "" {
|
if tb.Pk == "" {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
// Add namespaces
|
// Add namespaces
|
||||||
nameSpace := strings.Replace(NamespaceTPL, "{{nameSpace}}", tb.Name, -1)
|
nameSpace := strings.Replace(NamespaceTPL, "{{nameSpace}}", tb.Name, -1)
|
||||||
nameSpace = strings.Replace(nameSpace, "{{ctrlName}}", camelCase(tb.Name), -1)
|
nameSpace = strings.Replace(nameSpace, "{{ctrlName}}", utils.CamelCase(tb.Name), -1)
|
||||||
nameSpaces = append(nameSpaces, nameSpace)
|
nameSpaces = append(nameSpaces, nameSpace)
|
||||||
}
|
}
|
||||||
// Add export controller
|
// Add export controller
|
||||||
fpath := path.Join(rPath, "router.go")
|
fpath := filepath.Join(rPath, "router.go")
|
||||||
routerStr := strings.Replace(RouterTPL, "{{nameSpaces}}", strings.Join(nameSpaces, ""), 1)
|
routerStr := strings.Replace(RouterTPL, "{{nameSpaces}}", strings.Join(nameSpaces, ""), 1)
|
||||||
routerStr = strings.Replace(routerStr, "{{pkgPath}}", pkgPath, 1)
|
routerStr = strings.Replace(routerStr, "{{pkgPath}}", pkgPath, 1)
|
||||||
var f *os.File
|
var f *os.File
|
||||||
var err error
|
var err error
|
||||||
if isExist(fpath) {
|
if utils.IsExist(fpath) {
|
||||||
logger.Warnf("'%s' already exists. Do you want to overwrite it? [Yes|No] ", fpath)
|
beeLogger.Log.Warnf("'%s' already exists. Do you want to overwrite it? [Yes|No] ", fpath)
|
||||||
if askForConfirmation() {
|
if utils.AskForConfirmation() {
|
||||||
f, err = os.OpenFile(fpath, os.O_RDWR|os.O_TRUNC, 0666)
|
f, err = os.OpenFile(fpath, os.O_RDWR|os.O_TRUNC, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Warnf("%s", err)
|
beeLogger.Log.Warnf("%s", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
logger.Warnf("Skipped create file '%s'", fpath)
|
beeLogger.Log.Warnf("Skipped create file '%s'", fpath)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
f, err = os.OpenFile(fpath, os.O_CREATE|os.O_RDWR, 0666)
|
f, err = os.OpenFile(fpath, os.O_CREATE|os.O_RDWR, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Warnf("%s", err)
|
beeLogger.Log.Warnf("%s", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if _, err := f.WriteString(routerStr); err != nil {
|
if _, err := f.WriteString(routerStr); err != nil {
|
||||||
logger.Fatalf("Could not write router file to '%s': %s", fpath, err)
|
beeLogger.Log.Fatalf("Could not write router file to '%s': %s", fpath, err)
|
||||||
}
|
}
|
||||||
CloseFile(f)
|
utils.CloseFile(f)
|
||||||
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
||||||
formatSourceCode(fpath)
|
utils.FormatSourceCode(fpath)
|
||||||
}
|
}
|
||||||
|
|
||||||
func isSQLTemporalType(t string) bool {
|
func isSQLTemporalType(t string) bool {
|
||||||
|
@ -927,9 +925,9 @@ func extractColSize(colType string) string {
|
||||||
}
|
}
|
||||||
|
|
||||||
func extractIntSignness(colType string) string {
|
func extractIntSignness(colType string) string {
|
||||||
regex := regexp.MustCompile(`(int|smallint|mediumint|bigint)\([0-9]+\)(.*)`)
|
regex := regexp.MustCompile(`(int|smallint|mediumint|bigint).*`)
|
||||||
signRegex := regex.FindStringSubmatch(colType)
|
signRegex := regex.FindStringSubmatch(colType)
|
||||||
return strings.Trim(signRegex[2], " ")
|
return strings.Trim(signRegex[1], " ")
|
||||||
}
|
}
|
||||||
|
|
||||||
func extractDecimal(colType string) (digits string, decimals string) {
|
func extractDecimal(colType string) (digits string, decimals string) {
|
||||||
|
@ -952,19 +950,47 @@ func getFileName(tbName string) (filename string) {
|
||||||
func getPackagePath(curpath string) (packpath string) {
|
func getPackagePath(curpath string) (packpath string) {
|
||||||
gopath := os.Getenv("GOPATH")
|
gopath := os.Getenv("GOPATH")
|
||||||
if gopath == "" {
|
if gopath == "" {
|
||||||
logger.Fatal("GOPATH environment variable is not set or empty")
|
info := "GOPATH environment variable is not set or empty"
|
||||||
|
gomodpath := filepath.Join(curpath, `go.mod`)
|
||||||
|
re, err := regexp.Compile(`^module\s+(.+)$`)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(info)
|
||||||
|
beeLogger.Log.Fatalf("try `go.mod` generate regexp error:%s", err)
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
fd, err := os.Open(gomodpath)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(info)
|
||||||
|
beeLogger.Log.Fatalf("try `go.mod` Error while reading 'go.mod',%s", gomodpath)
|
||||||
|
}
|
||||||
|
reader := bufio.NewReader(fd)
|
||||||
|
for {
|
||||||
|
byteLine, _, er := reader.ReadLine()
|
||||||
|
if er != nil && er != io.EOF {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
if er == io.EOF {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
line := string(byteLine)
|
||||||
|
s := re.FindStringSubmatch(line)
|
||||||
|
if len(s) >= 2 {
|
||||||
|
return s[1]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
beeLogger.Log.Error(info)
|
||||||
|
beeLogger.Log.Fatalf("try `go.mod` Error while parse 'go.mod',%s", gomodpath)
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Debugf("GOPATH: %s", utils.FILE(), utils.LINE(), gopath)
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Debugf("GOPATH: %s", __FILE__(), __LINE__(), gopath)
|
|
||||||
|
|
||||||
appsrcpath := ""
|
appsrcpath := ""
|
||||||
haspath := false
|
haspath := false
|
||||||
wgopath := filepath.SplitList(gopath)
|
wgopath := filepath.SplitList(gopath)
|
||||||
|
|
||||||
for _, wg := range wgopath {
|
for _, wg := range wgopath {
|
||||||
wg, _ = filepath.EvalSymlinks(path.Join(wg, "src"))
|
wg, _ = filepath.EvalSymlinks(filepath.Join(wg, "src"))
|
||||||
|
if strings.HasPrefix(strings.ToLower(curpath), strings.ToLower(wg)) {
|
||||||
if filepath.HasPrefix(strings.ToLower(curpath), strings.ToLower(wg)) {
|
|
||||||
haspath = true
|
haspath = true
|
||||||
appsrcpath = wg
|
appsrcpath = wg
|
||||||
break
|
break
|
||||||
|
@ -972,11 +998,11 @@ func getPackagePath(curpath string) (packpath string) {
|
||||||
}
|
}
|
||||||
|
|
||||||
if !haspath {
|
if !haspath {
|
||||||
logger.Fatalf("Cannot generate application code outside of GOPATH '%s'", gopath)
|
beeLogger.Log.Fatalf("Cannot generate application code outside of GOPATH '%s' compare with CWD '%s'", gopath, curpath)
|
||||||
}
|
}
|
||||||
|
|
||||||
if curpath == appsrcpath {
|
if curpath == appsrcpath {
|
||||||
logger.Fatal("Cannot generate application code outside of application path")
|
beeLogger.Log.Fatal("Cannot generate application code outside of application path")
|
||||||
}
|
}
|
||||||
|
|
||||||
packpath = strings.Join(strings.Split(curpath[len(appsrcpath)+1:], string(filepath.Separator)), "/")
|
packpath = strings.Join(strings.Split(curpath[len(appsrcpath)+1:], string(filepath.Separator)), "/")
|
||||||
|
@ -997,7 +1023,7 @@ import (
|
||||||
"reflect"
|
"reflect"
|
||||||
"strings"
|
"strings"
|
||||||
{{timePkg}}
|
{{timePkg}}
|
||||||
"github.com/astaxie/beego/orm"
|
"github.com/beego/beego/v2/client/orm"
|
||||||
)
|
)
|
||||||
|
|
||||||
{{modelStruct}}
|
{{modelStruct}}
|
||||||
|
@ -1146,7 +1172,7 @@ import (
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
// {{ctrlName}}Controller operations for {{ctrlName}}
|
// {{ctrlName}}Controller operations for {{ctrlName}}
|
||||||
|
@ -1321,7 +1347,7 @@ package routers
|
||||||
import (
|
import (
|
||||||
"{{pkgPath}}/controllers"
|
"{{pkgPath}}/controllers"
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
func init() {
|
|
@ -12,17 +12,21 @@
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
|
|
||||||
package main
|
package generate
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
func generateController(cname, currpath string) {
|
func GenerateController(cname, currpath string) {
|
||||||
w := NewColorWriter(os.Stdout)
|
w := colors.NewColorWriter(os.Stdout)
|
||||||
|
|
||||||
p, f := path.Split(cname)
|
p, f := path.Split(cname)
|
||||||
controllerName := strings.Title(f)
|
controllerName := strings.Title(f)
|
||||||
|
@ -33,26 +37,26 @@ func generateController(cname, currpath string) {
|
||||||
packageName = p[i+1 : len(p)-1]
|
packageName = p[i+1 : len(p)-1]
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Infof("Using '%s' as controller name", controllerName)
|
beeLogger.Log.Infof("Using '%s' as controller name", controllerName)
|
||||||
logger.Infof("Using '%s' as package name", packageName)
|
beeLogger.Log.Infof("Using '%s' as package name", packageName)
|
||||||
|
|
||||||
fp := path.Join(currpath, "controllers", p)
|
fp := path.Join(currpath, "controllers", p)
|
||||||
if _, err := os.Stat(fp); os.IsNotExist(err) {
|
if _, err := os.Stat(fp); os.IsNotExist(err) {
|
||||||
// Create the controller's directory
|
// Create the controller's directory
|
||||||
if err := os.MkdirAll(fp, 0777); err != nil {
|
if err := os.MkdirAll(fp, 0777); err != nil {
|
||||||
logger.Fatalf("Could not create controllers directory: %s", err)
|
beeLogger.Log.Fatalf("Could not create controllers directory: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fpath := path.Join(fp, strings.ToLower(controllerName)+".go")
|
fpath := path.Join(fp, strings.ToLower(controllerName)+".go")
|
||||||
if f, err := os.OpenFile(fpath, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
if f, err := os.OpenFile(fpath, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
||||||
defer CloseFile(f)
|
defer utils.CloseFile(f)
|
||||||
|
|
||||||
modelPath := path.Join(currpath, "models", strings.ToLower(controllerName)+".go")
|
modelPath := path.Join(currpath, "models", strings.ToLower(controllerName)+".go")
|
||||||
|
|
||||||
var content string
|
var content string
|
||||||
if _, err := os.Stat(modelPath); err == nil {
|
if _, err := os.Stat(modelPath); err == nil {
|
||||||
logger.Infof("Using matching model '%s'", controllerName)
|
beeLogger.Log.Infof("Using matching model '%s'", controllerName)
|
||||||
content = strings.Replace(controllerModelTpl, "{{packageName}}", packageName, -1)
|
content = strings.Replace(controllerModelTpl, "{{packageName}}", packageName, -1)
|
||||||
pkgPath := getPackagePath(currpath)
|
pkgPath := getPackagePath(currpath)
|
||||||
content = strings.Replace(content, "{{pkgPath}}", pkgPath, -1)
|
content = strings.Replace(content, "{{pkgPath}}", pkgPath, -1)
|
||||||
|
@ -64,17 +68,17 @@ func generateController(cname, currpath string) {
|
||||||
f.WriteString(content)
|
f.WriteString(content)
|
||||||
|
|
||||||
// Run 'gofmt' on the generated source code
|
// Run 'gofmt' on the generated source code
|
||||||
formatSourceCode(fpath)
|
utils.FormatSourceCode(fpath)
|
||||||
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
||||||
} else {
|
} else {
|
||||||
logger.Fatalf("Could not create controller file: %s", err)
|
beeLogger.Log.Fatalf("Could not create controller file: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
var controllerTpl = `package {{packageName}}
|
var controllerTpl = `package {{packageName}}
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
// {{controllerName}}Controller operations for {{controllerName}}
|
// {{controllerName}}Controller operations for {{controllerName}}
|
||||||
|
@ -162,10 +166,10 @@ import (
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
// {{controllerName}}Controller oprations for {{controllerName}}
|
// {{controllerName}}Controller operations for {{controllerName}}
|
||||||
type {{controllerName}}Controller struct {
|
type {{controllerName}}Controller struct {
|
||||||
beego.Controller
|
beego.Controller
|
||||||
}
|
}
|
|
@ -15,7 +15,7 @@
|
||||||
* *
|
* *
|
||||||
\**********************************************************/
|
\**********************************************************/
|
||||||
|
|
||||||
package main
|
package generate
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
|
@ -24,11 +24,256 @@ import (
|
||||||
"path"
|
"path"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
_ "github.com/go-sql-driver/mysql"
|
_ "github.com/go-sql-driver/mysql"
|
||||||
_ "github.com/lib/pq"
|
_ "github.com/lib/pq"
|
||||||
)
|
)
|
||||||
|
|
||||||
func generateHproseAppcode(driver, connStr, level, tables, currpath string) {
|
var Hproseconf = `appname = {{.Appname}}
|
||||||
|
httpport = 8080
|
||||||
|
runmode = dev
|
||||||
|
autorender = false
|
||||||
|
copyrequestbody = true
|
||||||
|
EnableDocs = true
|
||||||
|
`
|
||||||
|
var HproseMaingo = `package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"reflect"
|
||||||
|
|
||||||
|
"{{.Appname}}/models"
|
||||||
|
"github.com/hprose/hprose-golang/rpc"
|
||||||
|
|
||||||
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
|
)
|
||||||
|
|
||||||
|
func logInvokeHandler(
|
||||||
|
name string,
|
||||||
|
args []reflect.Value,
|
||||||
|
context rpc.Context,
|
||||||
|
next rpc.NextInvokeHandler) (results []reflect.Value, err error) {
|
||||||
|
fmt.Printf("%s(%v) = ", name, args)
|
||||||
|
results, err = next(name, args, context)
|
||||||
|
fmt.Printf("%v %v\r\n", results, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// Create WebSocketServer
|
||||||
|
// service := rpc.NewWebSocketService()
|
||||||
|
|
||||||
|
// Create Http Server
|
||||||
|
service := rpc.NewHTTPService()
|
||||||
|
|
||||||
|
// Use Logger Middleware
|
||||||
|
service.AddInvokeHandler(logInvokeHandler)
|
||||||
|
|
||||||
|
// Publish Functions
|
||||||
|
service.AddFunction("AddOne", models.AddOne)
|
||||||
|
service.AddFunction("GetOne", models.GetOne)
|
||||||
|
|
||||||
|
// Start Service
|
||||||
|
beego.Handler("/", service)
|
||||||
|
beego.Run()
|
||||||
|
}
|
||||||
|
`
|
||||||
|
|
||||||
|
var HproseMainconngo = `package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"reflect"
|
||||||
|
|
||||||
|
"{{.Appname}}/models"
|
||||||
|
"github.com/hprose/hprose-golang/rpc"
|
||||||
|
|
||||||
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
|
"github.com/beego/beego/v2/client/orm"
|
||||||
|
{{.DriverPkg}}
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
orm.RegisterDataBase("default", "{{.DriverName}}", "{{.conn}}")
|
||||||
|
}
|
||||||
|
|
||||||
|
func logInvokeHandler(
|
||||||
|
name string,
|
||||||
|
args []reflect.Value,
|
||||||
|
context rpc.Context,
|
||||||
|
next rpc.NextInvokeHandler) (results []reflect.Value, err error) {
|
||||||
|
fmt.Printf("%s(%v) = ", name, args)
|
||||||
|
results, err = next(name, args, context)
|
||||||
|
fmt.Printf("%v %v\r\n", results, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// Create WebSocketServer
|
||||||
|
// service := rpc.NewWebSocketService()
|
||||||
|
|
||||||
|
// Create Http Server
|
||||||
|
service := rpc.NewHTTPService()
|
||||||
|
|
||||||
|
// Use Logger Middleware
|
||||||
|
service.AddInvokeHandler(logInvokeHandler)
|
||||||
|
|
||||||
|
{{HproseFunctionList}}
|
||||||
|
|
||||||
|
// Start Service
|
||||||
|
beego.Handler("/", service)
|
||||||
|
beego.Run()
|
||||||
|
}
|
||||||
|
|
||||||
|
`
|
||||||
|
|
||||||
|
var HproseModels = `package models
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
Objects map[string]*Object
|
||||||
|
)
|
||||||
|
|
||||||
|
type Object struct {
|
||||||
|
ObjectId string
|
||||||
|
Score int64
|
||||||
|
PlayerName string
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
Objects = make(map[string]*Object)
|
||||||
|
Objects["hjkhsbnmn123"] = &Object{"hjkhsbnmn123", 100, "astaxie"}
|
||||||
|
Objects["mjjkxsxsaa23"] = &Object{"mjjkxsxsaa23", 101, "someone"}
|
||||||
|
}
|
||||||
|
|
||||||
|
func AddOne(object Object) (ObjectId string) {
|
||||||
|
object.ObjectId = "astaxie" + strconv.FormatInt(time.Now().UnixNano(), 10)
|
||||||
|
Objects[object.ObjectId] = &object
|
||||||
|
return object.ObjectId
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetOne(ObjectId string) (object *Object, err error) {
|
||||||
|
if v, ok := Objects[ObjectId]; ok {
|
||||||
|
return v, nil
|
||||||
|
}
|
||||||
|
return nil, errors.New("ObjectId Not Exist")
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetAll() map[string]*Object {
|
||||||
|
return Objects
|
||||||
|
}
|
||||||
|
|
||||||
|
func Update(ObjectId string, Score int64) (err error) {
|
||||||
|
if v, ok := Objects[ObjectId]; ok {
|
||||||
|
v.Score = Score
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return errors.New("ObjectId Not Exist")
|
||||||
|
}
|
||||||
|
|
||||||
|
func Delete(ObjectId string) {
|
||||||
|
delete(Objects, ObjectId)
|
||||||
|
}
|
||||||
|
|
||||||
|
`
|
||||||
|
|
||||||
|
var HproseModels2 = `package models
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
UserList map[string]*User
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
UserList = make(map[string]*User)
|
||||||
|
u := User{"user_11111", "astaxie", "11111", Profile{"male", 20, "Singapore", "astaxie@gmail.com"}}
|
||||||
|
UserList["user_11111"] = &u
|
||||||
|
}
|
||||||
|
|
||||||
|
type User struct {
|
||||||
|
Id string
|
||||||
|
Username string
|
||||||
|
Password string
|
||||||
|
Profile Profile
|
||||||
|
}
|
||||||
|
|
||||||
|
type Profile struct {
|
||||||
|
Gender string
|
||||||
|
Age int
|
||||||
|
Address string
|
||||||
|
Email string
|
||||||
|
}
|
||||||
|
|
||||||
|
func AddUser(u User) string {
|
||||||
|
u.Id = "user_" + strconv.FormatInt(time.Now().UnixNano(), 10)
|
||||||
|
UserList[u.Id] = &u
|
||||||
|
return u.Id
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetUser(uid string) (u *User, err error) {
|
||||||
|
if u, ok := UserList[uid]; ok {
|
||||||
|
return u, nil
|
||||||
|
}
|
||||||
|
return nil, errors.New("User not exists")
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetAllUsers() map[string]*User {
|
||||||
|
return UserList
|
||||||
|
}
|
||||||
|
|
||||||
|
func UpdateUser(uid string, uu *User) (a *User, err error) {
|
||||||
|
if u, ok := UserList[uid]; ok {
|
||||||
|
if uu.Username != "" {
|
||||||
|
u.Username = uu.Username
|
||||||
|
}
|
||||||
|
if uu.Password != "" {
|
||||||
|
u.Password = uu.Password
|
||||||
|
}
|
||||||
|
if uu.Profile.Age != 0 {
|
||||||
|
u.Profile.Age = uu.Profile.Age
|
||||||
|
}
|
||||||
|
if uu.Profile.Address != "" {
|
||||||
|
u.Profile.Address = uu.Profile.Address
|
||||||
|
}
|
||||||
|
if uu.Profile.Gender != "" {
|
||||||
|
u.Profile.Gender = uu.Profile.Gender
|
||||||
|
}
|
||||||
|
if uu.Profile.Email != "" {
|
||||||
|
u.Profile.Email = uu.Profile.Email
|
||||||
|
}
|
||||||
|
return u, nil
|
||||||
|
}
|
||||||
|
return nil, errors.New("User Not Exist")
|
||||||
|
}
|
||||||
|
|
||||||
|
func Login(username, password string) bool {
|
||||||
|
for _, u := range UserList {
|
||||||
|
if u.Username == username && u.Password == password {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func DeleteUser(uid string) {
|
||||||
|
delete(UserList, uid)
|
||||||
|
}
|
||||||
|
`
|
||||||
|
var HproseAddFunctions = []string{}
|
||||||
|
|
||||||
|
func GenerateHproseAppcode(driver, connStr, level, tables, currpath string) {
|
||||||
var mode byte
|
var mode byte
|
||||||
switch level {
|
switch level {
|
||||||
case "1":
|
case "1":
|
||||||
|
@ -38,7 +283,7 @@ func generateHproseAppcode(driver, connStr, level, tables, currpath string) {
|
||||||
case "3":
|
case "3":
|
||||||
mode = OModel | OController | ORouter
|
mode = OModel | OController | ORouter
|
||||||
default:
|
default:
|
||||||
logger.Fatal("Invalid 'level' option. Level must be either \"1\", \"2\" or \"3\"")
|
beeLogger.Log.Fatal("Invalid 'level' option. Level must be either \"1\", \"2\" or \"3\"")
|
||||||
}
|
}
|
||||||
var selectedTables map[string]bool
|
var selectedTables map[string]bool
|
||||||
if tables != "" {
|
if tables != "" {
|
||||||
|
@ -51,9 +296,9 @@ func generateHproseAppcode(driver, connStr, level, tables, currpath string) {
|
||||||
case "mysql":
|
case "mysql":
|
||||||
case "postgres":
|
case "postgres":
|
||||||
case "sqlite":
|
case "sqlite":
|
||||||
logger.Fatal("Generating app code from SQLite database is not supported yet")
|
beeLogger.Log.Fatal("Generating app code from SQLite database is not supported yet")
|
||||||
default:
|
default:
|
||||||
logger.Fatalf("Unknown database driver '%s'. Driver must be one of mysql, postgres or sqlite", driver)
|
beeLogger.Log.Fatalf("Unknown database driver '%s'. Driver must be one of mysql, postgres or sqlite", driver)
|
||||||
}
|
}
|
||||||
genHprose(driver, connStr, mode, selectedTables, currpath)
|
genHprose(driver, connStr, mode, selectedTables, currpath)
|
||||||
}
|
}
|
||||||
|
@ -63,11 +308,11 @@ func generateHproseAppcode(driver, connStr, level, tables, currpath string) {
|
||||||
func genHprose(dbms, connStr string, mode byte, selectedTableNames map[string]bool, currpath string) {
|
func genHprose(dbms, connStr string, mode byte, selectedTableNames map[string]bool, currpath string) {
|
||||||
db, err := sql.Open(dbms, connStr)
|
db, err := sql.Open(dbms, connStr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Could not connect to '%s' database using '%s': %s", dbms, connStr, err)
|
beeLogger.Log.Fatalf("Could not connect to '%s' database using '%s': %s", dbms, connStr, err)
|
||||||
}
|
}
|
||||||
defer db.Close()
|
defer db.Close()
|
||||||
if trans, ok := dbDriver[dbms]; ok {
|
if trans, ok := dbDriver[dbms]; ok {
|
||||||
logger.Info("Analyzing database tables...")
|
beeLogger.Log.Info("Analyzing database tables...")
|
||||||
tableNames := trans.GetTableNames(db)
|
tableNames := trans.GetTableNames(db)
|
||||||
tables := getTableObjects(tableNames, db, trans)
|
tables := getTableObjects(tableNames, db, trans)
|
||||||
mvcPath := new(MvcPath)
|
mvcPath := new(MvcPath)
|
||||||
|
@ -76,7 +321,7 @@ func genHprose(dbms, connStr string, mode byte, selectedTableNames map[string]bo
|
||||||
pkgPath := getPackagePath(currpath)
|
pkgPath := getPackagePath(currpath)
|
||||||
writeHproseSourceFiles(pkgPath, tables, mode, mvcPath, selectedTableNames)
|
writeHproseSourceFiles(pkgPath, tables, mode, mvcPath, selectedTableNames)
|
||||||
} else {
|
} else {
|
||||||
logger.Fatalf("Generating app code from '%s' database is not supported yet", dbms)
|
beeLogger.Log.Fatalf("Generating app code from '%s' database is not supported yet", dbms)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -85,14 +330,14 @@ func genHprose(dbms, connStr string, mode byte, selectedTableNames map[string]bo
|
||||||
// Newly geneated files will be inside these folders.
|
// Newly geneated files will be inside these folders.
|
||||||
func writeHproseSourceFiles(pkgPath string, tables []*Table, mode byte, paths *MvcPath, selectedTables map[string]bool) {
|
func writeHproseSourceFiles(pkgPath string, tables []*Table, mode byte, paths *MvcPath, selectedTables map[string]bool) {
|
||||||
if (OModel & mode) == OModel {
|
if (OModel & mode) == OModel {
|
||||||
logger.Info("Creating model files...")
|
beeLogger.Log.Info("Creating model files...")
|
||||||
writeHproseModelFiles(tables, paths.ModelPath, selectedTables)
|
writeHproseModelFiles(tables, paths.ModelPath, selectedTables)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// writeHproseModelFiles generates model files
|
// writeHproseModelFiles generates model files
|
||||||
func writeHproseModelFiles(tables []*Table, mPath string, selectedTables map[string]bool) {
|
func writeHproseModelFiles(tables []*Table, mPath string, selectedTables map[string]bool) {
|
||||||
w := NewColorWriter(os.Stdout)
|
w := colors.NewColorWriter(os.Stdout)
|
||||||
|
|
||||||
for _, tb := range tables {
|
for _, tb := range tables {
|
||||||
// if selectedTables map is not nil and this table is not selected, ignore it
|
// if selectedTables map is not nil and this table is not selected, ignore it
|
||||||
|
@ -105,34 +350,34 @@ func writeHproseModelFiles(tables []*Table, mPath string, selectedTables map[str
|
||||||
fpath := path.Join(mPath, filename+".go")
|
fpath := path.Join(mPath, filename+".go")
|
||||||
var f *os.File
|
var f *os.File
|
||||||
var err error
|
var err error
|
||||||
if isExist(fpath) {
|
if utils.IsExist(fpath) {
|
||||||
logger.Warnf("'%s' already exists. Do you want to overwrite it? [Yes|No] ", fpath)
|
beeLogger.Log.Warnf("'%s' already exists. Do you want to overwrite it? [Yes|No] ", fpath)
|
||||||
if askForConfirmation() {
|
if utils.AskForConfirmation() {
|
||||||
f, err = os.OpenFile(fpath, os.O_RDWR|os.O_TRUNC, 0666)
|
f, err = os.OpenFile(fpath, os.O_RDWR|os.O_TRUNC, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Warnf("%s", err)
|
beeLogger.Log.Warnf("%s", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
logger.Warnf("Skipped create file '%s'", fpath)
|
beeLogger.Log.Warnf("Skipped create file '%s'", fpath)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
f, err = os.OpenFile(fpath, os.O_CREATE|os.O_RDWR, 0666)
|
f, err = os.OpenFile(fpath, os.O_CREATE|os.O_RDWR, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Warnf("%s", err)
|
beeLogger.Log.Warnf("%s", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
template := ""
|
var template string
|
||||||
if tb.Pk == "" {
|
if tb.Pk == "" {
|
||||||
template = HproseStructModelTPL
|
template = HproseStructModelTPL
|
||||||
} else {
|
} else {
|
||||||
template = HproseModelTPL
|
template = HproseModelTPL
|
||||||
hproseAddFunctions = append(hproseAddFunctions, strings.Replace(HproseAddFunction, "{{modelName}}", camelCase(tb.Name), -1))
|
HproseAddFunctions = append(HproseAddFunctions, strings.Replace(HproseAddFunction, "{{modelName}}", utils.CamelCase(tb.Name), -1))
|
||||||
}
|
}
|
||||||
fileStr := strings.Replace(template, "{{modelStruct}}", tb.String(), 1)
|
fileStr := strings.Replace(template, "{{modelStruct}}", tb.String(), 1)
|
||||||
fileStr = strings.Replace(fileStr, "{{modelName}}", camelCase(tb.Name), -1)
|
fileStr = strings.Replace(fileStr, "{{modelName}}", utils.CamelCase(tb.Name), -1)
|
||||||
// if table contains time field, import time.Time package
|
// if table contains time field, import time.Time package
|
||||||
timePkg := ""
|
timePkg := ""
|
||||||
importTimePkg := ""
|
importTimePkg := ""
|
||||||
|
@ -143,11 +388,11 @@ func writeHproseModelFiles(tables []*Table, mPath string, selectedTables map[str
|
||||||
fileStr = strings.Replace(fileStr, "{{timePkg}}", timePkg, -1)
|
fileStr = strings.Replace(fileStr, "{{timePkg}}", timePkg, -1)
|
||||||
fileStr = strings.Replace(fileStr, "{{importTimePkg}}", importTimePkg, -1)
|
fileStr = strings.Replace(fileStr, "{{importTimePkg}}", importTimePkg, -1)
|
||||||
if _, err := f.WriteString(fileStr); err != nil {
|
if _, err := f.WriteString(fileStr); err != nil {
|
||||||
logger.Fatalf("Could not write model file to '%s'", fpath)
|
beeLogger.Log.Fatalf("Could not write model file to '%s'", fpath)
|
||||||
}
|
}
|
||||||
CloseFile(f)
|
utils.CloseFile(f)
|
||||||
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
||||||
formatSourceCode(fpath)
|
utils.FormatSourceCode(fpath)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -174,7 +419,7 @@ import (
|
||||||
"reflect"
|
"reflect"
|
||||||
"strings"
|
"strings"
|
||||||
{{timePkg}}
|
{{timePkg}}
|
||||||
"github.com/astaxie/beego/orm"
|
"github.com/beego/beego/v2/client/orm"
|
||||||
)
|
)
|
||||||
|
|
||||||
{{modelStruct}}
|
{{modelStruct}}
|
|
@ -12,7 +12,7 @@
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
|
|
||||||
package main
|
package generate
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
@ -20,6 +20,10 @@ import (
|
||||||
"path"
|
"path"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
|
@ -29,18 +33,18 @@ const (
|
||||||
)
|
)
|
||||||
|
|
||||||
type DBDriver interface {
|
type DBDriver interface {
|
||||||
generateCreateUp(tableName string) string
|
GenerateCreateUp(tableName string) string
|
||||||
generateCreateDown(tableName string) string
|
GenerateCreateDown(tableName string) string
|
||||||
}
|
}
|
||||||
|
|
||||||
type mysqlDriver struct{}
|
type mysqlDriver struct{}
|
||||||
|
|
||||||
func (m mysqlDriver) generateCreateUp(tableName string) string {
|
func (m mysqlDriver) GenerateCreateUp(tableName string) string {
|
||||||
upsql := `m.SQL("CREATE TABLE ` + tableName + "(" + m.generateSQLFromFields(fields.String()) + `)");`
|
upsql := `m.SQL("CREATE TABLE ` + tableName + "(" + m.generateSQLFromFields(Fields.String()) + `)");`
|
||||||
return upsql
|
return upsql
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m mysqlDriver) generateCreateDown(tableName string) string {
|
func (m mysqlDriver) GenerateCreateDown(tableName string) string {
|
||||||
downsql := `m.SQL("DROP TABLE ` + "`" + tableName + "`" + `")`
|
downsql := `m.SQL("DROP TABLE ` + "`" + tableName + "`" + `")`
|
||||||
return downsql
|
return downsql
|
||||||
}
|
}
|
||||||
|
@ -51,21 +55,21 @@ func (m mysqlDriver) generateSQLFromFields(fields string) string {
|
||||||
for i, v := range fds {
|
for i, v := range fds {
|
||||||
kv := strings.SplitN(v, ":", 2)
|
kv := strings.SplitN(v, ":", 2)
|
||||||
if len(kv) != 2 {
|
if len(kv) != 2 {
|
||||||
logger.Error("Fields format is wrong. Should be: key:type,key:type " + v)
|
beeLogger.Log.Error("Fields format is wrong. Should be: key:type,key:type " + v)
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
typ, tag := m.getSQLType(kv[1])
|
typ, tag := m.getSQLType(kv[1])
|
||||||
if typ == "" {
|
if typ == "" {
|
||||||
logger.Error("Fields format is wrong. Should be: key:type,key:type " + v)
|
beeLogger.Log.Error("Fields format is wrong. Should be: key:type,key:type " + v)
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
if i == 0 && strings.ToLower(kv[0]) != "id" {
|
if i == 0 && strings.ToLower(kv[0]) != "id" {
|
||||||
sql += "`id` int(11) NOT NULL AUTO_INCREMENT,"
|
sql += "`id` int(11) NOT NULL AUTO_INCREMENT,"
|
||||||
tags = tags + "PRIMARY KEY (`id`),"
|
tags = tags + "PRIMARY KEY (`id`),"
|
||||||
}
|
}
|
||||||
sql += "`" + snakeString(kv[0]) + "` " + typ + ","
|
sql += "`" + utils.SnakeString(kv[0]) + "` " + typ + ","
|
||||||
if tag != "" {
|
if tag != "" {
|
||||||
tags = tags + fmt.Sprintf(tag, "`"+snakeString(kv[0])+"`") + ","
|
tags = tags + fmt.Sprintf(tag, "`"+utils.SnakeString(kv[0])+"`") + ","
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
sql = strings.TrimRight(sql+tags, ",")
|
sql = strings.TrimRight(sql+tags, ",")
|
||||||
|
@ -104,12 +108,12 @@ func (m mysqlDriver) getSQLType(ktype string) (tp, tag string) {
|
||||||
|
|
||||||
type postgresqlDriver struct{}
|
type postgresqlDriver struct{}
|
||||||
|
|
||||||
func (m postgresqlDriver) generateCreateUp(tableName string) string {
|
func (m postgresqlDriver) GenerateCreateUp(tableName string) string {
|
||||||
upsql := `m.SQL("CREATE TABLE ` + tableName + "(" + m.generateSQLFromFields(fields.String()) + `)");`
|
upsql := `m.SQL("CREATE TABLE ` + tableName + "(" + m.generateSQLFromFields(Fields.String()) + `)");`
|
||||||
return upsql
|
return upsql
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m postgresqlDriver) generateCreateDown(tableName string) string {
|
func (m postgresqlDriver) GenerateCreateDown(tableName string) string {
|
||||||
downsql := `m.SQL("DROP TABLE ` + tableName + `")`
|
downsql := `m.SQL("DROP TABLE ` + tableName + `")`
|
||||||
return downsql
|
return downsql
|
||||||
}
|
}
|
||||||
|
@ -120,20 +124,20 @@ func (m postgresqlDriver) generateSQLFromFields(fields string) string {
|
||||||
for i, v := range fds {
|
for i, v := range fds {
|
||||||
kv := strings.SplitN(v, ":", 2)
|
kv := strings.SplitN(v, ":", 2)
|
||||||
if len(kv) != 2 {
|
if len(kv) != 2 {
|
||||||
logger.Error("Fields format is wrong. Should be: key:type,key:type " + v)
|
beeLogger.Log.Error("Fields format is wrong. Should be: key:type,key:type " + v)
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
typ, tag := m.getSQLType(kv[1])
|
typ, tag := m.getSQLType(kv[1])
|
||||||
if typ == "" {
|
if typ == "" {
|
||||||
logger.Error("Fields format is wrong. Should be: key:type,key:type " + v)
|
beeLogger.Log.Error("Fields format is wrong. Should be: key:type,key:type " + v)
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
if i == 0 && strings.ToLower(kv[0]) != "id" {
|
if i == 0 && strings.ToLower(kv[0]) != "id" {
|
||||||
sql += "id serial primary key,"
|
sql += "id serial primary key,"
|
||||||
}
|
}
|
||||||
sql += snakeString(kv[0]) + " " + typ + ","
|
sql += utils.SnakeString(kv[0]) + " " + typ + ","
|
||||||
if tag != "" {
|
if tag != "" {
|
||||||
tags = tags + fmt.Sprintf(tag, snakeString(kv[0])) + ","
|
tags = tags + fmt.Sprintf(tag, utils.SnakeString(kv[0])) + ","
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if tags != "" {
|
if tags != "" {
|
||||||
|
@ -170,14 +174,14 @@ func (m postgresqlDriver) getSQLType(ktype string) (tp, tag string) {
|
||||||
return "", ""
|
return "", ""
|
||||||
}
|
}
|
||||||
|
|
||||||
func newDBDriver() DBDriver {
|
func NewDBDriver() DBDriver {
|
||||||
switch driver {
|
switch SQLDriver {
|
||||||
case "mysql":
|
case "mysql":
|
||||||
return mysqlDriver{}
|
return mysqlDriver{}
|
||||||
case "postgres":
|
case "postgres":
|
||||||
return postgresqlDriver{}
|
return postgresqlDriver{}
|
||||||
default:
|
default:
|
||||||
logger.Fatal("Driver not supported")
|
beeLogger.Log.Fatal("Driver not supported")
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -185,60 +189,102 @@ func newDBDriver() DBDriver {
|
||||||
// generateMigration generates migration file template for database schema update.
|
// generateMigration generates migration file template for database schema update.
|
||||||
// The generated file template consists of an up() method for updating schema and
|
// The generated file template consists of an up() method for updating schema and
|
||||||
// a down() method for reverting the update.
|
// a down() method for reverting the update.
|
||||||
func generateMigration(mname, upsql, downsql, curpath string) {
|
func GenerateMigration(mname, upsql, downsql, curpath string) {
|
||||||
w := NewColorWriter(os.Stdout)
|
w := colors.NewColorWriter(os.Stdout)
|
||||||
migrationFilePath := path.Join(curpath, DBPath, MPath)
|
migrationFilePath := path.Join(curpath, DBPath, MPath)
|
||||||
if _, err := os.Stat(migrationFilePath); os.IsNotExist(err) {
|
if _, err := os.Stat(migrationFilePath); os.IsNotExist(err) {
|
||||||
// create migrations directory
|
// create migrations directory
|
||||||
if err := os.MkdirAll(migrationFilePath, 0777); err != nil {
|
if err := os.MkdirAll(migrationFilePath, 0777); err != nil {
|
||||||
logger.Fatalf("Could not create migration directory: %s", err)
|
beeLogger.Log.Fatalf("Could not create migration directory: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// create file
|
// create file
|
||||||
today := time.Now().Format(MDateFormat)
|
today := time.Now().Format(MDateFormat)
|
||||||
fpath := path.Join(migrationFilePath, fmt.Sprintf("%s_%s.go", today, mname))
|
fpath := path.Join(migrationFilePath, fmt.Sprintf("%s_%s.go", today, mname))
|
||||||
if f, err := os.OpenFile(fpath, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
if f, err := os.OpenFile(fpath, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
||||||
defer CloseFile(f)
|
defer utils.CloseFile(f)
|
||||||
content := strings.Replace(MigrationTPL, "{{StructName}}", camelCase(mname)+"_"+today, -1)
|
ddlSpec := ""
|
||||||
content = strings.Replace(content, "{{CurrTime}}", today, -1)
|
spec := ""
|
||||||
content = strings.Replace(content, "{{UpSQL}}", upsql, -1)
|
up := ""
|
||||||
content = strings.Replace(content, "{{DownSQL}}", downsql, -1)
|
down := ""
|
||||||
f.WriteString(content)
|
if DDL != "" {
|
||||||
|
ddlSpec = "m.ddlSpec()"
|
||||||
|
switch strings.Title(DDL.String()) {
|
||||||
|
case "Create":
|
||||||
|
spec = strings.Replace(DDLSpecCreate, "{{StructName}}", utils.CamelCase(mname)+"_"+today, -1)
|
||||||
|
case "Alter":
|
||||||
|
spec = strings.Replace(DDLSpecAlter, "{{StructName}}", utils.CamelCase(mname)+"_"+today, -1)
|
||||||
|
}
|
||||||
|
spec = strings.Replace(spec, "{{tableName}}", mname, -1)
|
||||||
|
} else {
|
||||||
|
up = strings.Replace(MigrationUp, "{{UpSQL}}", upsql, -1)
|
||||||
|
up = strings.Replace(up, "{{StructName}}", utils.CamelCase(mname)+"_"+today, -1)
|
||||||
|
down = strings.Replace(MigrationDown, "{{DownSQL}}", downsql, -1)
|
||||||
|
down = strings.Replace(down, "{{StructName}}", utils.CamelCase(mname)+"_"+today, -1)
|
||||||
|
}
|
||||||
|
|
||||||
|
header := strings.Replace(MigrationHeader, "{{StructName}}", utils.CamelCase(mname)+"_"+today, -1)
|
||||||
|
header = strings.Replace(header, "{{ddlSpec}}", ddlSpec, -1)
|
||||||
|
header = strings.Replace(header, "{{CurrTime}}", today, -1)
|
||||||
|
f.WriteString(header + spec + up + down)
|
||||||
// Run 'gofmt' on the generated source code
|
// Run 'gofmt' on the generated source code
|
||||||
formatSourceCode(fpath)
|
utils.FormatSourceCode(fpath)
|
||||||
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
||||||
} else {
|
} else {
|
||||||
logger.Fatalf("Could not create migration file: %s", err)
|
beeLogger.Log.Fatalf("Could not create migration file: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const MigrationTPL = `package main
|
const (
|
||||||
|
MigrationHeader = `package main
|
||||||
|
import (
|
||||||
|
"github.com/beego/beego/v2/client/orm/migration"
|
||||||
|
)
|
||||||
|
|
||||||
import (
|
// DO NOT MODIFY
|
||||||
"github.com/astaxie/beego/migration"
|
type {{StructName}} struct {
|
||||||
|
migration.Migration
|
||||||
|
}
|
||||||
|
|
||||||
|
// DO NOT MODIFY
|
||||||
|
func init() {
|
||||||
|
m := &{{StructName}}{}
|
||||||
|
m.Created = "{{CurrTime}}"
|
||||||
|
{{ddlSpec}}
|
||||||
|
migration.Register("{{StructName}}", m)
|
||||||
|
}
|
||||||
|
`
|
||||||
|
|
||||||
|
DDLSpecCreate = `
|
||||||
|
/*
|
||||||
|
refer beego/migration/doc.go
|
||||||
|
*/
|
||||||
|
func(m *{{StructName}}) ddlSpec(){
|
||||||
|
m.CreateTable("{{tableName}}", "InnoDB", "utf8")
|
||||||
|
m.PriCol("id").SetAuto(true).SetNullable(false).SetDataType("INT(10)").SetUnsigned(true)
|
||||||
|
|
||||||
|
}
|
||||||
|
`
|
||||||
|
DDLSpecAlter = `
|
||||||
|
/*
|
||||||
|
refer beego/migration/doc.go
|
||||||
|
*/
|
||||||
|
func(m *{{StructName}}) ddlSpec(){
|
||||||
|
m.AlterTable("{{tableName}}")
|
||||||
|
|
||||||
|
}
|
||||||
|
`
|
||||||
|
MigrationUp = `
|
||||||
|
// Run the migrations
|
||||||
|
func (m *{{StructName}}) Up() {
|
||||||
|
// use m.SQL("CREATE TABLE ...") to make schema update
|
||||||
|
{{UpSQL}}
|
||||||
|
}`
|
||||||
|
MigrationDown = `
|
||||||
|
// Reverse the migrations
|
||||||
|
func (m *{{StructName}}) Down() {
|
||||||
|
// use m.SQL("DROP TABLE ...") to reverse schema update
|
||||||
|
{{DownSQL}}
|
||||||
|
}
|
||||||
|
`
|
||||||
)
|
)
|
||||||
|
|
||||||
// DO NOT MODIFY
|
|
||||||
type {{StructName}} struct {
|
|
||||||
migration.Migration
|
|
||||||
}
|
|
||||||
|
|
||||||
// DO NOT MODIFY
|
|
||||||
func init() {
|
|
||||||
m := &{{StructName}}{}
|
|
||||||
m.Created = "{{CurrTime}}"
|
|
||||||
migration.Register("{{StructName}}", m)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Run the migrations
|
|
||||||
func (m *{{StructName}}) Up() {
|
|
||||||
// use m.SQL("CREATE TABLE ...") to make schema update
|
|
||||||
{{UpSQL}}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Reverse the migrations
|
|
||||||
func (m *{{StructName}}) Down() {
|
|
||||||
// use m.SQL("DROP TABLE ...") to reverse schema update
|
|
||||||
{{DownSQL}}
|
|
||||||
}
|
|
||||||
`
|
|
|
@ -12,7 +12,7 @@
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
|
|
||||||
package main
|
package generate
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
@ -20,10 +20,14 @@ import (
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
func generateModel(mname, fields, currpath string) {
|
func GenerateModel(mname, fields, currpath string) {
|
||||||
w := NewColorWriter(os.Stdout)
|
w := colors.NewColorWriter(os.Stdout)
|
||||||
|
|
||||||
p, f := path.Split(mname)
|
p, f := path.Split(mname)
|
||||||
modelName := strings.Title(f)
|
modelName := strings.Title(f)
|
||||||
|
@ -35,23 +39,23 @@ func generateModel(mname, fields, currpath string) {
|
||||||
|
|
||||||
modelStruct, hastime, err := getStruct(modelName, fields)
|
modelStruct, hastime, err := getStruct(modelName, fields)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Could not generate the model struct: %s", err)
|
beeLogger.Log.Fatalf("Could not generate the model struct: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.Infof("Using '%s' as model name", modelName)
|
beeLogger.Log.Infof("Using '%s' as model name", modelName)
|
||||||
logger.Infof("Using '%s' as package name", packageName)
|
beeLogger.Log.Infof("Using '%s' as package name", packageName)
|
||||||
|
|
||||||
fp := path.Join(currpath, "models", p)
|
fp := path.Join(currpath, "models", p)
|
||||||
if _, err := os.Stat(fp); os.IsNotExist(err) {
|
if _, err := os.Stat(fp); os.IsNotExist(err) {
|
||||||
// Create the model's directory
|
// Create the model's directory
|
||||||
if err := os.MkdirAll(fp, 0777); err != nil {
|
if err := os.MkdirAll(fp, 0777); err != nil {
|
||||||
logger.Fatalf("Could not create the model directory: %s", err)
|
beeLogger.Log.Fatalf("Could not create the model directory: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fpath := path.Join(fp, strings.ToLower(modelName)+".go")
|
fpath := path.Join(fp, strings.ToLower(modelName)+".go")
|
||||||
if f, err := os.OpenFile(fpath, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
if f, err := os.OpenFile(fpath, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
||||||
defer CloseFile(f)
|
defer utils.CloseFile(f)
|
||||||
content := strings.Replace(modelTpl, "{{packageName}}", packageName, -1)
|
content := strings.Replace(modelTpl, "{{packageName}}", packageName, -1)
|
||||||
content = strings.Replace(content, "{{modelName}}", modelName, -1)
|
content = strings.Replace(content, "{{modelName}}", modelName, -1)
|
||||||
content = strings.Replace(content, "{{modelStruct}}", modelStruct, -1)
|
content = strings.Replace(content, "{{modelStruct}}", modelStruct, -1)
|
||||||
|
@ -62,10 +66,10 @@ func generateModel(mname, fields, currpath string) {
|
||||||
}
|
}
|
||||||
f.WriteString(content)
|
f.WriteString(content)
|
||||||
// Run 'gofmt' on the generated source code
|
// Run 'gofmt' on the generated source code
|
||||||
formatSourceCode(fpath)
|
utils.FormatSourceCode(fpath)
|
||||||
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", fpath, "\x1b[0m")
|
||||||
} else {
|
} else {
|
||||||
logger.Fatalf("Could not create model file: %s", err)
|
beeLogger.Log.Fatalf("Could not create model file: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -95,7 +99,7 @@ func getStruct(structname, fields string) (string, bool, error) {
|
||||||
if hastimeinner {
|
if hastimeinner {
|
||||||
hastime = true
|
hastime = true
|
||||||
}
|
}
|
||||||
structStr = structStr + camelString(kv[0]) + " " + typ + " " + tag + "\n"
|
structStr = structStr + utils.CamelString(kv[0]) + " " + typ + " " + tag + "\n"
|
||||||
}
|
}
|
||||||
structStr += "}\n"
|
structStr += "}\n"
|
||||||
return structStr, hastime, nil
|
return structStr, hastime, nil
|
||||||
|
@ -141,7 +145,7 @@ import (
|
||||||
"reflect"
|
"reflect"
|
||||||
"strings"
|
"strings"
|
||||||
{{timePkg}}
|
{{timePkg}}
|
||||||
"github.com/astaxie/beego/orm"
|
"github.com/beego/beego/v2/client/orm"
|
||||||
)
|
)
|
||||||
|
|
||||||
{{modelStruct}}
|
{{modelStruct}}
|
||||||
|
@ -163,7 +167,7 @@ func Add{{modelName}}(m *{{modelName}}) (id int64, err error) {
|
||||||
func Get{{modelName}}ById(id int64) (v *{{modelName}}, err error) {
|
func Get{{modelName}}ById(id int64) (v *{{modelName}}, err error) {
|
||||||
o := orm.NewOrm()
|
o := orm.NewOrm()
|
||||||
v = &{{modelName}}{Id: id}
|
v = &{{modelName}}{Id: id}
|
||||||
if err = o.Read(v); err == nil {
|
if err = o.QueryTable(new({{modelName}})).Filter("Id", id).RelatedSel().One(v); err == nil {
|
||||||
return v, nil
|
return v, nil
|
||||||
}
|
}
|
||||||
return nil, err
|
return nil, err
|
||||||
|
@ -221,7 +225,7 @@ func GetAll{{modelName}}(query map[string]string, fields []string, sortby []stri
|
||||||
}
|
}
|
||||||
|
|
||||||
var l []{{modelName}}
|
var l []{{modelName}}
|
||||||
qs = qs.OrderBy(sortFields...)
|
qs = qs.OrderBy(sortFields...).RelatedSel()
|
||||||
if _, err = qs.Limit(limit, offset).All(&l, fields...); err == nil {
|
if _, err = qs.Limit(limit, offset).All(&l, fields...); err == nil {
|
||||||
if len(fields) == 0 {
|
if len(fields) == 0 {
|
||||||
for _, v := range l {
|
for _, v := range l {
|
50
generate/g_scaffold.go
Normal file
50
generate/g_scaffold.go
Normal file
|
@ -0,0 +1,50 @@
|
||||||
|
package generate
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd/commands/migrate"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
func GenerateScaffold(sname, fields, currpath, driver, conn string) {
|
||||||
|
beeLogger.Log.Infof("Do you want to create a '%s' model? [Yes|No] ", sname)
|
||||||
|
|
||||||
|
// Generate the model
|
||||||
|
if utils.AskForConfirmation() {
|
||||||
|
GenerateModel(sname, fields, currpath)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generate the controller
|
||||||
|
beeLogger.Log.Infof("Do you want to create a '%s' controller? [Yes|No] ", sname)
|
||||||
|
if utils.AskForConfirmation() {
|
||||||
|
GenerateController(sname, currpath)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generate the views
|
||||||
|
beeLogger.Log.Infof("Do you want to create views for this '%s' resource? [Yes|No] ", sname)
|
||||||
|
if utils.AskForConfirmation() {
|
||||||
|
GenerateView(sname, currpath)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generate a migration
|
||||||
|
beeLogger.Log.Infof("Do you want to create a '%s' migration and schema for this resource? [Yes|No] ", sname)
|
||||||
|
if utils.AskForConfirmation() {
|
||||||
|
upsql := ""
|
||||||
|
downsql := ""
|
||||||
|
if fields != "" {
|
||||||
|
dbMigrator := NewDBDriver()
|
||||||
|
upsql = dbMigrator.GenerateCreateUp(sname)
|
||||||
|
downsql = dbMigrator.GenerateCreateDown(sname)
|
||||||
|
}
|
||||||
|
GenerateMigration(sname, upsql, downsql, currpath)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Run the migration
|
||||||
|
beeLogger.Log.Infof("Do you want to migrate the database? [Yes|No] ")
|
||||||
|
if utils.AskForConfirmation() {
|
||||||
|
migrate.MigrateUpdate(currpath, driver, conn, "")
|
||||||
|
}
|
||||||
|
beeLogger.Log.Successf("All done! Don't forget to add beego.Router(\"/%s\" ,&controllers.%sController{}) to routers/route.go\n", sname, strings.Title(sname))
|
||||||
|
}
|
|
@ -12,60 +12,64 @@
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
|
|
||||||
package main
|
package generate
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
// recipe
|
// recipe
|
||||||
// admin/recipe
|
// admin/recipe
|
||||||
func generateView(viewpath, currpath string) {
|
func GenerateView(viewpath, currpath string) {
|
||||||
w := NewColorWriter(os.Stdout)
|
w := colors.NewColorWriter(os.Stdout)
|
||||||
|
|
||||||
logger.Info("Generating view...")
|
beeLogger.Log.Info("Generating view...")
|
||||||
|
|
||||||
absViewPath := path.Join(currpath, "views", viewpath)
|
absViewPath := path.Join(currpath, "views", viewpath)
|
||||||
err := os.MkdirAll(absViewPath, os.ModePerm)
|
err := os.MkdirAll(absViewPath, os.ModePerm)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("Could not create '%s' view: %s", viewpath, err)
|
beeLogger.Log.Fatalf("Could not create '%s' view: %s", viewpath, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
cfile := path.Join(absViewPath, "index.tpl")
|
cfile := path.Join(absViewPath, "index.tpl")
|
||||||
if f, err := os.OpenFile(cfile, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
if f, err := os.OpenFile(cfile, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
||||||
defer CloseFile(f)
|
defer utils.CloseFile(f)
|
||||||
f.WriteString(cfile)
|
f.WriteString(cfile)
|
||||||
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", cfile, "\x1b[0m")
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", cfile, "\x1b[0m")
|
||||||
} else {
|
} else {
|
||||||
logger.Fatalf("Could not create view file: %s", err)
|
beeLogger.Log.Fatalf("Could not create view file: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
cfile = path.Join(absViewPath, "show.tpl")
|
cfile = path.Join(absViewPath, "show.tpl")
|
||||||
if f, err := os.OpenFile(cfile, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
if f, err := os.OpenFile(cfile, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
||||||
defer CloseFile(f)
|
defer utils.CloseFile(f)
|
||||||
f.WriteString(cfile)
|
f.WriteString(cfile)
|
||||||
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", cfile, "\x1b[0m")
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", cfile, "\x1b[0m")
|
||||||
} else {
|
} else {
|
||||||
logger.Fatalf("Could not create view file: %s", err)
|
beeLogger.Log.Fatalf("Could not create view file: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
cfile = path.Join(absViewPath, "create.tpl")
|
cfile = path.Join(absViewPath, "create.tpl")
|
||||||
if f, err := os.OpenFile(cfile, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
if f, err := os.OpenFile(cfile, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
||||||
defer CloseFile(f)
|
defer utils.CloseFile(f)
|
||||||
f.WriteString(cfile)
|
f.WriteString(cfile)
|
||||||
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", cfile, "\x1b[0m")
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", cfile, "\x1b[0m")
|
||||||
} else {
|
} else {
|
||||||
logger.Fatalf("Could not create view file: %s", err)
|
beeLogger.Log.Fatalf("Could not create view file: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
cfile = path.Join(absViewPath, "edit.tpl")
|
cfile = path.Join(absViewPath, "edit.tpl")
|
||||||
if f, err := os.OpenFile(cfile, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
if f, err := os.OpenFile(cfile, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil {
|
||||||
defer CloseFile(f)
|
defer utils.CloseFile(f)
|
||||||
f.WriteString(cfile)
|
f.WriteString(cfile)
|
||||||
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", cfile, "\x1b[0m")
|
fmt.Fprintf(w, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", cfile, "\x1b[0m")
|
||||||
} else {
|
} else {
|
||||||
logger.Fatalf("Could not create view file: %s", err)
|
beeLogger.Log.Fatalf("Could not create view file: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
549
generate/gen_routes.go
Normal file
549
generate/gen_routes.go
Normal file
|
@ -0,0 +1,549 @@
|
||||||
|
// Copyright 2014 beego Author. All Rights Reserved.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package generate
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"go/ast"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"regexp"
|
||||||
|
"sort"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"unicode"
|
||||||
|
|
||||||
|
"github.com/beego/beego/v2/server/web"
|
||||||
|
"golang.org/x/tools/go/packages"
|
||||||
|
|
||||||
|
"github.com/beego/beego/v2/core/logs"
|
||||||
|
|
||||||
|
"github.com/beego/beego/v2/server/web/context/param"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
var globalRouterTemplate = `package {{.routersDir}}
|
||||||
|
|
||||||
|
import (
|
||||||
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
|
"github.com/beego/beego/v2/server/web/context/param"{{.globalimport}}
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
{{.globalinfo}}
|
||||||
|
}
|
||||||
|
`
|
||||||
|
|
||||||
|
var (
|
||||||
|
genInfoList map[string][]web.ControllerComments
|
||||||
|
|
||||||
|
routerHooks = map[string]int{
|
||||||
|
"beego.BeforeStatic": web.BeforeStatic,
|
||||||
|
"beego.BeforeRouter": web.BeforeRouter,
|
||||||
|
"beego.BeforeExec": web.BeforeExec,
|
||||||
|
"beego.AfterExec": web.AfterExec,
|
||||||
|
"beego.FinishRouter": web.FinishRouter,
|
||||||
|
}
|
||||||
|
|
||||||
|
routerHooksMapping = map[int]string{
|
||||||
|
web.BeforeStatic: "beego.BeforeStatic",
|
||||||
|
web.BeforeRouter: "beego.BeforeRouter",
|
||||||
|
web.BeforeExec: "beego.BeforeExec",
|
||||||
|
web.AfterExec: "beego.AfterExec",
|
||||||
|
web.FinishRouter: "beego.FinishRouter",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
func GenRouters() {
|
||||||
|
ctrlPath := ControllerDirectory.String()
|
||||||
|
if len(ctrlPath) == 0 {
|
||||||
|
ctrlPath = "controllers"
|
||||||
|
}
|
||||||
|
routersPath := RoutersFile.String()
|
||||||
|
if len(routersPath) == 0 {
|
||||||
|
routersPath = "routers/commentsRouter.go"
|
||||||
|
}
|
||||||
|
genRouters(ctrlPath, routersPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
func genRouters(ctrlPath string, routersPath string) {
|
||||||
|
beeLogger.Log.Infof("read controller info from directory[%s], and output routers to [%s]", ctrlPath, routersPath)
|
||||||
|
err := parserPkg(ctrlPath, routersPath)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Could not generate routers. Please check your input: %+v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func parserPkg(ctrlPath string, routersPath string) error {
|
||||||
|
genInfoList = make(map[string][]web.ControllerComments)
|
||||||
|
pkgs, err := packages.Load(&packages.Config{
|
||||||
|
Mode: packages.NeedName | packages.NeedFiles | packages.NeedCompiledGoFiles | packages.NeedSyntax,
|
||||||
|
Dir: ctrlPath,
|
||||||
|
}, "./...")
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, pkg := range pkgs {
|
||||||
|
for _, fl := range pkg.Syntax {
|
||||||
|
for _, d := range fl.Decls {
|
||||||
|
switch specDecl := d.(type) {
|
||||||
|
case *ast.FuncDecl:
|
||||||
|
if specDecl.Recv != nil {
|
||||||
|
exp, ok := specDecl.Recv.List[0].Type.(*ast.StarExpr) // Check that the type is correct first beforing throwing to parser
|
||||||
|
if ok {
|
||||||
|
err = parserComments(specDecl, fmt.Sprint(exp.X), pkg.PkgPath)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return genRouterCode(routersPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
type parsedComment struct {
|
||||||
|
routerPath string
|
||||||
|
methods []string
|
||||||
|
params map[string]parsedParam
|
||||||
|
filters []parsedFilter
|
||||||
|
imports []parsedImport
|
||||||
|
}
|
||||||
|
|
||||||
|
type parsedImport struct {
|
||||||
|
importPath string
|
||||||
|
importAlias string
|
||||||
|
}
|
||||||
|
|
||||||
|
type parsedFilter struct {
|
||||||
|
pattern string
|
||||||
|
pos int
|
||||||
|
filter string
|
||||||
|
params []bool
|
||||||
|
}
|
||||||
|
|
||||||
|
type parsedParam struct {
|
||||||
|
name string
|
||||||
|
datatype string
|
||||||
|
location string
|
||||||
|
defValue string
|
||||||
|
required bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func parserComments(f *ast.FuncDecl, controllerName, pkgpath string) error {
|
||||||
|
if f.Doc != nil {
|
||||||
|
parsedComments, err := parseComment(f.Doc.List)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, parsedComment := range parsedComments {
|
||||||
|
if parsedComment.routerPath != "" {
|
||||||
|
key := pkgpath + ":" + controllerName
|
||||||
|
cc := web.ControllerComments{}
|
||||||
|
cc.Method = f.Name.String()
|
||||||
|
cc.Router = parsedComment.routerPath
|
||||||
|
cc.AllowHTTPMethods = parsedComment.methods
|
||||||
|
cc.MethodParams = buildMethodParams(f.Type.Params.List, parsedComment)
|
||||||
|
cc.FilterComments = buildFilters(parsedComment.filters)
|
||||||
|
cc.ImportComments = buildImports(parsedComment.imports)
|
||||||
|
genInfoList[key] = append(genInfoList[key], cc)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func buildImports(pis []parsedImport) []*web.ControllerImportComments {
|
||||||
|
var importComments []*web.ControllerImportComments
|
||||||
|
|
||||||
|
for _, pi := range pis {
|
||||||
|
importComments = append(importComments, &web.ControllerImportComments{
|
||||||
|
ImportPath: pi.importPath,
|
||||||
|
ImportAlias: pi.importAlias,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return importComments
|
||||||
|
}
|
||||||
|
|
||||||
|
func buildFilters(pfs []parsedFilter) []*web.ControllerFilterComments {
|
||||||
|
var filterComments []*web.ControllerFilterComments
|
||||||
|
|
||||||
|
for _, pf := range pfs {
|
||||||
|
var (
|
||||||
|
returnOnOutput bool
|
||||||
|
resetParams bool
|
||||||
|
)
|
||||||
|
|
||||||
|
if len(pf.params) >= 1 {
|
||||||
|
returnOnOutput = pf.params[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(pf.params) >= 2 {
|
||||||
|
resetParams = pf.params[1]
|
||||||
|
}
|
||||||
|
|
||||||
|
filterComments = append(filterComments, &web.ControllerFilterComments{
|
||||||
|
Filter: pf.filter,
|
||||||
|
Pattern: pf.pattern,
|
||||||
|
Pos: pf.pos,
|
||||||
|
ReturnOnOutput: returnOnOutput,
|
||||||
|
ResetParams: resetParams,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return filterComments
|
||||||
|
}
|
||||||
|
|
||||||
|
func buildMethodParams(funcParams []*ast.Field, pc *parsedComment) []*param.MethodParam {
|
||||||
|
result := make([]*param.MethodParam, 0, len(funcParams))
|
||||||
|
for _, fparam := range funcParams {
|
||||||
|
for _, pName := range fparam.Names {
|
||||||
|
methodParam := buildMethodParam(fparam, pName.Name, pc)
|
||||||
|
result = append(result, methodParam)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
func buildMethodParam(fparam *ast.Field, name string, pc *parsedComment) *param.MethodParam {
|
||||||
|
options := []param.MethodParamOption{}
|
||||||
|
if cparam, ok := pc.params[name]; ok {
|
||||||
|
// Build param from comment info
|
||||||
|
name = cparam.name
|
||||||
|
if cparam.required {
|
||||||
|
options = append(options, param.IsRequired)
|
||||||
|
}
|
||||||
|
switch cparam.location {
|
||||||
|
case "body":
|
||||||
|
options = append(options, param.InBody)
|
||||||
|
case "header":
|
||||||
|
options = append(options, param.InHeader)
|
||||||
|
case "path":
|
||||||
|
options = append(options, param.InPath)
|
||||||
|
}
|
||||||
|
if cparam.defValue != "" {
|
||||||
|
options = append(options, param.Default(cparam.defValue))
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if paramInPath(name, pc.routerPath) {
|
||||||
|
options = append(options, param.InPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return param.New(name, options...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func paramInPath(name, route string) bool {
|
||||||
|
return strings.HasSuffix(route, ":"+name) ||
|
||||||
|
strings.Contains(route, ":"+name+"/")
|
||||||
|
}
|
||||||
|
|
||||||
|
var routeRegex = regexp.MustCompile(`@router\s+(\S+)(?:\s+\[(\S+)\])?`)
|
||||||
|
|
||||||
|
func parseComment(lines []*ast.Comment) (pcs []*parsedComment, err error) {
|
||||||
|
pcs = []*parsedComment{}
|
||||||
|
params := map[string]parsedParam{}
|
||||||
|
filters := []parsedFilter{}
|
||||||
|
imports := []parsedImport{}
|
||||||
|
|
||||||
|
const doubleSlash = "/"
|
||||||
|
for _, c := range lines {
|
||||||
|
t := strings.TrimSpace(strings.TrimLeft(c.Text, doubleSlash))
|
||||||
|
const paramPrefix = "@Param"
|
||||||
|
if strings.HasPrefix(t, paramPrefix) {
|
||||||
|
pv := getparams(strings.TrimSpace(strings.TrimPrefix(t, paramPrefix)))
|
||||||
|
if len(pv) < 4 {
|
||||||
|
logs.Error("Invalid @Param format. Needs at least 4 parameters")
|
||||||
|
}
|
||||||
|
p := parsedParam{}
|
||||||
|
names := strings.SplitN(pv[0], "=>", 2)
|
||||||
|
p.name = names[0]
|
||||||
|
funcParamName := p.name
|
||||||
|
if len(names) > 1 {
|
||||||
|
funcParamName = names[1]
|
||||||
|
}
|
||||||
|
p.location = pv[1]
|
||||||
|
p.datatype = pv[2]
|
||||||
|
switch len(pv) {
|
||||||
|
case 5:
|
||||||
|
p.required, _ = strconv.ParseBool(pv[3])
|
||||||
|
case 6:
|
||||||
|
p.defValue = pv[3]
|
||||||
|
p.required, _ = strconv.ParseBool(pv[4])
|
||||||
|
}
|
||||||
|
params[funcParamName] = p
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range lines {
|
||||||
|
t := strings.TrimSpace(strings.TrimLeft(c.Text, doubleSlash))
|
||||||
|
if strings.HasPrefix(t, "@Import") {
|
||||||
|
iv := getparams(strings.TrimSpace(strings.TrimLeft(t, "@Import")))
|
||||||
|
if len(iv) == 0 || len(iv) > 2 {
|
||||||
|
logs.Error("Invalid @Import format. Only accepts 1 or 2 parameters")
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
p := parsedImport{}
|
||||||
|
p.importPath = iv[0]
|
||||||
|
|
||||||
|
if len(iv) == 2 {
|
||||||
|
p.importAlias = iv[1]
|
||||||
|
}
|
||||||
|
|
||||||
|
imports = append(imports, p)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
filterLoop:
|
||||||
|
for _, c := range lines {
|
||||||
|
t := strings.TrimSpace(strings.TrimLeft(c.Text, doubleSlash))
|
||||||
|
if strings.HasPrefix(t, "@Filter") {
|
||||||
|
fv := getparams(strings.TrimSpace(strings.TrimLeft(t, "@Filter")))
|
||||||
|
if len(fv) < 3 {
|
||||||
|
logs.Error("Invalid @Filter format. Needs at least 3 parameters")
|
||||||
|
continue filterLoop
|
||||||
|
}
|
||||||
|
|
||||||
|
p := parsedFilter{}
|
||||||
|
p.pattern = fv[0]
|
||||||
|
posName := fv[1]
|
||||||
|
if pos, exists := routerHooks[posName]; exists {
|
||||||
|
p.pos = pos
|
||||||
|
} else {
|
||||||
|
logs.Error("Invalid @Filter pos: ", posName)
|
||||||
|
continue filterLoop
|
||||||
|
}
|
||||||
|
|
||||||
|
p.filter = fv[2]
|
||||||
|
fvParams := fv[3:]
|
||||||
|
for _, fvParam := range fvParams {
|
||||||
|
switch fvParam {
|
||||||
|
case "true":
|
||||||
|
p.params = append(p.params, true)
|
||||||
|
case "false":
|
||||||
|
p.params = append(p.params, false)
|
||||||
|
default:
|
||||||
|
logs.Error("Invalid @Filter param: ", fvParam)
|
||||||
|
continue filterLoop
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
filters = append(filters, p)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range lines {
|
||||||
|
var pc = &parsedComment{}
|
||||||
|
pc.params = params
|
||||||
|
pc.filters = filters
|
||||||
|
pc.imports = imports
|
||||||
|
|
||||||
|
t := strings.TrimSpace(strings.TrimLeft(c.Text, doubleSlash))
|
||||||
|
if strings.HasPrefix(t, "@router") {
|
||||||
|
t := strings.TrimSpace(strings.TrimLeft(c.Text, doubleSlash))
|
||||||
|
matches := routeRegex.FindStringSubmatch(t)
|
||||||
|
if len(matches) == 3 {
|
||||||
|
pc.routerPath = matches[1]
|
||||||
|
methods := matches[2]
|
||||||
|
if methods == "" {
|
||||||
|
pc.methods = []string{"get"}
|
||||||
|
// pc.hasGet = true
|
||||||
|
} else {
|
||||||
|
pc.methods = strings.Split(methods, ",")
|
||||||
|
// pc.hasGet = strings.Contains(methods, "get")
|
||||||
|
}
|
||||||
|
pcs = append(pcs, pc)
|
||||||
|
} else {
|
||||||
|
return nil, errors.New("router information is missing")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// direct copy from bee\g_docs.go
|
||||||
|
// analysis params return []string
|
||||||
|
// @Param query form string true "The email for login"
|
||||||
|
// [query form string true "The email for login"]
|
||||||
|
func getparams(str string) []string {
|
||||||
|
var s []rune
|
||||||
|
var j int
|
||||||
|
var start bool
|
||||||
|
var r []string
|
||||||
|
var quoted int8
|
||||||
|
for _, c := range str {
|
||||||
|
if unicode.IsSpace(c) && quoted == 0 {
|
||||||
|
if !start {
|
||||||
|
continue
|
||||||
|
} else {
|
||||||
|
start = false
|
||||||
|
j++
|
||||||
|
r = append(r, string(s))
|
||||||
|
s = make([]rune, 0)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
start = true
|
||||||
|
if c == '"' {
|
||||||
|
quoted ^= 1
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
s = append(s, c)
|
||||||
|
}
|
||||||
|
if len(s) > 0 {
|
||||||
|
r = append(r, string(s))
|
||||||
|
}
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func genRouterCode(routersPath string) error {
|
||||||
|
logs.Info("generate router from comments")
|
||||||
|
var (
|
||||||
|
globalinfo string
|
||||||
|
globalimport string
|
||||||
|
sortKey []string
|
||||||
|
)
|
||||||
|
for k := range genInfoList {
|
||||||
|
sortKey = append(sortKey, k)
|
||||||
|
}
|
||||||
|
sort.Strings(sortKey)
|
||||||
|
for _, k := range sortKey {
|
||||||
|
cList := genInfoList[k]
|
||||||
|
sort.Sort(web.ControllerCommentsSlice(cList))
|
||||||
|
for _, c := range cList {
|
||||||
|
allmethod := "nil"
|
||||||
|
if len(c.AllowHTTPMethods) > 0 {
|
||||||
|
allmethod = "[]string{"
|
||||||
|
for _, m := range c.AllowHTTPMethods {
|
||||||
|
allmethod += `"` + m + `",`
|
||||||
|
}
|
||||||
|
allmethod = strings.TrimRight(allmethod, ",") + "}"
|
||||||
|
}
|
||||||
|
|
||||||
|
params := "nil"
|
||||||
|
if len(c.Params) > 0 {
|
||||||
|
params = "[]map[string]string{"
|
||||||
|
for _, p := range c.Params {
|
||||||
|
for k, v := range p {
|
||||||
|
params = params + `map[string]string{` + k + `:"` + v + `"},`
|
||||||
|
}
|
||||||
|
}
|
||||||
|
params = strings.TrimRight(params, ",") + "}"
|
||||||
|
}
|
||||||
|
|
||||||
|
methodParams := "param.Make("
|
||||||
|
if len(c.MethodParams) > 0 {
|
||||||
|
lines := make([]string, 0, len(c.MethodParams))
|
||||||
|
for _, m := range c.MethodParams {
|
||||||
|
lines = append(lines, fmt.Sprint(m))
|
||||||
|
}
|
||||||
|
methodParams += "\n " +
|
||||||
|
strings.Join(lines, ",\n ") +
|
||||||
|
",\n "
|
||||||
|
}
|
||||||
|
methodParams += ")"
|
||||||
|
|
||||||
|
imports := ""
|
||||||
|
if len(c.ImportComments) > 0 {
|
||||||
|
for _, i := range c.ImportComments {
|
||||||
|
var s string
|
||||||
|
if i.ImportAlias != "" {
|
||||||
|
s = fmt.Sprintf(`
|
||||||
|
%s "%s"`, i.ImportAlias, i.ImportPath)
|
||||||
|
} else {
|
||||||
|
s = fmt.Sprintf(`
|
||||||
|
"%s"`, i.ImportPath)
|
||||||
|
}
|
||||||
|
if !strings.Contains(globalimport, s) {
|
||||||
|
imports += s
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
filters := ""
|
||||||
|
if len(c.FilterComments) > 0 {
|
||||||
|
for _, f := range c.FilterComments {
|
||||||
|
filters += fmt.Sprintf(` &beego.ControllerFilter{
|
||||||
|
Pattern: "%s",
|
||||||
|
Pos: %s,
|
||||||
|
Filter: %s,
|
||||||
|
ReturnOnOutput: %v,
|
||||||
|
ResetParams: %v,
|
||||||
|
},`, f.Pattern, routerHooksMapping[f.Pos], f.Filter, f.ReturnOnOutput, f.ResetParams)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if filters == "" {
|
||||||
|
filters = "nil"
|
||||||
|
} else {
|
||||||
|
filters = fmt.Sprintf(`[]*beego.ControllerFilter{
|
||||||
|
%s
|
||||||
|
}`, filters)
|
||||||
|
}
|
||||||
|
|
||||||
|
globalimport += imports
|
||||||
|
|
||||||
|
globalinfo = globalinfo + `
|
||||||
|
beego.GlobalControllerRouter["` + k + `"] = append(beego.GlobalControllerRouter["` + k + `"],
|
||||||
|
beego.ControllerComments{
|
||||||
|
Method: "` + strings.TrimSpace(c.Method) + `",
|
||||||
|
` + "Router: `" + c.Router + "`" + `,
|
||||||
|
AllowHTTPMethods: ` + allmethod + `,
|
||||||
|
MethodParams: ` + methodParams + `,
|
||||||
|
Filters: ` + filters + `,
|
||||||
|
Params: ` + params + `})
|
||||||
|
`
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if globalinfo != "" {
|
||||||
|
|
||||||
|
routersPathDir := filepath.Dir(routersPath)
|
||||||
|
err := os.MkdirAll(routersPathDir, os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
f, err := os.Create(routersPath)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
routersDir := RouterPkg.String()
|
||||||
|
if len(routersDir) == 0{
|
||||||
|
routersDir = "routers"
|
||||||
|
}
|
||||||
|
|
||||||
|
beeLogger.Log.Infof("using %s as routers file's package", routersDir)
|
||||||
|
|
||||||
|
content := strings.Replace(globalRouterTemplate, "{{.globalinfo}}", globalinfo, -1)
|
||||||
|
content = strings.Replace(content, "{{.routersDir}}", routersDir, -1)
|
||||||
|
content = strings.Replace(content, "{{.globalimport}}", globalimport, -1)
|
||||||
|
_, err = f.WriteString(content)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
1455
generate/swaggergen/g_docs.go
Normal file
1455
generate/swaggergen/g_docs.go
Normal file
File diff suppressed because it is too large
Load Diff
224
generate/swaggergen/go_docs_test.go
Normal file
224
generate/swaggergen/go_docs_test.go
Normal file
|
@ -0,0 +1,224 @@
|
||||||
|
// Copyright 2013 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
|
||||||
|
package swaggergen
|
||||||
|
|
||||||
|
import (
|
||||||
|
_ "github.com/shopspring/decimal"
|
||||||
|
"go/ast"
|
||||||
|
"go/build"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// package model
|
||||||
|
//
|
||||||
|
// import (
|
||||||
|
// "sync"
|
||||||
|
//
|
||||||
|
// "example.com/pkgnotexist"
|
||||||
|
// "github.com/shopspring/decimal"
|
||||||
|
// )
|
||||||
|
//
|
||||||
|
// type Object struct {
|
||||||
|
// Field1 decimal.Decimal
|
||||||
|
// Field2 pkgnotexist.TestType
|
||||||
|
// Field3 sync.Map
|
||||||
|
// }
|
||||||
|
func TestCheckAndLoadPackageOnGoMod(t *testing.T) {
|
||||||
|
defer os.Setenv("GO111MODULE", os.Getenv("GO111MODULE"))
|
||||||
|
os.Setenv("GO111MODULE", "on")
|
||||||
|
|
||||||
|
testCases := []struct {
|
||||||
|
pkgName string
|
||||||
|
pkgImportPath string
|
||||||
|
imports []*ast.ImportSpec
|
||||||
|
realType string
|
||||||
|
curPkgName string
|
||||||
|
expected bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
pkgName: "decimal",
|
||||||
|
pkgImportPath: "github.com/shopspring/decimal",
|
||||||
|
imports: []*ast.ImportSpec{
|
||||||
|
{
|
||||||
|
Path: &ast.BasicLit{
|
||||||
|
Value: "github.com/shopspring/decimal",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
realType: "decimal.Decimal",
|
||||||
|
curPkgName: "model",
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
pkgName: "pkgnotexist",
|
||||||
|
pkgImportPath: "example.com/pkgnotexist",
|
||||||
|
imports: []*ast.ImportSpec{
|
||||||
|
{
|
||||||
|
Path: &ast.BasicLit{
|
||||||
|
Value: "example.com/pkgnotexist",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
realType: "pkgnotexist.TestType",
|
||||||
|
curPkgName: "model",
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
pkgName: "sync",
|
||||||
|
pkgImportPath: "sync",
|
||||||
|
imports: []*ast.ImportSpec{
|
||||||
|
{
|
||||||
|
Path: &ast.BasicLit{
|
||||||
|
Value: "sync",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
realType: "sync.Map",
|
||||||
|
curPkgName: "model",
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, test := range testCases {
|
||||||
|
checkAndLoadPackage(test.imports, test.realType, test.curPkgName)
|
||||||
|
result := false
|
||||||
|
for _, v := range astPkgs {
|
||||||
|
if v.Name == test.pkgName {
|
||||||
|
result = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if test.expected != result {
|
||||||
|
t.Fatalf("load module error, expected: %v, result: %v", test.expected, result)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// package model
|
||||||
|
//
|
||||||
|
// import (
|
||||||
|
// "sync"
|
||||||
|
//
|
||||||
|
// "example.com/comm"
|
||||||
|
// "example.com/pkgnotexist"
|
||||||
|
// )
|
||||||
|
//
|
||||||
|
// type Object struct {
|
||||||
|
// Field1 comm.Common
|
||||||
|
// Field2 pkgnotexist.TestType
|
||||||
|
// Field3 sync.Map
|
||||||
|
// }
|
||||||
|
func TestCheckAndLoadPackageOnGoPath(t *testing.T) {
|
||||||
|
var (
|
||||||
|
testCommPkg = `
|
||||||
|
package comm
|
||||||
|
|
||||||
|
type Common struct {
|
||||||
|
Code string
|
||||||
|
Error string
|
||||||
|
}
|
||||||
|
`
|
||||||
|
)
|
||||||
|
|
||||||
|
gopath, err := ioutil.TempDir("", "gobuild-gopath")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer os.RemoveAll(gopath)
|
||||||
|
|
||||||
|
if err := os.MkdirAll(filepath.Join(gopath, "src/example.com/comm"), 0777); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := ioutil.WriteFile(filepath.Join(gopath, "src/example.com/comm/comm.go"), []byte(testCommPkg), 0666); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer os.Setenv("GO111MODULE", os.Getenv("GO111MODULE"))
|
||||||
|
os.Setenv("GO111MODULE", "off")
|
||||||
|
defer os.Setenv("GOPATH", os.Getenv("GOPATH"))
|
||||||
|
os.Setenv("GOPATH", gopath)
|
||||||
|
build.Default.GOPATH = gopath
|
||||||
|
|
||||||
|
testCases := []struct {
|
||||||
|
pkgName string
|
||||||
|
pkgImportPath string
|
||||||
|
imports []*ast.ImportSpec
|
||||||
|
realType string
|
||||||
|
curPkgName string
|
||||||
|
expected bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
pkgName: "comm",
|
||||||
|
pkgImportPath: "example.com/comm",
|
||||||
|
imports: []*ast.ImportSpec{
|
||||||
|
{
|
||||||
|
Path: &ast.BasicLit{
|
||||||
|
Value: "example.com/comm",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
realType: "comm.Common",
|
||||||
|
curPkgName: "model",
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
pkgName: "pkgnotexist",
|
||||||
|
pkgImportPath: "example.com/pkgnotexist",
|
||||||
|
imports: []*ast.ImportSpec{
|
||||||
|
{
|
||||||
|
Path: &ast.BasicLit{
|
||||||
|
Value: "example.com/pkgnotexist",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
realType: "pkgnotexist.TestType",
|
||||||
|
curPkgName: "model",
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
pkgName: "sync",
|
||||||
|
pkgImportPath: "sync",
|
||||||
|
imports: []*ast.ImportSpec{
|
||||||
|
{
|
||||||
|
Path: &ast.BasicLit{
|
||||||
|
Value: "sync",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
realType: "sync.Map",
|
||||||
|
curPkgName: "model",
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, test := range testCases {
|
||||||
|
checkAndLoadPackage(test.imports, test.realType, test.curPkgName)
|
||||||
|
result := false
|
||||||
|
for _, v := range astPkgs {
|
||||||
|
if v.Name == test.pkgName {
|
||||||
|
result = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if test.expected != result {
|
||||||
|
t.Fatalf("load module error, expected: %v, result: %v", test.expected, result)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
63
go.mod
Normal file
63
go.mod
Normal file
|
@ -0,0 +1,63 @@
|
||||||
|
module github.com/beego/bee/v2
|
||||||
|
|
||||||
|
go 1.18
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/beego/beego/v2 v2.1.0
|
||||||
|
github.com/davecgh/go-spew v1.1.1
|
||||||
|
github.com/flosch/pongo2 v0.0.0-20200529170236-5abacdfa4915
|
||||||
|
github.com/fsnotify/fsnotify v1.4.9
|
||||||
|
github.com/go-delve/delve v1.20.2
|
||||||
|
github.com/go-sql-driver/mysql v1.7.0
|
||||||
|
github.com/gorilla/websocket v1.4.2
|
||||||
|
github.com/lib/pq v1.10.5
|
||||||
|
github.com/pelletier/go-toml v1.9.2
|
||||||
|
github.com/shopspring/decimal v1.3.1
|
||||||
|
github.com/smartwalle/pongo2render v1.0.1
|
||||||
|
github.com/spf13/viper v1.7.0
|
||||||
|
golang.org/x/tools v0.1.12
|
||||||
|
gopkg.in/yaml.v2 v2.4.0
|
||||||
|
)
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/beorn7/perks v1.0.1 // indirect
|
||||||
|
github.com/cespare/xxhash/v2 v2.2.0 // indirect
|
||||||
|
github.com/cilium/ebpf v0.7.0 // indirect
|
||||||
|
github.com/cosiner/argv v0.1.0 // indirect
|
||||||
|
github.com/derekparker/trie v0.0.0-20221213183930-4c74548207f4 // indirect
|
||||||
|
github.com/go-delve/liner v1.2.3-0.20220127212407-d32d89dd2a5d // indirect
|
||||||
|
github.com/golang/protobuf v1.5.3 // indirect
|
||||||
|
github.com/google/go-dap v0.7.0 // indirect
|
||||||
|
github.com/hashicorp/golang-lru v0.5.4 // indirect
|
||||||
|
github.com/hashicorp/hcl v1.0.0 // indirect
|
||||||
|
github.com/konsorten/go-windows-terminal-sequences v1.0.3 // indirect
|
||||||
|
github.com/magiconair/properties v1.8.1 // indirect
|
||||||
|
github.com/mattn/go-colorable v0.0.9 // indirect
|
||||||
|
github.com/mattn/go-isatty v0.0.3 // indirect
|
||||||
|
github.com/mattn/go-runewidth v0.0.13 // indirect
|
||||||
|
github.com/matttproud/golang_protobuf_extensions v1.0.4 // indirect
|
||||||
|
github.com/mitchellh/mapstructure v1.5.0 // indirect
|
||||||
|
github.com/pkg/errors v0.9.1 // indirect
|
||||||
|
github.com/prometheus/client_golang v1.15.1 // indirect
|
||||||
|
github.com/prometheus/client_model v0.3.0 // indirect
|
||||||
|
github.com/prometheus/common v0.42.0 // indirect
|
||||||
|
github.com/prometheus/procfs v0.9.0 // indirect
|
||||||
|
github.com/rivo/uniseg v0.2.0 // indirect
|
||||||
|
github.com/shiena/ansicolor v0.0.0-20200904210342-c7312218db18 // indirect
|
||||||
|
github.com/sirupsen/logrus v1.6.0 // indirect
|
||||||
|
github.com/spf13/afero v1.1.2 // indirect
|
||||||
|
github.com/spf13/cast v1.3.0 // indirect
|
||||||
|
github.com/spf13/jwalterweatherman v1.0.0 // indirect
|
||||||
|
github.com/spf13/pflag v1.0.5 // indirect
|
||||||
|
github.com/subosito/gotenv v1.2.0 // indirect
|
||||||
|
go.starlark.net v0.0.0-20220816155156-cfacd8902214 // indirect
|
||||||
|
golang.org/x/arch v0.0.0-20190927153633-4e8777c89be4 // indirect
|
||||||
|
golang.org/x/crypto v0.0.0-20220315160706-3147a52a75dd // indirect
|
||||||
|
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4 // indirect
|
||||||
|
golang.org/x/net v0.7.0 // indirect
|
||||||
|
golang.org/x/sys v0.6.0 // indirect
|
||||||
|
golang.org/x/text v0.7.0 // indirect
|
||||||
|
google.golang.org/protobuf v1.30.0 // indirect
|
||||||
|
gopkg.in/ini.v1 v1.51.0 // indirect
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
|
)
|
467
go.sum
Normal file
467
go.sum
Normal file
|
@ -0,0 +1,467 @@
|
||||||
|
cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=
|
||||||
|
cloud.google.com/go v0.34.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=
|
||||||
|
cloud.google.com/go v0.38.0/go.mod h1:990N+gfupTy94rShfmMCWGDn0LpTmnzTp2qbd1dvSRU=
|
||||||
|
cloud.google.com/go v0.44.1/go.mod h1:iSa0KzasP4Uvy3f1mN/7PiObzGgflwredwwASm/v6AU=
|
||||||
|
cloud.google.com/go v0.44.2/go.mod h1:60680Gw3Yr4ikxnPRS/oxxkBccT6SA1yMk63TGekxKY=
|
||||||
|
cloud.google.com/go v0.45.1/go.mod h1:RpBamKRgapWJb87xiFSdk4g1CME7QZg3uwTez+TSTjc=
|
||||||
|
cloud.google.com/go v0.46.3/go.mod h1:a6bKKbmY7er1mI7TEI4lsAkts/mkhTSZK8w33B4RAg0=
|
||||||
|
cloud.google.com/go/bigquery v1.0.1/go.mod h1:i/xbL2UlR5RvWAURpBYZTtm/cXjCha9lbfbpx4poX+o=
|
||||||
|
cloud.google.com/go/datastore v1.0.0/go.mod h1:LXYbyblFSglQ5pkeyhO+Qmw7ukd3C+pD7TKLgZqpHYE=
|
||||||
|
cloud.google.com/go/firestore v1.1.0/go.mod h1:ulACoGHTpvq5r8rxGJ4ddJZBZqakUQqClKRT5SZwBmk=
|
||||||
|
cloud.google.com/go/pubsub v1.0.1/go.mod h1:R0Gpsv3s54REJCy4fxDixWD93lHJMoZTyQ2kNxGRt3I=
|
||||||
|
cloud.google.com/go/storage v1.0.0/go.mod h1:IhtSnM/ZTZV8YYJWCY8RULGVqBDmpoyjwiyrjsg+URw=
|
||||||
|
dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7gZCb22OMCxBHrMx7a5I7Hp++hsVxbQ4BYO7hU=
|
||||||
|
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
|
||||||
|
github.com/BurntSushi/xgb v0.0.0-20160522181843-27f122750802/go.mod h1:IVnqGOEym/WlBOVXweHU+Q+/VP0lqqI8lqeDx9IjBqo=
|
||||||
|
github.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU=
|
||||||
|
github.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=
|
||||||
|
github.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=
|
||||||
|
github.com/armon/circbuf v0.0.0-20150827004946-bbbad097214e/go.mod h1:3U/XgcO3hCbHZ8TKRvWD2dDTCfh9M9ya+I9JpbB7O8o=
|
||||||
|
github.com/armon/go-metrics v0.0.0-20180917152333-f0300d1749da/go.mod h1:Q73ZrmVTwzkszR9V5SSuryQ31EELlFMUz1kKyl939pY=
|
||||||
|
github.com/armon/go-radix v0.0.0-20180808171621-7fddfc383310/go.mod h1:ufUuZ+zHj4x4TnLV4JWEpy2hxWSpsRywHrMgIH9cCH8=
|
||||||
|
github.com/beego/beego/v2 v2.1.0 h1:Lk0FtQGvDQCx5V5yEu4XwDsIgt+QOlNjt5emUa3/ZmA=
|
||||||
|
github.com/beego/beego/v2 v2.1.0/go.mod h1:6h36ISpaxNrrpJ27siTpXBG8d/Icjzsc7pU1bWpp0EE=
|
||||||
|
github.com/beorn7/perks v0.0.0-20180321164747-3a771d992973/go.mod h1:Dwedo/Wpr24TaqPxmxbtue+5NUziq4I4S80YR8gNf3Q=
|
||||||
|
github.com/beorn7/perks v1.0.0/go.mod h1:KWe93zE9D1o94FZ5RNwFwVgaQK1VOXiVxmqh+CedLV8=
|
||||||
|
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
||||||
|
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
||||||
|
github.com/bgentry/speakeasy v0.1.0/go.mod h1:+zsyZBPWlz7T6j88CTgSN5bM796AkVf0kBD4zp0CCIs=
|
||||||
|
github.com/bketelsen/crypt v0.0.3-0.20200106085610-5cbc8cc4026c/go.mod h1:MKsuJmJgSg28kpZDP6UIiPt0e0Oz0kqKNGyRaWEPv84=
|
||||||
|
github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=
|
||||||
|
github.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc=
|
||||||
|
github.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=
|
||||||
|
github.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||||
|
github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
|
||||||
|
github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
|
||||||
|
github.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=
|
||||||
|
github.com/cilium/ebpf v0.7.0 h1:1k/q3ATgxSXRdrmPfH8d7YK0GfqVsEKZAX9dQZvs56k=
|
||||||
|
github.com/cilium/ebpf v0.7.0/go.mod h1:/oI2+1shJiTGAMgl6/RgJr36Eo1jzrRcAWbcXO2usCA=
|
||||||
|
github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw=
|
||||||
|
github.com/coreos/bbolt v1.3.2/go.mod h1:iRUV2dpdMOn7Bo10OQBFzIJO9kkE559Wcmn+qkEiiKk=
|
||||||
|
github.com/coreos/etcd v3.3.13+incompatible/go.mod h1:uF7uidLiAD3TWHmW31ZFd/JWoc32PjwdhPthX9715RE=
|
||||||
|
github.com/coreos/go-semver v0.3.0/go.mod h1:nnelYz7RCh+5ahJtPPxZlU+153eP4D4r3EedlOD2RNk=
|
||||||
|
github.com/coreos/go-systemd v0.0.0-20190321100706-95778dfbb74e/go.mod h1:F5haX7vjVVG0kc13fIWeqUViNPyEJxv/OmvnBo0Yme4=
|
||||||
|
github.com/coreos/pkg v0.0.0-20180928190104-399ea9e2e55f/go.mod h1:E3G3o1h8I7cfcXa63jLwjI0eiQQMgzzUDFVpN/nH/eA=
|
||||||
|
github.com/cosiner/argv v0.1.0 h1:BVDiEL32lwHukgJKP87btEPenzrrHUjajs/8yzaqcXg=
|
||||||
|
github.com/cosiner/argv v0.1.0/go.mod h1:EusR6TucWKX+zFgtdUsKT2Cvg45K5rtpCcWz4hK06d8=
|
||||||
|
github.com/cpuguy83/go-md2man/v2 v2.0.0/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU=
|
||||||
|
github.com/creack/pty v1.1.9 h1:uDmaGzcdjhF4i/plgjmEsriH11Y0o7RKapEf/LDaM3w=
|
||||||
|
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
|
||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/derekparker/trie v0.0.0-20221213183930-4c74548207f4 h1:atN94qKNhLpy+9BwbE5nxvFj4rScJi6W3x/NfFmMDg4=
|
||||||
|
github.com/derekparker/trie v0.0.0-20221213183930-4c74548207f4/go.mod h1:C7Es+DLenIpPc9J6IYw4jrK0h7S9bKj4DNl8+KxGEXU=
|
||||||
|
github.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ=
|
||||||
|
github.com/dgryski/go-sip13 v0.0.0-20181026042036-e10d5fee7954/go.mod h1:vAd38F8PWV+bWy6jNmig1y/TA+kYO4g3RSRF0IAv0no=
|
||||||
|
github.com/elazarl/go-bindata-assetfs v1.0.1 h1:m0kkaHRKEu7tUIUFVwhGGGYClXvyl4RE03qmvRTNfbw=
|
||||||
|
github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
|
||||||
|
github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=
|
||||||
|
github.com/fatih/color v1.7.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4=
|
||||||
|
github.com/flosch/pongo2 v0.0.0-20190707114632-bbf5a6c351f4/go.mod h1:T9YF2M40nIgbVgp3rreNmTged+9HrbNTIQf1PsaIiTA=
|
||||||
|
github.com/flosch/pongo2 v0.0.0-20200529170236-5abacdfa4915 h1:rNVrewdFbSujcoKZifC6cHJfqCTbCIR7XTLHW5TqUWU=
|
||||||
|
github.com/flosch/pongo2 v0.0.0-20200529170236-5abacdfa4915/go.mod h1:fB4mx6dzqFinCxIf3a7Mf5yLk+18Bia9mPAnuejcvDA=
|
||||||
|
github.com/frankban/quicktest v1.11.3 h1:8sXhOn0uLys67V8EsXLc6eszDs8VXWxL3iRvebPhedY=
|
||||||
|
github.com/frankban/quicktest v1.11.3/go.mod h1:wRf/ReqHper53s+kmmSZizM8NamnL3IM0I9ntUbOk+k=
|
||||||
|
github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
|
||||||
|
github.com/fsnotify/fsnotify v1.4.9 h1:hsms1Qyu0jgnwNXIxa+/V/PDsU6CfLf6CNO8H7IWoS4=
|
||||||
|
github.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ=
|
||||||
|
github.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04=
|
||||||
|
github.com/go-check/check v0.0.0-20180628173108-788fd7840127 h1:0gkP6mzaMqkmpcJYCFOLkIBwI7xFExG03bbkOkCvUPI=
|
||||||
|
github.com/go-check/check v0.0.0-20180628173108-788fd7840127/go.mod h1:9ES+weclKsC9YodN5RgxqK/VD9HM9JsCSh7rNhMZE98=
|
||||||
|
github.com/go-delve/delve v1.20.2 h1:rgPK7Iqb1oQk+i2Ilg0fpH6p5LqyixYiAt4N3Lhx4/Y=
|
||||||
|
github.com/go-delve/delve v1.20.2/go.mod h1:KQtnLRy2M+cNHCRnDzURxljVNbRTdvVDD5Mb10KGP18=
|
||||||
|
github.com/go-delve/liner v1.2.3-0.20220127212407-d32d89dd2a5d h1:pxjSLshkZJGLVm0wv20f/H0oTWiq/egkoJQ2ja6LEvo=
|
||||||
|
github.com/go-delve/liner v1.2.3-0.20220127212407-d32d89dd2a5d/go.mod h1:biJCRbqp51wS+I92HMqn5H8/A0PAhxn2vyOT+JqhiGI=
|
||||||
|
github.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1/go.mod h1:vR7hzQXu2zJy9AVAgeJqvqgH9Q5CA+iKCZ2gyEVpxRU=
|
||||||
|
github.com/go-kit/kit v0.8.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=
|
||||||
|
github.com/go-logfmt/logfmt v0.3.0/go.mod h1:Qt1PoO58o5twSAckw1HlFXLmHsOX5/0LbT9GBnD5lWE=
|
||||||
|
github.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk=
|
||||||
|
github.com/go-sql-driver/mysql v1.7.0 h1:ueSltNNllEqE3qcWBTD0iQd3IpL/6U+mJxLkazJ7YPc=
|
||||||
|
github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=
|
||||||
|
github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=
|
||||||
|
github.com/gogo/protobuf v1.1.1/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=
|
||||||
|
github.com/gogo/protobuf v1.2.1/go.mod h1:hp+jE20tsWTFYpLwKvXlhS1hjn+gTNwPg2I6zVXpSg4=
|
||||||
|
github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=
|
||||||
|
github.com/golang/groupcache v0.0.0-20190129154638-5b532d6fd5ef/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=
|
||||||
|
github.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=
|
||||||
|
github.com/golang/mock v1.2.0/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=
|
||||||
|
github.com/golang/mock v1.3.1/go.mod h1:sBzyDLLjw3U8JLTeZvSv8jJB+tU5PVekmnlKIyFUx0Y=
|
||||||
|
github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
||||||
|
github.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
||||||
|
github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
||||||
|
github.com/golang/protobuf v1.3.5/go.mod h1:6O5/vntMXwX2lRkT1hjjk0nAC1IDOTvTlVgjlRvqsdk=
|
||||||
|
github.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8=
|
||||||
|
github.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA=
|
||||||
|
github.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs=
|
||||||
|
github.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w=
|
||||||
|
github.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0=
|
||||||
|
github.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8=
|
||||||
|
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
|
||||||
|
github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg=
|
||||||
|
github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
|
||||||
|
github.com/google/btree v0.0.0-20180813153112-4030bb1f1f0c/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ=
|
||||||
|
github.com/google/btree v1.0.0/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ=
|
||||||
|
github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=
|
||||||
|
github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
|
||||||
|
github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
|
||||||
|
github.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
|
github.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
|
github.com/google/go-cmp v0.5.1/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
|
github.com/google/go-cmp v0.5.4/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
|
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
|
github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
|
||||||
|
github.com/google/go-dap v0.7.0 h1:088PdKBUkxAxrXrnY8FREUJXpS6Y6jhAyZIuJv3OGOM=
|
||||||
|
github.com/google/go-dap v0.7.0/go.mod h1:5q8aYQFnHOAZEMP+6vmq25HKYAEwE+LF5yh7JKrrhSQ=
|
||||||
|
github.com/google/martian v2.1.0+incompatible/go.mod h1:9I4somxYTbIHy5NJKHRl3wXiIaQGbYVAs8BPL6v8lEs=
|
||||||
|
github.com/google/pprof v0.0.0-20181206194817-3ea8567a2e57/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc=
|
||||||
|
github.com/google/pprof v0.0.0-20190515194954-54271f7e092f/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc=
|
||||||
|
github.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=
|
||||||
|
github.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+vpHVxEJEs9eg=
|
||||||
|
github.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk=
|
||||||
|
github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1 h1:EGx4pi6eqNxGaHF6qqu48+N2wcFQ5qg5FXgOdqsJ5d8=
|
||||||
|
github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY=
|
||||||
|
github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=
|
||||||
|
github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
||||||
|
github.com/grpc-ecosystem/go-grpc-middleware v1.0.0/go.mod h1:FiyG127CGDf3tlThmgyCl78X/SZQqEOJBCDaAfeWzPs=
|
||||||
|
github.com/grpc-ecosystem/go-grpc-prometheus v1.2.0/go.mod h1:8NvIoxWQoOIhqOTXgfV/d3M/q6VIi02HzZEHgUlZvzk=
|
||||||
|
github.com/grpc-ecosystem/grpc-gateway v1.9.0/go.mod h1:vNeuVxBJEsws4ogUvrchl83t/GYV9WGTSLVdBhOQFDY=
|
||||||
|
github.com/hashicorp/consul/api v1.1.0/go.mod h1:VmuI/Lkw1nC05EYQWNKwWGbkg+FbDBtguAZLlVdkD9Q=
|
||||||
|
github.com/hashicorp/consul/sdk v0.1.1/go.mod h1:VKf9jXwCTEY1QZP2MOLRhb5i/I/ssyNV1vwHyQBF0x8=
|
||||||
|
github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
|
||||||
|
github.com/hashicorp/go-cleanhttp v0.5.1/go.mod h1:JpRdi6/HCYpAwUzNwuwqhbovhLtngrth3wmdIIUrZ80=
|
||||||
|
github.com/hashicorp/go-immutable-radix v1.0.0/go.mod h1:0y9vanUI8NX6FsYoO3zeMjhV/C5i9g4Q3DwcSNZ4P60=
|
||||||
|
github.com/hashicorp/go-msgpack v0.5.3/go.mod h1:ahLV/dePpqEmjfWmKiqvPkv/twdG7iPBM1vqhUKIvfM=
|
||||||
|
github.com/hashicorp/go-multierror v1.0.0/go.mod h1:dHtQlpGsu+cZNNAkkCN/P3hoUDHhCYQXV3UM06sGGrk=
|
||||||
|
github.com/hashicorp/go-rootcerts v1.0.0/go.mod h1:K6zTfqpRlCUIjkwsN4Z+hiSfzSTQa6eBIzfwKfwNnHU=
|
||||||
|
github.com/hashicorp/go-sockaddr v1.0.0/go.mod h1:7Xibr9yA9JjQq1JpNB2Vw7kxv8xerXegt+ozgdvDeDU=
|
||||||
|
github.com/hashicorp/go-syslog v1.0.0/go.mod h1:qPfqrKkXGihmCqbJM2mZgkZGvKG1dFdvsLplgctolz4=
|
||||||
|
github.com/hashicorp/go-uuid v1.0.0/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro=
|
||||||
|
github.com/hashicorp/go-uuid v1.0.1/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro=
|
||||||
|
github.com/hashicorp/go.net v0.0.1/go.mod h1:hjKkEWcCURg++eb33jQU7oqQcI9XDCnUzHA0oac0k90=
|
||||||
|
github.com/hashicorp/golang-lru v0.5.0/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=
|
||||||
|
github.com/hashicorp/golang-lru v0.5.1/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=
|
||||||
|
github.com/hashicorp/golang-lru v0.5.4 h1:YDjusn29QI/Das2iO9M0BHnIbxPeyuCHsjMW+lJfyTc=
|
||||||
|
github.com/hashicorp/golang-lru v0.5.4/go.mod h1:iADmTwqILo4mZ8BN3D2Q6+9jd8WM5uGBxy+E8yxSoD4=
|
||||||
|
github.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4=
|
||||||
|
github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=
|
||||||
|
github.com/hashicorp/logutils v1.0.0/go.mod h1:QIAnNjmIWmVIIkWDTG1z5v++HQmx9WQRO+LraFDTW64=
|
||||||
|
github.com/hashicorp/mdns v1.0.0/go.mod h1:tL+uN++7HEJ6SQLQ2/p+z2pH24WQKWjBPkE0mNTz8vQ=
|
||||||
|
github.com/hashicorp/memberlist v0.1.3/go.mod h1:ajVTdAv/9Im8oMAAj5G31PhhMCZJV2pPBoIllUwCN7I=
|
||||||
|
github.com/hashicorp/serf v0.8.2/go.mod h1:6hOLApaqBFA1NXqRQAsxw9QxuDEvNxSQRwA/JwenrHc=
|
||||||
|
github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8=
|
||||||
|
github.com/jonboulle/clockwork v0.1.0/go.mod h1:Ii8DK3G1RaLaWxj9trq07+26W01tbo22gdxWY5EU2bo=
|
||||||
|
github.com/json-iterator/go v1.1.6/go.mod h1:+SdeFBvtyEkXs7REEP0seUULqWtbJapLOCVDaaPEHmU=
|
||||||
|
github.com/jstemmer/go-junit-report v0.0.0-20190106144839-af01ea7f8024/go.mod h1:6v2b51hI/fHJwM22ozAgKL4VKDeJcHhJFhtBdhmNjmU=
|
||||||
|
github.com/jtolds/gls v4.20.0+incompatible h1:xdiiI2gbIgH/gLH7ADydsJ1uDOEzR8yvV7C0MuV77Wo=
|
||||||
|
github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU=
|
||||||
|
github.com/juju/errors v0.0.0-20181118221551-089d3ea4e4d5/go.mod h1:W54LbzXuIE0boCoNJfwqpmkKJ1O4TCTZMetAt6jGk7Q=
|
||||||
|
github.com/juju/errors v0.0.0-20190930114154-d42613fe1ab9/go.mod h1:W54LbzXuIE0boCoNJfwqpmkKJ1O4TCTZMetAt6jGk7Q=
|
||||||
|
github.com/juju/loggo v0.0.0-20180524022052-584905176618/go.mod h1:vgyd7OREkbtVEN/8IXZe5Ooef3LQePvuBm9UWj6ZL8U=
|
||||||
|
github.com/juju/testing v0.0.0-20180920084828-472a3e8b2073/go.mod h1:63prj8cnj0tU0S9OHjGJn+b1h0ZghCndfnbQolrYTwA=
|
||||||
|
github.com/julienschmidt/httprouter v1.2.0/go.mod h1:SYymIcj16QtmaHHD7aYtjjsJG7VTCxuUUipMqKk8s4w=
|
||||||
|
github.com/kisielk/errcheck v1.1.0/go.mod h1:EZBBE59ingxPouuu3KfxchcWSUPOHkagtvWXihfKN4Q=
|
||||||
|
github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=
|
||||||
|
github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
|
||||||
|
github.com/konsorten/go-windows-terminal-sequences v1.0.3 h1:CE8S1cTafDpPvMhIxNJKvHsGVBgn1xWYf1NbHQhywc8=
|
||||||
|
github.com/konsorten/go-windows-terminal-sequences v1.0.3/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
|
||||||
|
github.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc=
|
||||||
|
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||||
|
github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
|
||||||
|
github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
|
||||||
|
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||||
|
github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
|
||||||
|
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||||
|
github.com/lib/pq v1.10.5 h1:J+gdV2cUmX7ZqL2B0lFcW0m+egaHC2V3lpO8nWxyYiQ=
|
||||||
|
github.com/lib/pq v1.10.5/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
|
||||||
|
github.com/magiconair/properties v1.8.1 h1:ZC2Vc7/ZFkGmsVC9KvOjumD+G5lXy2RtTKyzRKO2BQ4=
|
||||||
|
github.com/magiconair/properties v1.8.1/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ=
|
||||||
|
github.com/mattn/go-colorable v0.0.9 h1:UVL0vNpWh04HeJXV0KLcaT7r06gOH2l4OW6ddYRUIY4=
|
||||||
|
github.com/mattn/go-colorable v0.0.9/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU=
|
||||||
|
github.com/mattn/go-isatty v0.0.3 h1:ns/ykhmWi7G9O+8a448SecJU3nSMBXJfqQkl0upE1jI=
|
||||||
|
github.com/mattn/go-isatty v0.0.3/go.mod h1:M+lRXTBqGeGNdLjl/ufCoiOlB5xdOkqRJdNxMWT7Zi4=
|
||||||
|
github.com/mattn/go-runewidth v0.0.3/go.mod h1:LwmH8dsx7+W8Uxz3IHJYH5QSwggIsqBzpuz5H//U1FU=
|
||||||
|
github.com/mattn/go-runewidth v0.0.13 h1:lTGmDsbAYt5DmK6OnoV7EuIF1wEIFAcxld6ypU4OSgU=
|
||||||
|
github.com/mattn/go-runewidth v0.0.13/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
|
||||||
|
github.com/mattn/goveralls v0.0.2/go.mod h1:8d1ZMHsd7fW6IRPKQh46F2WRpyib5/X4FOpevwGNQEw=
|
||||||
|
github.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0=
|
||||||
|
github.com/matttproud/golang_protobuf_extensions v1.0.4 h1:mmDVorXM7PCGKw94cs5zkfA9PSy5pEvNWRP0ET0TIVo=
|
||||||
|
github.com/matttproud/golang_protobuf_extensions v1.0.4/go.mod h1:BSXmuO+STAnVfrANrmjBb36TMTDstsz7MSK+HVaYKv4=
|
||||||
|
github.com/miekg/dns v1.0.14/go.mod h1:W1PPwlIAgtquWBMBEV9nkV9Cazfe8ScdGz/Lj7v3Nrg=
|
||||||
|
github.com/mitchellh/cli v1.0.0/go.mod h1:hNIlj7HEI86fIcpObd7a0FcrxTWetlwJDGcceTlRvqc=
|
||||||
|
github.com/mitchellh/go-homedir v1.0.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=
|
||||||
|
github.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=
|
||||||
|
github.com/mitchellh/go-testing-interface v1.0.0/go.mod h1:kRemZodwjscx+RGhAo8eIhFbs2+BFgRtFPeD/KE+zxI=
|
||||||
|
github.com/mitchellh/gox v0.4.0/go.mod h1:Sd9lOJ0+aimLBi73mGofS1ycjY8lL3uZM3JPS42BGNg=
|
||||||
|
github.com/mitchellh/iochan v1.0.0/go.mod h1:JwYml1nuB7xOzsp52dPpHFffvOCDupsG0QubkSMEySY=
|
||||||
|
github.com/mitchellh/mapstructure v0.0.0-20160808181253-ca63d7c062ee/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=
|
||||||
|
github.com/mitchellh/mapstructure v1.1.2/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=
|
||||||
|
github.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyuac5Z2hdY=
|
||||||
|
github.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo=
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||||
|
github.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
|
||||||
|
github.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=
|
||||||
|
github.com/oklog/ulid v1.3.1/go.mod h1:CirwcVhetQ6Lv90oh/F+FBtV6XMibvdAFo93nm5qn4U=
|
||||||
|
github.com/pascaldekloe/goe v0.0.0-20180627143212-57f6aae5913c/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc=
|
||||||
|
github.com/pelletier/go-toml v1.2.0/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic=
|
||||||
|
github.com/pelletier/go-toml v1.9.2 h1:7NiByeVF4jKSG1lDF3X8LTIkq2/bu+1uYbIm1eS5tzk=
|
||||||
|
github.com/pelletier/go-toml v1.9.2/go.mod h1:u1nR/EPcESfeI/szUZKdtJ0xRNbUoANCkoOuaOx1Y+c=
|
||||||
|
github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
|
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
|
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
||||||
|
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/posener/complete v1.1.1/go.mod h1:em0nMJCgc9GFtwrmVmEMR/ZL6WyhyjMBndrE9hABlRI=
|
||||||
|
github.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw=
|
||||||
|
github.com/prometheus/client_golang v0.9.3/go.mod h1:/TN21ttK/J9q6uSwhBd54HahCDft0ttaMvbicHlPoso=
|
||||||
|
github.com/prometheus/client_golang v1.15.1 h1:8tXpTmJbyH5lydzFPoxSIJ0J46jdh3tylbvM1xCv0LI=
|
||||||
|
github.com/prometheus/client_golang v1.15.1/go.mod h1:e9yaBhRPU2pPNsZwE+JdQl0KEt1N9XgF6zxWmaC0xOk=
|
||||||
|
github.com/prometheus/client_model v0.0.0-20180712105110-5c3871d89910/go.mod h1:MbSGuTsp3dbXC40dX6PRTWyKYBIrTGTE9sqQNg2J8bo=
|
||||||
|
github.com/prometheus/client_model v0.0.0-20190129233127-fd36f4220a90/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=
|
||||||
|
github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=
|
||||||
|
github.com/prometheus/client_model v0.3.0 h1:UBgGFHqYdG/TPFD1B1ogZywDqEkwp3fBMvqdiQ7Xew4=
|
||||||
|
github.com/prometheus/client_model v0.3.0/go.mod h1:LDGWKZIo7rky3hgvBe+caln+Dr3dPggB5dvjtD7w9+w=
|
||||||
|
github.com/prometheus/common v0.0.0-20181113130724-41aa239b4cce/go.mod h1:daVV7qP5qjZbuso7PdcryaAu0sAZbrN9i7WWcTMWvro=
|
||||||
|
github.com/prometheus/common v0.4.0/go.mod h1:TNfzLD0ON7rHzMJeJkieUDPYmFC7Snx/y86RQel1bk4=
|
||||||
|
github.com/prometheus/common v0.42.0 h1:EKsfXEYo4JpWMHH5cg+KOUWeuJSov1Id8zGR8eeI1YM=
|
||||||
|
github.com/prometheus/common v0.42.0/go.mod h1:xBwqVerjNdUDjgODMpudtOMwlOwf2SaTr1yjz4b7Zbc=
|
||||||
|
github.com/prometheus/procfs v0.0.0-20181005140218-185b4288413d/go.mod h1:c3At6R/oaqEKCNdg8wHV1ftS6bRYblBhIjjI8uT2IGk=
|
||||||
|
github.com/prometheus/procfs v0.0.0-20190507164030-5867b95ac084/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA=
|
||||||
|
github.com/prometheus/procfs v0.9.0 h1:wzCHvIvM5SxWqYvwgVL7yJY8Lz3PKn49KQtpgMYJfhI=
|
||||||
|
github.com/prometheus/procfs v0.9.0/go.mod h1:+pB4zwohETzFnmlpe6yd2lSc+0/46IYZRB/chUwxUZY=
|
||||||
|
github.com/prometheus/tsdb v0.7.1/go.mod h1:qhTCs0VvXwvX/y3TZrWD7rabWM+ijKTux40TwIPHuXU=
|
||||||
|
github.com/rivo/uniseg v0.2.0 h1:S1pD9weZBuJdFmowNwbpi7BJ8TNftyUImj/0WQi72jY=
|
||||||
|
github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
|
||||||
|
github.com/rogpeppe/fastuuid v0.0.0-20150106093220-6724a57986af/go.mod h1:XWv6SoW27p1b0cqNHllgS5HIMJraePCO15w5zCzIWYg=
|
||||||
|
github.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=
|
||||||
|
github.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ=
|
||||||
|
github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
|
||||||
|
github.com/ryanuber/columnize v0.0.0-20160712163229-9b3edd62028f/go.mod h1:sm1tb6uqfes/u+d4ooFouqFdy9/2g9QGwK3SQygK0Ts=
|
||||||
|
github.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529/go.mod h1:DxrIzT+xaE7yg65j358z/aeFdxmN0P9QXhEzd20vsDc=
|
||||||
|
github.com/shiena/ansicolor v0.0.0-20200904210342-c7312218db18 h1:DAYUYH5869yV94zvCES9F51oYtN5oGlwjxJJz7ZCnik=
|
||||||
|
github.com/shiena/ansicolor v0.0.0-20200904210342-c7312218db18/go.mod h1:nkxAfR/5quYxwPZhyDxgasBMnRtBZd0FCEpawpjMUFg=
|
||||||
|
github.com/shopspring/decimal v1.3.1 h1:2Usl1nmF/WZucqkFZhnfFYxxxu8LG21F6nPQBE5gKV8=
|
||||||
|
github.com/shopspring/decimal v1.3.1/go.mod h1:DKyhrW/HYNuLGql+MJL6WCR6knT2jwCFRcu2hWCYk4o=
|
||||||
|
github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=
|
||||||
|
github.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=
|
||||||
|
github.com/sirupsen/logrus v1.6.0 h1:UBcNElsrwanuuMsnGSlYmtmgbb23qDR5dG+6X6Oo89I=
|
||||||
|
github.com/sirupsen/logrus v1.6.0/go.mod h1:7uNnSEd1DgxDLC74fIahvMZmmYsHGZGEOFrfsX/uA88=
|
||||||
|
github.com/smartwalle/pongo2render v1.0.1 h1:rsPnDTu/+zIT5HEB5RbMjxKY5hisov26j0isZL/7YS0=
|
||||||
|
github.com/smartwalle/pongo2render v1.0.1/go.mod h1:MGnTzND7nEMz7g194kjlnw8lx/V5JJlb1hr5kDXEO0I=
|
||||||
|
github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d h1:zE9ykElWQ6/NYmHa3jpm/yHnI4xSofP+UP6SpjHcSeM=
|
||||||
|
github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc=
|
||||||
|
github.com/smartystreets/goconvey v1.6.4 h1:fv0U8FUIMPNf1L9lnHLvLhgicrIVChEkdzIKYqbNC9s=
|
||||||
|
github.com/smartystreets/goconvey v1.6.4/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA=
|
||||||
|
github.com/soheilhy/cmux v0.1.4/go.mod h1:IM3LyeVVIOuxMH7sFAkER9+bJ4dT7Ms6E4xg4kGIyLM=
|
||||||
|
github.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=
|
||||||
|
github.com/spf13/afero v1.1.2 h1:m8/z1t7/fwjysjQRYbP0RD+bUIF/8tJwPdEZsI83ACI=
|
||||||
|
github.com/spf13/afero v1.1.2/go.mod h1:j4pytiNVoe2o6bmDsKpLACNPDBIoEAkihy7loJ1B0CQ=
|
||||||
|
github.com/spf13/cast v1.3.0 h1:oget//CVOEoFewqQxwr0Ej5yjygnqGkvggSE/gB35Q8=
|
||||||
|
github.com/spf13/cast v1.3.0/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE=
|
||||||
|
github.com/spf13/cobra v1.1.3/go.mod h1:pGADOWyqRD/YMrPZigI/zbliZ2wVD/23d+is3pSWzOo=
|
||||||
|
github.com/spf13/jwalterweatherman v1.0.0 h1:XHEdyB+EcvlqZamSM4ZOMGlc93t6AcsBEu9Gc1vn7yk=
|
||||||
|
github.com/spf13/jwalterweatherman v1.0.0/go.mod h1:cQK4TGJAtQXfYWX+Ddv3mKDzgVb68N+wFjFa4jdeBTo=
|
||||||
|
github.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4=
|
||||||
|
github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
|
||||||
|
github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
|
||||||
|
github.com/spf13/viper v1.7.0 h1:xVKxvI7ouOI5I+U9s2eeiUfMaWBVoXA3AWskkrqK0VM=
|
||||||
|
github.com/spf13/viper v1.7.0/go.mod h1:8WkrPz2fc9jxqZNCJI/76HCieCp4Q8HaLFoCha5qpdg=
|
||||||
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
||||||
|
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||||
|
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk=
|
||||||
|
github.com/subosito/gotenv v1.2.0 h1:Slr1R9HxAlEKefgq5jn9U+DnETlIUa6HfgEzj0g5d7s=
|
||||||
|
github.com/subosito/gotenv v1.2.0/go.mod h1:N0PQaV/YGNqwC0u51sEeR/aUtSLEXKX9iv69rRypqCw=
|
||||||
|
github.com/tmc/grpc-websocket-proxy v0.0.0-20190109142713-0ad062ec5ee5/go.mod h1:ncp9v5uamzpCO7NfCPTXjqaC+bZgJeR0sMTm6dMHP7U=
|
||||||
|
github.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2/go.mod h1:UETIi67q53MR2AWcXfiuqkDkRtnGDLqkBTpCHuJHxtU=
|
||||||
|
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
|
||||||
|
go.etcd.io/bbolt v1.3.2/go.mod h1:IbVyRI1SCnLcuJnV2u8VeU0CEYM7e686BmAb1XKL+uU=
|
||||||
|
go.opencensus.io v0.21.0/go.mod h1:mSImk1erAIZhrmZN+AvHh14ztQfjbGwt4TtuofqLduU=
|
||||||
|
go.opencensus.io v0.22.0/go.mod h1:+kGneAE2xo2IficOXnaByMWTGM9T73dGwxeWcUqIpI8=
|
||||||
|
go.starlark.net v0.0.0-20220816155156-cfacd8902214 h1:MqijAN3S61c7KWasOk+zIqIjHQPN6WUra/X3+YAkQxQ=
|
||||||
|
go.starlark.net v0.0.0-20220816155156-cfacd8902214/go.mod h1:VZcBMdr3cT3PnBoWunTabuSEXwVAH+ZJ5zxfs3AdASk=
|
||||||
|
go.uber.org/atomic v1.4.0/go.mod h1:gD2HeocX3+yG+ygLZcrzQJaqmWj9AIm7n08wl/qW/PE=
|
||||||
|
go.uber.org/multierr v1.1.0/go.mod h1:wR5kodmAFQ0UK8QlbwjlSNy0Z68gJhDJUG5sjR94q/0=
|
||||||
|
go.uber.org/zap v1.10.0/go.mod h1:vwi/ZaCAaUcBkycHslxD9B2zi4UTXhF60s6SWpuDF0Q=
|
||||||
|
golang.org/x/arch v0.0.0-20190927153633-4e8777c89be4 h1:QlVATYS7JBoZMVaf+cNjb90WD/beKVHnIxFKT4QaHVI=
|
||||||
|
golang.org/x/arch v0.0.0-20190927153633-4e8777c89be4/go.mod h1:flIaEI6LNU6xOCD5PaJvn9wGP0agmIOqjrtsKGRguv4=
|
||||||
|
golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
|
||||||
|
golang.org/x/crypto v0.0.0-20181029021203-45a5f77698d3/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
|
||||||
|
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||||
|
golang.org/x/crypto v0.0.0-20190510104115-cbcb75029529/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
|
||||||
|
golang.org/x/crypto v0.0.0-20190605123033-f99c8df09eb5/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
|
||||||
|
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
||||||
|
golang.org/x/crypto v0.0.0-20220315160706-3147a52a75dd h1:XcWmESyNjXJMLahc3mqVQJcgSTDxFxhETVlfk9uGc38=
|
||||||
|
golang.org/x/crypto v0.0.0-20220315160706-3147a52a75dd/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=
|
||||||
|
golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
|
||||||
|
golang.org/x/exp v0.0.0-20190306152737-a1d7652674e8/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
|
||||||
|
golang.org/x/exp v0.0.0-20190510132918-efd6b22b2522/go.mod h1:ZjyILWgesfNpC6sMxTJOJm9Kp84zZh5NQWvqDGG3Qr8=
|
||||||
|
golang.org/x/exp v0.0.0-20190829153037-c13cbed26979/go.mod h1:86+5VVa7VpoJ4kLfm080zCjGlMRFzhUhsZKEZO7MGek=
|
||||||
|
golang.org/x/exp v0.0.0-20191030013958-a1ab85dbe136/go.mod h1:JXzH8nQsPlswgeRAPE3MuO9GYsAcnJvJ4vnMwN/5qkY=
|
||||||
|
golang.org/x/image v0.0.0-20190227222117-0694c2d4d067/go.mod h1:kZ7UVZpmo3dzQBMxlp+ypCbDeSB+sBbTgSJuh5dn5js=
|
||||||
|
golang.org/x/image v0.0.0-20190802002840-cff245a6509b/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0=
|
||||||
|
golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=
|
||||||
|
golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=
|
||||||
|
golang.org/x/lint v0.0.0-20190301231843-5614ed5bae6f/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=
|
||||||
|
golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=
|
||||||
|
golang.org/x/lint v0.0.0-20190409202823-959b441ac422/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=
|
||||||
|
golang.org/x/lint v0.0.0-20190909230951-414d861bb4ac/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=
|
||||||
|
golang.org/x/lint v0.0.0-20190930215403-16217165b5de/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=
|
||||||
|
golang.org/x/mobile v0.0.0-20190312151609-d3739f865fa6/go.mod h1:z+o9i4GpDbdi3rU15maQ/Ox0txvL9dWGYEHz965HBQE=
|
||||||
|
golang.org/x/mobile v0.0.0-20190719004257-d2bd2a29d028/go.mod h1:E/iHnbuqvinMTCcRqshq8CkpyQDoeVncDDYHnLhea+o=
|
||||||
|
golang.org/x/mod v0.0.0-20190513183733-4bf6d317e70e/go.mod h1:mXi4GBBbnImb6dmsKGUJ2LatrhH/nqhxcFungHvyanc=
|
||||||
|
golang.org/x/mod v0.1.0/go.mod h1:0QHyrYULN0/3qlju5TqG8bIK38QM8yzMo5ekMj3DlcY=
|
||||||
|
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4 h1:6zppjxzCulZykYSLyVDYbneBfbaBIQPYMevg0bEwv2s=
|
||||||
|
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
||||||
|
golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
|
golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
|
golang.org/x/net v0.0.0-20181023162649-9b4f9f5ad519/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
|
golang.org/x/net v0.0.0-20181114220301-adae6a3d119a/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
|
golang.org/x/net v0.0.0-20181201002055-351d144fa1fc/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
|
golang.org/x/net v0.0.0-20181220203305-927f97764cc3/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
|
golang.org/x/net v0.0.0-20190108225652-1e06a53dbb7e/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
|
golang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
|
golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
|
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
|
golang.org/x/net v0.0.0-20190501004415-9ce7a6920f09/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
|
golang.org/x/net v0.0.0-20190503192946-f4e77d36d62c/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
|
golang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks=
|
||||||
|
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
|
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
||||||
|
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
|
||||||
|
golang.org/x/net v0.7.0 h1:rJrUqqhjsgNp7KqAIc25s9pZnjU7TUcSY7HcVZjdn1g=
|
||||||
|
golang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
|
||||||
|
golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
|
||||||
|
golang.org/x/oauth2 v0.0.0-20190226205417-e64efc72b421/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=
|
||||||
|
golang.org/x/oauth2 v0.0.0-20190604053449-0f29369cfe45/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=
|
||||||
|
golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20190227155943-e225da77a7e6/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sys v0.0.0-20180823144017-11551d06cbcc/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20181026203630-95b1ffbd15a5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20181107165924-66b7b1311ac8/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20181116152217-5ac8a444bdc5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20190312061237-fead79001313/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20190502145724-3ef323f4f1fd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20190507160741-ecd444e8653b/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20190606165138-5da285871e9c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20190624142023-c5567b49c5d0/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20191005200804-aed5e4c7ecf9/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.0.0-20210906170528-6f6e22806c34/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.0.0-20211117180635-dee7805ff2e1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.0.0-20220908164124-27713097b956/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.6.0 h1:MVltZSvRTcU2ljQOhs94SXPftV6DCNnZViHeQps87pQ=
|
||||||
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||||
|
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
||||||
|
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
|
golang.org/x/text v0.3.1-0.20180807135948-17ff2d5776d2/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
|
golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
|
||||||
|
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
|
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
||||||
|
golang.org/x/text v0.7.0 h1:4BRB4x83lYWy72KwLD/qYDuTu7q9PjSagHvijDw7cLo=
|
||||||
|
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
||||||
|
golang.org/x/time v0.0.0-20181108054448-85acf8d2951c/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
|
||||||
|
golang.org/x/time v0.0.0-20190308202827-9d24e82272b4/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
|
||||||
|
golang.org/x/tools v0.0.0-20180221164845-07fd8470d635/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/tools v0.0.0-20181221001348-537d06c36207/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=
|
||||||
|
golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
|
||||||
|
golang.org/x/tools v0.0.0-20190312151545-0bb0c0a6e846/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
|
||||||
|
golang.org/x/tools v0.0.0-20190312170243-e65039ee4138/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
|
||||||
|
golang.org/x/tools v0.0.0-20190328211700-ab21143f2384/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
|
||||||
|
golang.org/x/tools v0.0.0-20190425150028-36563e24a262/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
|
||||||
|
golang.org/x/tools v0.0.0-20190506145303-2d16b83fe98c/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
|
||||||
|
golang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
|
||||||
|
golang.org/x/tools v0.0.0-20190606124116-d0a3d012864b/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=
|
||||||
|
golang.org/x/tools v0.0.0-20190621195816-6e04913cbbac/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=
|
||||||
|
golang.org/x/tools v0.0.0-20190628153133-6cdbf07be9d0/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=
|
||||||
|
golang.org/x/tools v0.0.0-20190816200558-6889da9d5479/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
|
golang.org/x/tools v0.0.0-20190911174233-4f2ddba30aff/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
|
golang.org/x/tools v0.0.0-20191012152004-8de300cfc20a/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
|
golang.org/x/tools v0.0.0-20191112195655-aa38f8e97acc/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
|
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
|
golang.org/x/tools v0.1.12 h1:VveCTK38A2rkS8ZqFY25HIDFscX5X9OoEhJd3quQmXU=
|
||||||
|
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
||||||
|
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
google.golang.org/api v0.4.0/go.mod h1:8k5glujaEP+g9n7WNsDg8QP6cUVNI86fCNMcbazEtwE=
|
||||||
|
google.golang.org/api v0.7.0/go.mod h1:WtwebWUNSVBH/HAw79HIFXZNqEvBhG+Ra+ax0hx3E3M=
|
||||||
|
google.golang.org/api v0.8.0/go.mod h1:o4eAsZoiT+ibD93RtjEohWalFOjRDx6CVaqeizhEnKg=
|
||||||
|
google.golang.org/api v0.9.0/go.mod h1:o4eAsZoiT+ibD93RtjEohWalFOjRDx6CVaqeizhEnKg=
|
||||||
|
google.golang.org/api v0.13.0/go.mod h1:iLdEw5Ide6rF15KTC1Kkl0iskquN2gFfn9o9XIsbkAI=
|
||||||
|
google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=
|
||||||
|
google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=
|
||||||
|
google.golang.org/appengine v1.5.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=
|
||||||
|
google.golang.org/appengine v1.6.1/go.mod h1:i06prIuMbXzDqacNJfV5OdTW448YApPu5ww/cMBSeb0=
|
||||||
|
google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=
|
||||||
|
google.golang.org/genproto v0.0.0-20190307195333-5fe7a883aa19/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=
|
||||||
|
google.golang.org/genproto v0.0.0-20190418145605-e7d98fc518a7/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=
|
||||||
|
google.golang.org/genproto v0.0.0-20190425155659-357c62f0e4bb/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=
|
||||||
|
google.golang.org/genproto v0.0.0-20190502173448-54afdca5d873/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=
|
||||||
|
google.golang.org/genproto v0.0.0-20190801165951-fa694d86fc64/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc=
|
||||||
|
google.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc=
|
||||||
|
google.golang.org/genproto v0.0.0-20190911173649-1774047e7e51/go.mod h1:IbNlFCBrqXvoKpeg0TB2l7cyZUmoaFKYIwrEpbDKLA8=
|
||||||
|
google.golang.org/genproto v0.0.0-20191108220845-16a3f7862a1a/go.mod h1:n3cpQtvxv34hfy77yVDNjmbRyujviMdxYliBSkLhpCc=
|
||||||
|
google.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo=
|
||||||
|
google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c=
|
||||||
|
google.golang.org/grpc v1.20.1/go.mod h1:10oTOabMzJvdu6/UiuZezV6QK5dSlG84ov/aaiqXj38=
|
||||||
|
google.golang.org/grpc v1.21.1/go.mod h1:oYelfM1adQP15Ek0mdvEgi9Df8B9CZIaU1084ijfRaM=
|
||||||
|
google.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg=
|
||||||
|
google.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk=
|
||||||
|
google.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8=
|
||||||
|
google.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0=
|
||||||
|
google.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM=
|
||||||
|
google.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE=
|
||||||
|
google.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo=
|
||||||
|
google.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=
|
||||||
|
google.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=
|
||||||
|
google.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c=
|
||||||
|
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
|
||||||
|
google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
|
||||||
|
google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng=
|
||||||
|
google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
|
||||||
|
gopkg.in/alecthomas/kingpin.v2 v2.2.6/go.mod h1:FMv+mEhP44yOT+4EoQTLFTRgOQ1FBLkstjWtayDeSgw=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
|
||||||
|
gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=
|
||||||
|
gopkg.in/ini.v1 v1.51.0 h1:AQvPpx3LzTDM0AjnIRlVFwFFGC+npRopjZxLJj6gdno=
|
||||||
|
gopkg.in/ini.v1 v1.51.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=
|
||||||
|
gopkg.in/mgo.v2 v2.0.0-20180705113604-9856a29383ce/go.mod h1:yeKp02qBN3iKW1OzL3MGk2IdtZzaj7SFntXj72NppTA=
|
||||||
|
gopkg.in/resty.v1 v1.12.0/go.mod h1:mDo4pnntr5jdWRML875a/NmxYqAlA73dVijT2AXvQQo=
|
||||||
|
gopkg.in/yaml.v2 v2.0.0-20170812160011-eb3733d160e7/go.mod h1:JAlM8MvJe8wmxCU4Bli9HhUf9+ttbYbLASfIpnQbh74=
|
||||||
|
gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||||
|
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||||
|
gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||||
|
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
|
||||||
|
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
|
||||||
|
honnef.co/go/tools v0.0.0-20190106161140-3f1c8253044a/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
|
||||||
|
honnef.co/go/tools v0.0.0-20190418001031-e561f6794a2a/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
|
||||||
|
honnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
|
||||||
|
honnef.co/go/tools v0.0.1-2019.2.3/go.mod h1:a3bituU0lyd329TUQxRnasdCoJDkEUEAqEt0JzvZhAg=
|
||||||
|
rsc.io/binaryregexp v0.2.0/go.mod h1:qTv7/COck+e2FymRvadv62gMdZztPaShugOCi3I+8D8=
|
||||||
|
rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=
|
3
gosimple.ignore
Normal file
3
gosimple.ignore
Normal file
|
@ -0,0 +1,3 @@
|
||||||
|
github.com/beego/bee/v2/cmd/commands/run/*.go:S1024
|
||||||
|
github.com/beego/bee/v2/cmd/commands/dlv/*.go:S1024
|
||||||
|
github.com/beego/bee/v2/utils/*.go:S1026
|
365
hproseapp.go
365
hproseapp.go
|
@ -1,365 +0,0 @@
|
||||||
// Copyright 2013 bee authors
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
path "path/filepath"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
var cmdHproseapp = &Command{
|
|
||||||
// CustomFlags: true,
|
|
||||||
UsageLine: "hprose [appname]",
|
|
||||||
Short: "Creates an RPC application based on Hprose and Beego frameworks",
|
|
||||||
Long: `
|
|
||||||
The command 'hprose' creates an RPC application based on both Beego and Hprose (http://hprose.com/).
|
|
||||||
|
|
||||||
{{"To scaffold out your application, use:"|bold}}
|
|
||||||
|
|
||||||
$ bee hprose [appname] [-tables=""] [-driver=mysql] [-conn=root:@tcp(127.0.0.1:3306)/test]
|
|
||||||
|
|
||||||
If 'conn' is empty, the command will generate a sample application. Otherwise the command
|
|
||||||
will connect to your database and generate models based on the existing tables.
|
|
||||||
|
|
||||||
The command 'hprose' creates a folder named [appname] with the following structure:
|
|
||||||
|
|
||||||
├── main.go
|
|
||||||
├── {{"conf"|foldername}}
|
|
||||||
│ └── app.conf
|
|
||||||
└── {{"models"|foldername}}
|
|
||||||
└── object.go
|
|
||||||
└── user.go
|
|
||||||
`,
|
|
||||||
PreRun: func(cmd *Command, args []string) { ShowShortVersionBanner() },
|
|
||||||
Run: createhprose,
|
|
||||||
}
|
|
||||||
|
|
||||||
var hproseconf = `appname = {{.Appname}}
|
|
||||||
httpport = 8080
|
|
||||||
runmode = dev
|
|
||||||
autorender = false
|
|
||||||
copyrequestbody = true
|
|
||||||
EnableDocs = true
|
|
||||||
`
|
|
||||||
var hproseMaingo = `package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"reflect"
|
|
||||||
|
|
||||||
"{{.Appname}}/models"
|
|
||||||
"github.com/hprose/hprose-golang/rpc"
|
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
|
||||||
)
|
|
||||||
|
|
||||||
func logInvokeHandler(
|
|
||||||
name string,
|
|
||||||
args []reflect.Value,
|
|
||||||
context rpc.Context,
|
|
||||||
next rpc.NextInvokeHandler) (results []reflect.Value, err error) {
|
|
||||||
fmt.Printf("%s(%v) = ", name, args)
|
|
||||||
results, err = next(name, args, context)
|
|
||||||
fmt.Printf("%v %v\r\n", results, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
// Create WebSocketServer
|
|
||||||
// service := rpc.NewWebSocketService()
|
|
||||||
|
|
||||||
// Create Http Server
|
|
||||||
service := rpc.NewHTTPService()
|
|
||||||
|
|
||||||
// Use Logger Middleware
|
|
||||||
service.AddInvokeHandler(logInvokeHandler)
|
|
||||||
|
|
||||||
// Publish Functions
|
|
||||||
service.AddFunction("AddOne", models.AddOne)
|
|
||||||
service.AddFunction("GetOne", models.GetOne)
|
|
||||||
|
|
||||||
// Start Service
|
|
||||||
beego.Handler("/", service)
|
|
||||||
beego.Run()
|
|
||||||
}
|
|
||||||
`
|
|
||||||
|
|
||||||
var hproseMainconngo = `package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"reflect"
|
|
||||||
|
|
||||||
"{{.Appname}}/models"
|
|
||||||
"github.com/hprose/hprose-golang/rpc"
|
|
||||||
|
|
||||||
"github.com/astaxie/beego"
|
|
||||||
"github.com/astaxie/beego/orm"
|
|
||||||
{{.DriverPkg}}
|
|
||||||
)
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
orm.RegisterDataBase("default", "{{.DriverName}}", "{{.conn}}")
|
|
||||||
}
|
|
||||||
|
|
||||||
func logInvokeHandler(
|
|
||||||
name string,
|
|
||||||
args []reflect.Value,
|
|
||||||
context rpc.Context,
|
|
||||||
next rpc.NextInvokeHandler) (results []reflect.Value, err error) {
|
|
||||||
fmt.Printf("%s(%v) = ", name, args)
|
|
||||||
results, err = next(name, args, context)
|
|
||||||
fmt.Printf("%v %v\r\n", results, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
// Create WebSocketServer
|
|
||||||
// service := rpc.NewWebSocketService()
|
|
||||||
|
|
||||||
// Create Http Server
|
|
||||||
service := rpc.NewHTTPService()
|
|
||||||
|
|
||||||
// Use Logger Middleware
|
|
||||||
service.AddInvokeHandler(logInvokeHandler)
|
|
||||||
|
|
||||||
{{HproseFunctionList}}
|
|
||||||
|
|
||||||
// Start Service
|
|
||||||
beego.Handler("/", service)
|
|
||||||
beego.Run()
|
|
||||||
}
|
|
||||||
|
|
||||||
`
|
|
||||||
|
|
||||||
var hproseModels = `package models
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"strconv"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
Objects map[string]*Object
|
|
||||||
)
|
|
||||||
|
|
||||||
type Object struct {
|
|
||||||
ObjectId string
|
|
||||||
Score int64
|
|
||||||
PlayerName string
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
Objects = make(map[string]*Object)
|
|
||||||
Objects["hjkhsbnmn123"] = &Object{"hjkhsbnmn123", 100, "astaxie"}
|
|
||||||
Objects["mjjkxsxsaa23"] = &Object{"mjjkxsxsaa23", 101, "someone"}
|
|
||||||
}
|
|
||||||
|
|
||||||
func AddOne(object Object) (ObjectId string) {
|
|
||||||
object.ObjectId = "astaxie" + strconv.FormatInt(time.Now().UnixNano(), 10)
|
|
||||||
Objects[object.ObjectId] = &object
|
|
||||||
return object.ObjectId
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetOne(ObjectId string) (object *Object, err error) {
|
|
||||||
if v, ok := Objects[ObjectId]; ok {
|
|
||||||
return v, nil
|
|
||||||
}
|
|
||||||
return nil, errors.New("ObjectId Not Exist")
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetAll() map[string]*Object {
|
|
||||||
return Objects
|
|
||||||
}
|
|
||||||
|
|
||||||
func Update(ObjectId string, Score int64) (err error) {
|
|
||||||
if v, ok := Objects[ObjectId]; ok {
|
|
||||||
v.Score = Score
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return errors.New("ObjectId Not Exist")
|
|
||||||
}
|
|
||||||
|
|
||||||
func Delete(ObjectId string) {
|
|
||||||
delete(Objects, ObjectId)
|
|
||||||
}
|
|
||||||
|
|
||||||
`
|
|
||||||
|
|
||||||
var hproseModels2 = `package models
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"strconv"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
UserList map[string]*User
|
|
||||||
)
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
UserList = make(map[string]*User)
|
|
||||||
u := User{"user_11111", "astaxie", "11111", Profile{"male", 20, "Singapore", "astaxie@gmail.com"}}
|
|
||||||
UserList["user_11111"] = &u
|
|
||||||
}
|
|
||||||
|
|
||||||
type User struct {
|
|
||||||
Id string
|
|
||||||
Username string
|
|
||||||
Password string
|
|
||||||
Profile Profile
|
|
||||||
}
|
|
||||||
|
|
||||||
type Profile struct {
|
|
||||||
Gender string
|
|
||||||
Age int
|
|
||||||
Address string
|
|
||||||
Email string
|
|
||||||
}
|
|
||||||
|
|
||||||
func AddUser(u User) string {
|
|
||||||
u.Id = "user_" + strconv.FormatInt(time.Now().UnixNano(), 10)
|
|
||||||
UserList[u.Id] = &u
|
|
||||||
return u.Id
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetUser(uid string) (u *User, err error) {
|
|
||||||
if u, ok := UserList[uid]; ok {
|
|
||||||
return u, nil
|
|
||||||
}
|
|
||||||
return nil, errors.New("User not exists")
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetAllUsers() map[string]*User {
|
|
||||||
return UserList
|
|
||||||
}
|
|
||||||
|
|
||||||
func UpdateUser(uid string, uu *User) (a *User, err error) {
|
|
||||||
if u, ok := UserList[uid]; ok {
|
|
||||||
if uu.Username != "" {
|
|
||||||
u.Username = uu.Username
|
|
||||||
}
|
|
||||||
if uu.Password != "" {
|
|
||||||
u.Password = uu.Password
|
|
||||||
}
|
|
||||||
if uu.Profile.Age != 0 {
|
|
||||||
u.Profile.Age = uu.Profile.Age
|
|
||||||
}
|
|
||||||
if uu.Profile.Address != "" {
|
|
||||||
u.Profile.Address = uu.Profile.Address
|
|
||||||
}
|
|
||||||
if uu.Profile.Gender != "" {
|
|
||||||
u.Profile.Gender = uu.Profile.Gender
|
|
||||||
}
|
|
||||||
if uu.Profile.Email != "" {
|
|
||||||
u.Profile.Email = uu.Profile.Email
|
|
||||||
}
|
|
||||||
return u, nil
|
|
||||||
}
|
|
||||||
return nil, errors.New("User Not Exist")
|
|
||||||
}
|
|
||||||
|
|
||||||
func Login(username, password string) bool {
|
|
||||||
for _, u := range UserList {
|
|
||||||
if u.Username == username && u.Password == password {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func DeleteUser(uid string) {
|
|
||||||
delete(UserList, uid)
|
|
||||||
}
|
|
||||||
`
|
|
||||||
|
|
||||||
var hproseAddFunctions = []string{}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
cmdHproseapp.Flag.Var(&tables, "tables", "List of table names separated by a comma.")
|
|
||||||
cmdHproseapp.Flag.Var(&driver, "driver", "Database driver. Either mysql, postgres or sqlite.")
|
|
||||||
cmdHproseapp.Flag.Var(&conn, "conn", "Connection string used by the driver to connect to a database instance.")
|
|
||||||
}
|
|
||||||
|
|
||||||
func createhprose(cmd *Command, args []string) int {
|
|
||||||
output := cmd.Out()
|
|
||||||
|
|
||||||
curpath, _ := os.Getwd()
|
|
||||||
if len(args) > 1 {
|
|
||||||
cmd.Flag.Parse(args[1:])
|
|
||||||
}
|
|
||||||
apppath, packpath, err := checkEnv(args[0])
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("%s", err)
|
|
||||||
}
|
|
||||||
if driver == "" {
|
|
||||||
driver = "mysql"
|
|
||||||
}
|
|
||||||
if conn == "" {
|
|
||||||
}
|
|
||||||
|
|
||||||
logger.Info("Creating Hprose application...")
|
|
||||||
|
|
||||||
os.MkdirAll(apppath, 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", apppath, "\x1b[0m")
|
|
||||||
os.Mkdir(path.Join(apppath, "conf"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "conf"), "\x1b[0m")
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "conf", "app.conf"), "\x1b[0m")
|
|
||||||
WriteToFile(path.Join(apppath, "conf", "app.conf"),
|
|
||||||
strings.Replace(hproseconf, "{{.Appname}}", args[0], -1))
|
|
||||||
|
|
||||||
if conn != "" {
|
|
||||||
logger.Infof("Using '%s' as 'driver'", driver)
|
|
||||||
logger.Infof("Using '%s' as 'conn'", conn)
|
|
||||||
logger.Infof("Using '%s' as 'tables'", tables)
|
|
||||||
generateHproseAppcode(string(driver), string(conn), "1", string(tables), path.Join(curpath, args[0]))
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "main.go"), "\x1b[0m")
|
|
||||||
maingoContent := strings.Replace(hproseMainconngo, "{{.Appname}}", packpath, -1)
|
|
||||||
maingoContent = strings.Replace(maingoContent, "{{.DriverName}}", string(driver), -1)
|
|
||||||
maingoContent = strings.Replace(maingoContent, "{{HproseFunctionList}}", strings.Join(hproseAddFunctions, ""), -1)
|
|
||||||
if driver == "mysql" {
|
|
||||||
maingoContent = strings.Replace(maingoContent, "{{.DriverPkg}}", `_ "github.com/go-sql-driver/mysql"`, -1)
|
|
||||||
} else if driver == "postgres" {
|
|
||||||
maingoContent = strings.Replace(maingoContent, "{{.DriverPkg}}", `_ "github.com/lib/pq"`, -1)
|
|
||||||
}
|
|
||||||
WriteToFile(path.Join(apppath, "main.go"),
|
|
||||||
strings.Replace(
|
|
||||||
maingoContent,
|
|
||||||
"{{.conn}}",
|
|
||||||
conn.String(),
|
|
||||||
-1,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
} else {
|
|
||||||
os.Mkdir(path.Join(apppath, "models"), 0755)
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models"), "\x1b[0m")
|
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models", "object.go"), "\x1b[0m")
|
|
||||||
WriteToFile(path.Join(apppath, "models", "object.go"), apiModels)
|
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models", "user.go"), "\x1b[0m")
|
|
||||||
WriteToFile(path.Join(apppath, "models", "user.go"), apiModels2)
|
|
||||||
|
|
||||||
fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "main.go"), "\x1b[0m")
|
|
||||||
WriteToFile(path.Join(apppath, "main.go"),
|
|
||||||
strings.Replace(hproseMaingo, "{{.Appname}}", packpath, -1))
|
|
||||||
}
|
|
||||||
logger.Success("New Hprose application successfully created!")
|
|
||||||
return 0
|
|
||||||
}
|
|
23
internal/app/module/beegopro/config.go
Normal file
23
internal/app/module/beegopro/config.go
Normal file
|
@ -0,0 +1,23 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io/ioutil"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/internal/pkg/utils"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
var CompareExcept = []string{"@BeeGenerateTime"}
|
||||||
|
|
||||||
|
func (c *Container) GenConfig() {
|
||||||
|
if utils.IsExist(c.BeegoProFile) {
|
||||||
|
beeLogger.Log.Fatalf("beego pro toml exist")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
err := ioutil.WriteFile("beegopro.toml", []byte(BeegoToml), 0644)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("write beego pro toml err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
18
internal/app/module/beegopro/constx.go
Normal file
18
internal/app/module/beegopro/constx.go
Normal file
|
@ -0,0 +1,18 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
const BeegoToml = `
|
||||||
|
dsn = "root:123456@tcp(127.0.0.1:3306)/beego"
|
||||||
|
driver = "mysql"
|
||||||
|
proType = "default"
|
||||||
|
enableModule = []
|
||||||
|
apiPrefix = "/"
|
||||||
|
gitRemotePath = "https://github.com/beego-dev/beego-pro.git"
|
||||||
|
format = true
|
||||||
|
sourceGen = "text"
|
||||||
|
gitPull = true
|
||||||
|
[models.user]
|
||||||
|
name = ["uid"]
|
||||||
|
orm = ["auto"]
|
||||||
|
comment = ["Uid"]
|
||||||
|
|
||||||
|
`
|
254
internal/app/module/beegopro/container.go
Normal file
254
internal/app/module/beegopro/container.go
Normal file
|
@ -0,0 +1,254 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/internal/pkg/git"
|
||||||
|
"github.com/beego/bee/v2/internal/pkg/system"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
"github.com/davecgh/go-spew/spew"
|
||||||
|
"github.com/pelletier/go-toml"
|
||||||
|
"github.com/spf13/viper"
|
||||||
|
)
|
||||||
|
|
||||||
|
var GitRemotePath utils.DocValue
|
||||||
|
|
||||||
|
const MDateFormat = "20060102_150405"
|
||||||
|
|
||||||
|
var DefaultBeegoPro = &Container{
|
||||||
|
BeegoProFile: system.CurrentDir + "/beegopro.toml",
|
||||||
|
TimestampFile: system.CurrentDir + "/.beegopro.timestamp",
|
||||||
|
GoModFile: system.CurrentDir + "/go.mod",
|
||||||
|
UserOption: UserOption{
|
||||||
|
Debug: false,
|
||||||
|
ContextDebug: false,
|
||||||
|
Dsn: "",
|
||||||
|
Driver: "mysql",
|
||||||
|
ProType: "default",
|
||||||
|
ApiPrefix: "/api",
|
||||||
|
EnableModule: nil,
|
||||||
|
Models: make(map[string]TextModel),
|
||||||
|
GitRemotePath: "",
|
||||||
|
//GitRemotePath: "https://github.com/beego/beego-pro.git",
|
||||||
|
Branch: "master",
|
||||||
|
//GitLocalPath: system.BeegoHome + "/beego-pro",
|
||||||
|
EnableFormat: true,
|
||||||
|
SourceGen: "text",
|
||||||
|
EnableGitPull: true,
|
||||||
|
Path: map[string]string{
|
||||||
|
"beego": ".",
|
||||||
|
},
|
||||||
|
EnableGomod: true,
|
||||||
|
RefreshGitTime: 24 * 3600,
|
||||||
|
Extend: nil,
|
||||||
|
},
|
||||||
|
GenerateTime: time.Now().Format(MDateFormat),
|
||||||
|
GenerateTimeUnix: time.Now().Unix(),
|
||||||
|
TmplOption: TmplOption{},
|
||||||
|
CurPath: system.CurrentDir,
|
||||||
|
EnableModules: make(map[string]interface{}), // get the user configuration, get the enable module result
|
||||||
|
FunctionOnce: make(map[string]sync.Once), // get the tmpl configuration, get the function once result
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Container) Run() {
|
||||||
|
// init git refresh cache time
|
||||||
|
c.initTimestamp()
|
||||||
|
c.initUserOption()
|
||||||
|
c.initTemplateOption()
|
||||||
|
c.initParser()
|
||||||
|
c.initRender()
|
||||||
|
c.flushTimestamp()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Container) initUserOption() {
|
||||||
|
if !utils.IsExist(c.BeegoProFile) {
|
||||||
|
beeLogger.Log.Fatalf("beego pro config is not exist, beego json path: %s", c.BeegoProFile)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
viper.SetConfigFile(c.BeegoProFile)
|
||||||
|
err := viper.ReadInConfig()
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("read beego pro config content, err: %s", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
err = viper.Unmarshal(&c.UserOption)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("beego pro config unmarshal error, err: %s", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if c.UserOption.Debug {
|
||||||
|
viper.Debug()
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.UserOption.EnableGomod {
|
||||||
|
if !utils.IsExist(c.GoModFile) {
|
||||||
|
beeLogger.Log.Fatalf("go mod not exist, please create go mod file")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, value := range c.UserOption.EnableModule {
|
||||||
|
c.EnableModules[value] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(c.EnableModules) == 0 {
|
||||||
|
c.EnableModules["*"] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.UserOption.Debug {
|
||||||
|
fmt.Println("c.modules", c.EnableModules)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Container) initTemplateOption() {
|
||||||
|
c.GetLocalPath()
|
||||||
|
if c.UserOption.EnableGitPull && (c.GenerateTimeUnix-c.Timestamp.GitCacheLastRefresh > c.UserOption.RefreshGitTime) {
|
||||||
|
err := git.CloneORPullRepo(c.UserOption.GitRemotePath, c.UserOption.GitLocalPath)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("beego pro git clone or pull repo error, err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Timestamp.GitCacheLastRefresh = c.GenerateTimeUnix
|
||||||
|
}
|
||||||
|
|
||||||
|
tree, err := toml.LoadFile(c.UserOption.GitLocalPath + "/" + c.UserOption.ProType + "/bee.toml")
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("beego tmpl exec error, err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
err = tree.Unmarshal(&c.TmplOption)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("beego tmpl parse error, err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.UserOption.Debug {
|
||||||
|
spew.Dump("tmpl", c.TmplOption)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, value := range c.TmplOption.Descriptor {
|
||||||
|
if value.Once {
|
||||||
|
c.FunctionOnce[value.SrcName] = sync.Once{}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Container) initParser() {
|
||||||
|
driver, flag := ParserDriver[c.UserOption.SourceGen]
|
||||||
|
if !flag {
|
||||||
|
beeLogger.Log.Fatalf("parse driver not exit, source gen %s", c.UserOption.SourceGen)
|
||||||
|
}
|
||||||
|
driver.RegisterOption(c.UserOption, c.TmplOption)
|
||||||
|
c.Parser = driver
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Container) initRender() {
|
||||||
|
for _, desc := range c.TmplOption.Descriptor {
|
||||||
|
_, allFlag := c.EnableModules["*"]
|
||||||
|
_, moduleFlag := c.EnableModules[desc.Module]
|
||||||
|
if !allFlag && !moduleFlag {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
models := c.Parser.GetRenderInfos(desc)
|
||||||
|
// model table name, model table schema
|
||||||
|
for _, m := range models {
|
||||||
|
// some render exec once
|
||||||
|
syncOnce, flag := c.FunctionOnce[desc.SrcName]
|
||||||
|
if flag {
|
||||||
|
syncOnce.Do(func() {
|
||||||
|
c.renderModel(m)
|
||||||
|
})
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
c.renderModel(m)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Container) renderModel(m RenderInfo) {
|
||||||
|
// todo optimize
|
||||||
|
m.GenerateTime = c.GenerateTime
|
||||||
|
render := NewRender(m)
|
||||||
|
render.Exec(m.Descriptor.SrcName)
|
||||||
|
if render.Descriptor.IsExistScript() {
|
||||||
|
err := render.Descriptor.ExecScript(c.CurPath)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("beego exec shell error, err: %s", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Container) initTimestamp() {
|
||||||
|
if utils.IsExist(c.TimestampFile) {
|
||||||
|
tree, err := toml.LoadFile(c.TimestampFile)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("beego timestamp tmpl exec error, err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
err = tree.Unmarshal(&c.Timestamp)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("beego timestamp tmpl parse error, err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
c.Timestamp.Generate = c.GenerateTimeUnix
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Container) flushTimestamp() {
|
||||||
|
tomlByte, err := toml.Marshal(c.Timestamp)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("marshal timestamp tmpl parse error, err: %s", err)
|
||||||
|
}
|
||||||
|
err = ioutil.WriteFile(c.TimestampFile, tomlByte, 0644)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("flush timestamp tmpl parse error, err: %s", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Container) InitToml() {
|
||||||
|
if exist := utils.IsExist(c.BeegoProFile); exist {
|
||||||
|
beeLogger.Log.Fatalf("file beegopro.toml already exists")
|
||||||
|
}
|
||||||
|
sourceFile := c.UserOption.GitLocalPath + "/beegopro.toml"
|
||||||
|
input, err := ioutil.ReadFile(sourceFile)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("read beegopro.toml file err, %s", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
err = ioutil.WriteFile(c.BeegoProFile, input, 0644)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("create beegopro.toml file err, %s", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
beeLogger.Log.Success("Successfully created file beegopro.toml")
|
||||||
|
}
|
||||||
|
|
||||||
|
//form https://github.com/beego/beego-pro.git
|
||||||
|
//get beego/beego-pro
|
||||||
|
func (c *Container) GetLocalPath() {
|
||||||
|
if c.UserOption.GitLocalPath != "" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if GitRemotePath != "" {
|
||||||
|
c.UserOption.GitRemotePath = GitRemotePath.String()
|
||||||
|
}
|
||||||
|
if c.UserOption.GitRemotePath == "" {
|
||||||
|
c.UserOption.GitRemotePath = "https://github.com/beego/beego-pro.git"
|
||||||
|
}
|
||||||
|
parse, err := url.Parse(c.UserOption.GitRemotePath)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("git GitRemotePath err, %s", err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
s := parse.Path
|
||||||
|
s = strings.TrimRight(s, ".git")
|
||||||
|
c.UserOption.GitLocalPath = system.BeegoHome + s
|
||||||
|
}
|
80
internal/app/module/beegopro/migration.go
Normal file
80
internal/app/module/beegopro/migration.go
Normal file
|
@ -0,0 +1,80 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"io/ioutil"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
var SQL utils.DocValue
|
||||||
|
var SQLMode utils.DocValue
|
||||||
|
var SQLModePath utils.DocValue
|
||||||
|
|
||||||
|
var (
|
||||||
|
SQLModeUp = "up"
|
||||||
|
SQLModeDown = "down"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (c *Container) Migration(args []string) {
|
||||||
|
c.initUserOption()
|
||||||
|
db, err := sql.Open(c.UserOption.Driver, c.UserOption.Dsn)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not connect to '%s' database using '%s': %s", c.UserOption.Driver, c.UserOption.Dsn, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
switch SQLMode.String() {
|
||||||
|
case SQLModeUp:
|
||||||
|
doByMode(db, "up.sql")
|
||||||
|
case SQLModeDown:
|
||||||
|
doByMode(db, "down.sql")
|
||||||
|
default:
|
||||||
|
doBySqlFile(db)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func doBySqlFile(db *sql.DB) {
|
||||||
|
fileName := SQL.String()
|
||||||
|
if !utils.IsExist(fileName) {
|
||||||
|
beeLogger.Log.Fatalf("sql mode path not exist, path %s", SQL.String())
|
||||||
|
}
|
||||||
|
doDb(db, fileName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func doByMode(db *sql.DB, suffix string) {
|
||||||
|
pathName := SQLModePath.String()
|
||||||
|
if !utils.IsExist(pathName) {
|
||||||
|
beeLogger.Log.Fatalf("sql mode path not exist, path %s", SQLModePath.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
rd, err := ioutil.ReadDir(pathName)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("read dir err, path %s, err %s", pathName, err)
|
||||||
|
}
|
||||||
|
for _, fi := range rd {
|
||||||
|
if !fi.IsDir() {
|
||||||
|
if !strings.HasSuffix(fi.Name(), suffix) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
doDb(db, filepath.Join(pathName, fi.Name()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func doDb(db *sql.DB, filePath string) {
|
||||||
|
absFile, _ := filepath.Abs(filePath)
|
||||||
|
content, err := ioutil.ReadFile(filePath)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("read file err %s, abs file %s", err, absFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = db.Exec(string(content))
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("db exec err %s", err)
|
||||||
|
}
|
||||||
|
beeLogger.Log.Infof("db exec info %s", filePath)
|
||||||
|
}
|
13
internal/app/module/beegopro/parser.go
Normal file
13
internal/app/module/beegopro/parser.go
Normal file
|
@ -0,0 +1,13 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
type Parser interface {
|
||||||
|
RegisterOption(userOption UserOption, tmplOption TmplOption)
|
||||||
|
Parse(descriptor Descriptor)
|
||||||
|
GetRenderInfos(descriptor Descriptor) (output []RenderInfo)
|
||||||
|
Unregister()
|
||||||
|
}
|
||||||
|
|
||||||
|
var ParserDriver = map[string]Parser{
|
||||||
|
"text": &TextParser{},
|
||||||
|
"mysql": &MysqlParser{},
|
||||||
|
}
|
88
internal/app/module/beegopro/parser_mysql.go
Normal file
88
internal/app/module/beegopro/parser_mysql.go
Normal file
|
@ -0,0 +1,88 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/internal/pkg/utils"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
type MysqlParser struct {
|
||||||
|
userOption UserOption
|
||||||
|
tmplOption TmplOption
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MysqlParser) RegisterOption(userOption UserOption, tmplOption TmplOption) {
|
||||||
|
m.userOption = userOption
|
||||||
|
m.tmplOption = tmplOption
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*MysqlParser) Parse(descriptor Descriptor) {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MysqlParser) GetRenderInfos(descriptor Descriptor) (output []RenderInfo) {
|
||||||
|
tableSchemas, err := m.getTableSchemas()
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("get table schemas err %s", err)
|
||||||
|
}
|
||||||
|
models := tableSchemas.ToTableMap()
|
||||||
|
|
||||||
|
output = make([]RenderInfo, 0)
|
||||||
|
// model table name, model table schema
|
||||||
|
for modelName, content := range models {
|
||||||
|
output = append(output, RenderInfo{
|
||||||
|
Module: descriptor.Module,
|
||||||
|
ModelName: modelName,
|
||||||
|
Content: content,
|
||||||
|
Option: m.userOption,
|
||||||
|
Descriptor: descriptor,
|
||||||
|
TmplPath: m.tmplOption.RenderPath,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *MysqlParser) Unregister() {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MysqlParser) getTableSchemas() (resp TableSchemas, err error) {
|
||||||
|
dsn, err := utils.ParseDSN(m.userOption.Dsn)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("parse dsn err %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
conn, err := sql.Open("mysql", fmt.Sprintf("%s:%s@tcp(%s)/information_schema", dsn.User, dsn.Passwd, dsn.Addr))
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not connect to mysql database using '%s': %s", m.userOption.Dsn, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer conn.Close()
|
||||||
|
|
||||||
|
q := `SELECT TABLE_NAME, COLUMN_NAME, IS_NULLABLE, DATA_TYPE, CHARACTER_MAXIMUM_LENGTH,
|
||||||
|
NUMERIC_PRECISION, NUMERIC_SCALE,COLUMN_TYPE,COLUMN_KEY,COLUMN_COMMENT
|
||||||
|
FROM COLUMNS WHERE TABLE_SCHEMA = ? ORDER BY TABLE_NAME, ORDINAL_POSITION`
|
||||||
|
rows, err := conn.Query(q, dsn.DBName)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
columns := make(TableSchemas, 0)
|
||||||
|
for rows.Next() {
|
||||||
|
cs := TableSchema{}
|
||||||
|
err := rows.Scan(&cs.TableName, &cs.ColumnName, &cs.IsNullable, &cs.DataType,
|
||||||
|
&cs.CharacterMaximumLength, &cs.NumericPrecision, &cs.NumericScale,
|
||||||
|
&cs.ColumnType, &cs.ColumnKey, &cs.Comment)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
columns = append(columns, cs)
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return columns, nil
|
||||||
|
}
|
35
internal/app/module/beegopro/parser_text.go
Normal file
35
internal/app/module/beegopro/parser_text.go
Normal file
|
@ -0,0 +1,35 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
type TextParser struct {
|
||||||
|
userOption UserOption
|
||||||
|
tmplOption TmplOption
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TextParser) RegisterOption(userOption UserOption, tmplOption TmplOption) {
|
||||||
|
t.userOption = userOption
|
||||||
|
t.tmplOption = tmplOption
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*TextParser) Parse(descriptor Descriptor) {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TextParser) GetRenderInfos(descriptor Descriptor) (output []RenderInfo) {
|
||||||
|
output = make([]RenderInfo, 0)
|
||||||
|
// model table name, model table schema
|
||||||
|
for modelName, content := range t.userOption.Models {
|
||||||
|
output = append(output, RenderInfo{
|
||||||
|
Module: descriptor.Module,
|
||||||
|
ModelName: modelName,
|
||||||
|
Content: content.ToModelInfos(),
|
||||||
|
Option: t.userOption,
|
||||||
|
Descriptor: descriptor,
|
||||||
|
TmplPath: t.tmplOption.RenderPath,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TextParser) Unregister() {
|
||||||
|
|
||||||
|
}
|
59
internal/app/module/beegopro/pongo2.go
Normal file
59
internal/app/module/beegopro/pongo2.go
Normal file
|
@ -0,0 +1,59 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
"unicode/utf8"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
"github.com/flosch/pongo2"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
_ = pongo2.RegisterFilter("lowerFirst", pongo2LowerFirst)
|
||||||
|
_ = pongo2.RegisterFilter("upperFirst", pongo2UpperFirst)
|
||||||
|
_ = pongo2.RegisterFilter("snakeString", pongo2SnakeString)
|
||||||
|
_ = pongo2.RegisterFilter("camelString", pongo2CamelString)
|
||||||
|
}
|
||||||
|
|
||||||
|
func pongo2LowerFirst(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) {
|
||||||
|
if in.Len() <= 0 {
|
||||||
|
return pongo2.AsValue(""), nil
|
||||||
|
}
|
||||||
|
t := in.String()
|
||||||
|
r, size := utf8.DecodeRuneInString(t)
|
||||||
|
return pongo2.AsValue(strings.ToLower(string(r)) + t[size:]), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func pongo2UpperFirst(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) {
|
||||||
|
if in.Len() <= 0 {
|
||||||
|
return pongo2.AsValue(""), nil
|
||||||
|
}
|
||||||
|
t := in.String()
|
||||||
|
return pongo2.AsValue(strings.Replace(t, string(t[0]), strings.ToUpper(string(t[0])), 1)), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// snake string, XxYy to xx_yy
|
||||||
|
func pongo2SnakeString(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) {
|
||||||
|
if in.Len() <= 0 {
|
||||||
|
return pongo2.AsValue(""), nil
|
||||||
|
}
|
||||||
|
t := in.String()
|
||||||
|
return pongo2.AsValue(utils.SnakeString(t)), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// snake string, XxYy to xx_yy
|
||||||
|
func pongo2CamelString(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) {
|
||||||
|
if in.Len() <= 0 {
|
||||||
|
return pongo2.AsValue(""), nil
|
||||||
|
}
|
||||||
|
t := in.String()
|
||||||
|
return pongo2.AsValue(utils.CamelString(t)), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
//func upperFirst(str string) string {
|
||||||
|
// return strings.Replace(str, string(str[0]), strings.ToUpper(string(str[0])), 1)
|
||||||
|
//}
|
||||||
|
|
||||||
|
func lowerFirst(str string) string {
|
||||||
|
return strings.Replace(str, string(str[0]), strings.ToLower(string(str[0])), 1)
|
||||||
|
}
|
153
internal/app/module/beegopro/render.go
Normal file
153
internal/app/module/beegopro/render.go
Normal file
|
@ -0,0 +1,153 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
"go/format"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"path/filepath"
|
||||||
|
|
||||||
|
"github.com/davecgh/go-spew/spew"
|
||||||
|
"github.com/flosch/pongo2"
|
||||||
|
"github.com/smartwalle/pongo2render"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/internal/pkg/system"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
// render
|
||||||
|
type RenderFile struct {
|
||||||
|
*pongo2render.Render
|
||||||
|
Context pongo2.Context
|
||||||
|
GenerateTime string
|
||||||
|
Option UserOption
|
||||||
|
ModelName string
|
||||||
|
PackageName string
|
||||||
|
FlushFile string
|
||||||
|
PkgPath string
|
||||||
|
TmplPath string
|
||||||
|
Descriptor Descriptor
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewRender(m RenderInfo) *RenderFile {
|
||||||
|
var (
|
||||||
|
pathCtx pongo2.Context
|
||||||
|
newDescriptor Descriptor
|
||||||
|
)
|
||||||
|
|
||||||
|
// parse descriptor, get flush file path, beego path, etc...
|
||||||
|
newDescriptor, pathCtx = m.Descriptor.Parse(m.ModelName, m.Option.Path)
|
||||||
|
|
||||||
|
obj := &RenderFile{
|
||||||
|
Context: make(pongo2.Context),
|
||||||
|
Option: m.Option,
|
||||||
|
ModelName: m.ModelName,
|
||||||
|
GenerateTime: m.GenerateTime,
|
||||||
|
Descriptor: newDescriptor,
|
||||||
|
}
|
||||||
|
|
||||||
|
obj.FlushFile = newDescriptor.DstPath
|
||||||
|
|
||||||
|
// new render
|
||||||
|
obj.Render = pongo2render.NewRender(path.Join(obj.Option.GitLocalPath, obj.Option.ProType, m.TmplPath))
|
||||||
|
|
||||||
|
filePath := path.Dir(obj.FlushFile)
|
||||||
|
err := createPath(filePath)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not create the controllers directory: %s", err)
|
||||||
|
}
|
||||||
|
// get go package path
|
||||||
|
obj.PkgPath = getPackagePath()
|
||||||
|
|
||||||
|
relativePath, err := filepath.Rel(system.CurrentDir, obj.FlushFile)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not get the relative path: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
modelSchemas := m.Content.ToModelSchemas()
|
||||||
|
camelPrimaryKey := modelSchemas.GetPrimaryKey()
|
||||||
|
importMaps := make(map[string]struct{})
|
||||||
|
if modelSchemas.IsExistTime() {
|
||||||
|
importMaps["time"] = struct{}{}
|
||||||
|
}
|
||||||
|
obj.PackageName = filepath.Base(filepath.Dir(relativePath))
|
||||||
|
beeLogger.Log.Infof("Using '%s' as name", obj.ModelName)
|
||||||
|
|
||||||
|
beeLogger.Log.Infof("Using '%s' as package name from %s", obj.ModelName, obj.PackageName)
|
||||||
|
|
||||||
|
// package
|
||||||
|
obj.SetContext("packageName", obj.PackageName)
|
||||||
|
obj.SetContext("packageImports", importMaps)
|
||||||
|
|
||||||
|
// todo optimize
|
||||||
|
// todo Set the beego directory, should recalculate the package
|
||||||
|
if pathCtx["pathRelBeego"] == "." {
|
||||||
|
obj.SetContext("packagePath", obj.PkgPath)
|
||||||
|
} else {
|
||||||
|
obj.SetContext("packagePath", obj.PkgPath+"/"+pathCtx["pathRelBeego"].(string))
|
||||||
|
}
|
||||||
|
|
||||||
|
obj.SetContext("packageMod", obj.PkgPath)
|
||||||
|
|
||||||
|
obj.SetContext("modelSchemas", modelSchemas)
|
||||||
|
obj.SetContext("modelPrimaryKey", camelPrimaryKey)
|
||||||
|
|
||||||
|
for key, value := range pathCtx {
|
||||||
|
obj.SetContext(key, value)
|
||||||
|
}
|
||||||
|
|
||||||
|
obj.SetContext("apiPrefix", obj.Option.ApiPrefix)
|
||||||
|
obj.SetContext("generateTime", obj.GenerateTime)
|
||||||
|
|
||||||
|
if obj.Option.ContextDebug {
|
||||||
|
spew.Dump(obj.Context)
|
||||||
|
}
|
||||||
|
return obj
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *RenderFile) SetContext(key string, value interface{}) {
|
||||||
|
r.Context[key] = value
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *RenderFile) Exec(name string) {
|
||||||
|
var (
|
||||||
|
buf string
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
buf, err = r.Render.Template(name).Execute(r.Context)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not create the %s render tmpl: %s", name, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
_, err = os.Stat(r.Descriptor.DstPath)
|
||||||
|
var orgContent []byte
|
||||||
|
if err == nil {
|
||||||
|
if org, err := os.OpenFile(r.Descriptor.DstPath, os.O_RDONLY, 0666); err == nil {
|
||||||
|
orgContent, _ = ioutil.ReadAll(org)
|
||||||
|
org.Close()
|
||||||
|
} else {
|
||||||
|
beeLogger.Log.Infof("file err %s", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Replace or create when content changes
|
||||||
|
output := []byte(buf)
|
||||||
|
ext := filepath.Ext(r.FlushFile)
|
||||||
|
if r.Option.EnableFormat && ext == ".go" {
|
||||||
|
// format code
|
||||||
|
var bts []byte
|
||||||
|
bts, err = format.Source([]byte(buf))
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Warnf("format buf error %s", err.Error())
|
||||||
|
}
|
||||||
|
output = bts
|
||||||
|
}
|
||||||
|
|
||||||
|
if FileContentChange(orgContent, output, GetSeg(ext)) {
|
||||||
|
err = r.write(r.FlushFile, output)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not create file: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
beeLogger.Log.Infof("create file '%s' from %s", r.FlushFile, r.PackageName)
|
||||||
|
}
|
||||||
|
}
|
142
internal/app/module/beegopro/schema.go
Normal file
142
internal/app/module/beegopro/schema.go
Normal file
|
@ -0,0 +1,142 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/internal/pkg/command"
|
||||||
|
"github.com/beego/bee/v2/internal/pkg/system"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
"github.com/flosch/pongo2"
|
||||||
|
"github.com/smartwalle/pongo2render"
|
||||||
|
)
|
||||||
|
|
||||||
|
// store all data
|
||||||
|
type Container struct {
|
||||||
|
BeegoProFile string // beego pro toml
|
||||||
|
TimestampFile string // store ts file
|
||||||
|
GoModFile string // go mod file
|
||||||
|
UserOption UserOption // user option
|
||||||
|
TmplOption TmplOption // tmpl option
|
||||||
|
CurPath string // user current path
|
||||||
|
EnableModules map[string]interface{} // beego pro provider a collection of module
|
||||||
|
FunctionOnce map[string]sync.Once // exec function once
|
||||||
|
Timestamp Timestamp
|
||||||
|
GenerateTime string
|
||||||
|
GenerateTimeUnix int64
|
||||||
|
Parser Parser
|
||||||
|
}
|
||||||
|
|
||||||
|
// user option
|
||||||
|
type UserOption struct {
|
||||||
|
Debug bool `json:"debug"`
|
||||||
|
ContextDebug bool `json:"contextDebug"`
|
||||||
|
Dsn string `json:"dsn"`
|
||||||
|
Driver string `json:"driver"`
|
||||||
|
ProType string `json:"proType"`
|
||||||
|
ApiPrefix string `json:"apiPrefix"`
|
||||||
|
EnableModule []string `json:"enableModule"`
|
||||||
|
Models map[string]TextModel `json:"models"`
|
||||||
|
GitRemotePath string `json:"gitRemotePath"`
|
||||||
|
Branch string `json:"branch"`
|
||||||
|
GitLocalPath string `json:"gitLocalPath"`
|
||||||
|
EnableFormat bool `json:"enableFormat"`
|
||||||
|
SourceGen string `json:"sourceGen"`
|
||||||
|
EnableGitPull bool `json:"enbaleGitPull"`
|
||||||
|
Path map[string]string `json:"path"`
|
||||||
|
EnableGomod bool `json:"enableGomod"`
|
||||||
|
RefreshGitTime int64 `json:"refreshGitTime"`
|
||||||
|
Extend map[string]string `json:"extend"` // extend user data
|
||||||
|
}
|
||||||
|
|
||||||
|
// tmpl option
|
||||||
|
type TmplOption struct {
|
||||||
|
RenderPath string `toml:"renderPath"`
|
||||||
|
Descriptor []Descriptor
|
||||||
|
}
|
||||||
|
|
||||||
|
type Descriptor struct {
|
||||||
|
Module string `toml:"module"`
|
||||||
|
SrcName string `toml:"srcName"`
|
||||||
|
DstPath string `toml:"dstPath"`
|
||||||
|
Once bool `toml:"once"`
|
||||||
|
Script string `toml:"script"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (descriptor Descriptor) Parse(modelName string, paths map[string]string) (newDescriptor Descriptor, ctx pongo2.Context) {
|
||||||
|
var (
|
||||||
|
err error
|
||||||
|
relativeDstPath string
|
||||||
|
absFile string
|
||||||
|
relPath string
|
||||||
|
)
|
||||||
|
|
||||||
|
newDescriptor = descriptor
|
||||||
|
render := pongo2render.NewRender("")
|
||||||
|
ctx = make(pongo2.Context)
|
||||||
|
for key, value := range paths {
|
||||||
|
absFile, err = filepath.Abs(value)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("absolute path error %s from key %s and value %s", err, key, value)
|
||||||
|
}
|
||||||
|
relPath, err = filepath.Rel(system.CurrentDir, absFile)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not get the relative path: %s", err)
|
||||||
|
}
|
||||||
|
// user input path
|
||||||
|
ctx["path"+utils.CamelCase(key)] = value
|
||||||
|
// relativePath
|
||||||
|
ctx["pathRel"+utils.CamelCase(key)] = relPath
|
||||||
|
}
|
||||||
|
ctx["modelName"] = lowerFirst(utils.CamelString(modelName))
|
||||||
|
relativeDstPath, err = render.TemplateFromString(descriptor.DstPath).Execute(ctx)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("beego tmpl exec error, err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
newDescriptor.DstPath, err = filepath.Abs(relativeDstPath)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("absolute path error %s from flush file %s", err, relativeDstPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
newDescriptor.Script, err = render.TemplateFromString(descriptor.Script).Execute(ctx)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("parse script %s, error %s", descriptor.Script, err)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (descriptor Descriptor) IsExistScript() bool {
|
||||||
|
return descriptor.Script != ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d Descriptor) ExecScript(path string) (err error) {
|
||||||
|
arr := strings.Split(d.Script, " ")
|
||||||
|
if len(arr) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
stdout, stderr, err := command.ExecCmdDir(path, arr[0], arr[1:]...)
|
||||||
|
if err != nil {
|
||||||
|
return concatenateError(err, stderr)
|
||||||
|
}
|
||||||
|
|
||||||
|
beeLogger.Log.Info(stdout)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type Timestamp struct {
|
||||||
|
GitCacheLastRefresh int64 `toml:"gitCacheLastRefresh"`
|
||||||
|
Generate int64 `toml:"generate"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func concatenateError(err error, stderr string) error {
|
||||||
|
if len(stderr) == 0 {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return fmt.Errorf("%v: %s", err, stderr)
|
||||||
|
}
|
98
internal/app/module/beegopro/schema_model.go
Normal file
98
internal/app/module/beegopro/schema_model.go
Normal file
|
@ -0,0 +1,98 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
// parse get the model info
|
||||||
|
type ModelInfo struct {
|
||||||
|
Name string `json:"name"` // mysql name
|
||||||
|
InputType string `json:"inputType"` // user input type
|
||||||
|
MysqlType string `json:"mysqlType"` // mysql type
|
||||||
|
GoType string `json:"goType"` // go type
|
||||||
|
Orm string `json:"orm"` // orm tag
|
||||||
|
Comment string `json:"comment"` // mysql comment
|
||||||
|
Extend string `json:"extend"` // user extend info
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m ModelInfo) GetColumnKey() (columnKey string) {
|
||||||
|
if m.InputType == "auto" || m.Orm == "pk" {
|
||||||
|
columnKey = "PRI"
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m ModelInfo) IsPrimaryKey() (flag bool) {
|
||||||
|
if m.Orm == "auto" || m.Orm == "pk" || strings.ToLower(m.Name) == "id" {
|
||||||
|
flag = true
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
type ModelInfos []ModelInfo
|
||||||
|
|
||||||
|
// to render model schemas
|
||||||
|
func (modelInfos ModelInfos) ToModelSchemas() (output ModelSchemas) {
|
||||||
|
output = make(ModelSchemas, 0)
|
||||||
|
for i, value := range modelInfos {
|
||||||
|
if i == 0 && !value.IsPrimaryKey() {
|
||||||
|
inputType, goType, mysqlType, ormTag := getModelType("auto")
|
||||||
|
output = append(output, &ModelSchema{
|
||||||
|
Name: "id",
|
||||||
|
InputType: inputType,
|
||||||
|
ColumnKey: "PRI",
|
||||||
|
Comment: "ID",
|
||||||
|
MysqlType: mysqlType,
|
||||||
|
GoType: goType,
|
||||||
|
OrmTag: ormTag,
|
||||||
|
Extend: value.Extend,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
modelSchema := &ModelSchema{
|
||||||
|
Name: value.Name,
|
||||||
|
InputType: value.InputType,
|
||||||
|
ColumnKey: value.GetColumnKey(),
|
||||||
|
MysqlType: value.MysqlType,
|
||||||
|
Comment: value.Comment,
|
||||||
|
GoType: value.GoType,
|
||||||
|
OrmTag: value.Orm,
|
||||||
|
}
|
||||||
|
output = append(output, modelSchema)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
type ModelSchema struct {
|
||||||
|
Name string // column name
|
||||||
|
InputType string // user input type
|
||||||
|
MysqlType string // mysql type
|
||||||
|
ColumnKey string // PRI
|
||||||
|
Comment string // comment
|
||||||
|
GoType string // go type
|
||||||
|
OrmTag string // orm tag
|
||||||
|
Extend string
|
||||||
|
}
|
||||||
|
|
||||||
|
type ModelSchemas []*ModelSchema
|
||||||
|
|
||||||
|
func (m ModelSchemas) IsExistTime() bool {
|
||||||
|
for _, value := range m {
|
||||||
|
if value.InputType == "datetime" {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m ModelSchemas) GetPrimaryKey() string {
|
||||||
|
camelPrimaryKey := ""
|
||||||
|
for _, value := range m {
|
||||||
|
if value.ColumnKey == "PRI" {
|
||||||
|
camelPrimaryKey = utils.CamelString(value.Name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return camelPrimaryKey
|
||||||
|
}
|
75
internal/app/module/beegopro/schema_mysql_model.go
Normal file
75
internal/app/module/beegopro/schema_mysql_model.go
Normal file
|
@ -0,0 +1,75 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"errors"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TableSchema struct {
|
||||||
|
TableName string
|
||||||
|
ColumnName string
|
||||||
|
IsNullable string
|
||||||
|
DataType string
|
||||||
|
CharacterMaximumLength sql.NullInt64
|
||||||
|
NumericPrecision sql.NullInt64
|
||||||
|
NumericScale sql.NullInt64
|
||||||
|
ColumnType string
|
||||||
|
ColumnKey string
|
||||||
|
Comment string
|
||||||
|
}
|
||||||
|
|
||||||
|
type TableSchemas []TableSchema
|
||||||
|
|
||||||
|
func (tableSchemas TableSchemas) ToTableMap() (resp map[string]ModelInfos) {
|
||||||
|
|
||||||
|
resp = make(map[string]ModelInfos)
|
||||||
|
for _, value := range tableSchemas {
|
||||||
|
if _, ok := resp[value.TableName]; !ok {
|
||||||
|
resp[value.TableName] = make(ModelInfos, 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
modelInfos := resp[value.TableName]
|
||||||
|
inputType, goType, err := value.ToGoType()
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("parse go type err %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
modelInfo := ModelInfo{
|
||||||
|
Name: value.ColumnName,
|
||||||
|
InputType: inputType,
|
||||||
|
GoType: goType,
|
||||||
|
Comment: value.Comment,
|
||||||
|
}
|
||||||
|
|
||||||
|
if value.ColumnKey == "PRI" {
|
||||||
|
modelInfo.Orm = "pk"
|
||||||
|
}
|
||||||
|
resp[value.TableName] = append(modelInfos, modelInfo)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetGoDataType maps an SQL data type to Golang data type
|
||||||
|
func (col TableSchema) ToGoType() (inputType string, goType string, err error) {
|
||||||
|
switch col.DataType {
|
||||||
|
case "char", "varchar", "enum", "set", "text", "longtext", "mediumtext", "tinytext":
|
||||||
|
goType = "string"
|
||||||
|
case "blob", "mediumblob", "longblob", "varbinary", "binary":
|
||||||
|
goType = "[]byte"
|
||||||
|
case "date", "time", "datetime", "timestamp":
|
||||||
|
goType, inputType = "time.Time", "dateTime"
|
||||||
|
case "tinyint", "smallint", "int", "mediumint":
|
||||||
|
goType = "int"
|
||||||
|
case "bit", "bigint":
|
||||||
|
goType = "int64"
|
||||||
|
case "float", "decimal", "double":
|
||||||
|
goType = "float64"
|
||||||
|
}
|
||||||
|
if goType == "" {
|
||||||
|
err = errors.New("No compatible datatype (" + col.DataType + ", CamelName: " + col.ColumnName + ") found")
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
11
internal/app/module/beegopro/schema_render.go
Normal file
11
internal/app/module/beegopro/schema_render.go
Normal file
|
@ -0,0 +1,11 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
type RenderInfo struct {
|
||||||
|
Module string
|
||||||
|
ModelName string
|
||||||
|
Option UserOption
|
||||||
|
Content ModelInfos
|
||||||
|
Descriptor Descriptor
|
||||||
|
TmplPath string
|
||||||
|
GenerateTime string
|
||||||
|
}
|
50
internal/app/module/beegopro/schema_text_model.go
Normal file
50
internal/app/module/beegopro/schema_text_model.go
Normal file
|
@ -0,0 +1,50 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TextModel struct {
|
||||||
|
Names []string
|
||||||
|
Orms []string
|
||||||
|
Comments []string
|
||||||
|
Extends []string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (content TextModel) ToModelInfos() (output []ModelInfo) {
|
||||||
|
namesLen := len(content.Names)
|
||||||
|
ormsLen := len(content.Orms)
|
||||||
|
commentsLen := len(content.Comments)
|
||||||
|
if namesLen != ormsLen && namesLen != commentsLen {
|
||||||
|
beeLogger.Log.Fatalf("length error, namesLen is %d, ormsLen is %d, commentsLen is %d", namesLen, ormsLen, commentsLen)
|
||||||
|
}
|
||||||
|
extendLen := len(content.Extends)
|
||||||
|
if extendLen != 0 && extendLen != namesLen {
|
||||||
|
beeLogger.Log.Fatalf("extend length error, namesLen is %d, extendsLen is %d", namesLen, extendLen)
|
||||||
|
}
|
||||||
|
|
||||||
|
output = make([]ModelInfo, 0)
|
||||||
|
for i, name := range content.Names {
|
||||||
|
comment := content.Comments[i]
|
||||||
|
if comment == "" {
|
||||||
|
comment = name
|
||||||
|
}
|
||||||
|
inputType, goType, mysqlType, ormTag := getModelType(content.Orms[i])
|
||||||
|
|
||||||
|
m := ModelInfo{
|
||||||
|
Name: name,
|
||||||
|
InputType: inputType,
|
||||||
|
GoType: goType,
|
||||||
|
Orm: ormTag,
|
||||||
|
Comment: comment,
|
||||||
|
MysqlType: mysqlType,
|
||||||
|
Extend: "",
|
||||||
|
}
|
||||||
|
// extend value
|
||||||
|
if extendLen != 0 {
|
||||||
|
m.Extend = content.Extends[i]
|
||||||
|
}
|
||||||
|
output = append(output, m)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
215
internal/app/module/beegopro/util.go
Normal file
215
internal/app/module/beegopro/util.go
Normal file
|
@ -0,0 +1,215 @@
|
||||||
|
package beegopro
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/md5"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/internal/pkg/utils"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
// write to file
|
||||||
|
func (c *RenderFile) write(filename string, buf []byte) (err error) {
|
||||||
|
if utils.IsExist(filename) && !isNeedOverwrite(filename) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
filePath := filepath.Dir(filename)
|
||||||
|
err = createPath(filePath)
|
||||||
|
if err != nil {
|
||||||
|
err = errors.New("write create path " + err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
filePathBak := filePath + "/bak"
|
||||||
|
err = createPath(filePathBak)
|
||||||
|
if err != nil {
|
||||||
|
err = errors.New("write create path bak " + err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
name := path.Base(filename)
|
||||||
|
|
||||||
|
if utils.IsExist(filename) {
|
||||||
|
bakName := fmt.Sprintf("%s/%s.%s.bak", filePathBak, filepath.Base(name), time.Now().Format("2006.01.02.15.04.05"))
|
||||||
|
beeLogger.Log.Infof("bak file '%s'", bakName)
|
||||||
|
if err := os.Rename(filename, bakName); err != nil {
|
||||||
|
err = errors.New("file is bak error, path is " + bakName)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
file, err := os.Create(filename)
|
||||||
|
defer func() {
|
||||||
|
err = file.Close()
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("file close error, err %s", err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
if err != nil {
|
||||||
|
err = errors.New("write create file " + err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
err = ioutil.WriteFile(filename, buf, 0644)
|
||||||
|
if err != nil {
|
||||||
|
err = errors.New("write write file " + err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func isNeedOverwrite(fileName string) (flag bool) {
|
||||||
|
seg := GetSeg(filepath.Ext(fileName))
|
||||||
|
|
||||||
|
f, err := os.Open(fileName)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
overwrite := ""
|
||||||
|
var contentByte []byte
|
||||||
|
contentByte, err = ioutil.ReadAll(f)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, s := range strings.Split(string(contentByte), "\n") {
|
||||||
|
s = strings.TrimSpace(strings.TrimPrefix(s, seg))
|
||||||
|
if strings.HasPrefix(s, "@BeeOverwrite") {
|
||||||
|
overwrite = strings.TrimSpace(s[len("@BeeOverwrite"):])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if strings.ToLower(overwrite) == "yes" {
|
||||||
|
flag = true
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// createPath 调用os.MkdirAll递归创建文件夹
|
||||||
|
func createPath(filePath string) error {
|
||||||
|
if !utils.IsExist(filePath) {
|
||||||
|
err := os.MkdirAll(filePath, os.ModePerm)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func getPackagePath() (packagePath string) {
|
||||||
|
f, err := os.Open("go.mod")
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
var contentByte []byte
|
||||||
|
contentByte, err = ioutil.ReadAll(f)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, s := range strings.Split(string(contentByte), "\n") {
|
||||||
|
packagePath = strings.TrimSpace(strings.TrimPrefix(s, "module"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func getModelType(orm string) (inputType, goType, mysqlType, tag string) {
|
||||||
|
kv := strings.SplitN(orm, ",", 2)
|
||||||
|
inputType = kv[0]
|
||||||
|
switch inputType {
|
||||||
|
case "string":
|
||||||
|
goType = "string"
|
||||||
|
tag = "size(255)"
|
||||||
|
// todo use orm data
|
||||||
|
mysqlType = "varchar(255) NOT NULL"
|
||||||
|
case "text":
|
||||||
|
goType = "string"
|
||||||
|
tag = "type(longtext)"
|
||||||
|
mysqlType = "longtext NOT NULL"
|
||||||
|
case "auto":
|
||||||
|
goType = "int"
|
||||||
|
tag = "auto"
|
||||||
|
mysqlType = "int(11) NOT NULL AUTO_INCREMENT"
|
||||||
|
case "pk":
|
||||||
|
goType = "int"
|
||||||
|
tag = "pk"
|
||||||
|
mysqlType = "int(11) NOT NULL"
|
||||||
|
case "datetime":
|
||||||
|
goType = "time.Time"
|
||||||
|
tag = "type(datetime)"
|
||||||
|
mysqlType = "datetime NOT NULL"
|
||||||
|
case "int", "int8", "int16", "int32", "int64":
|
||||||
|
fallthrough
|
||||||
|
case "uint", "uint8", "uint16", "uint32", "uint64":
|
||||||
|
goType = inputType
|
||||||
|
tag = ""
|
||||||
|
mysqlType = "int(11) DEFAULT NULL"
|
||||||
|
case "bool":
|
||||||
|
goType = inputType
|
||||||
|
tag = ""
|
||||||
|
mysqlType = "int(11) DEFAULT NULL"
|
||||||
|
case "float32", "float64":
|
||||||
|
goType = inputType
|
||||||
|
tag = ""
|
||||||
|
mysqlType = "float NOT NULL"
|
||||||
|
case "float":
|
||||||
|
goType = "float64"
|
||||||
|
tag = ""
|
||||||
|
mysqlType = "float NOT NULL"
|
||||||
|
default:
|
||||||
|
beeLogger.Log.Fatalf("not support type: %s", inputType)
|
||||||
|
}
|
||||||
|
// user set orm tag
|
||||||
|
if len(kv) == 2 {
|
||||||
|
tag = kv[1]
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func FileContentChange(org, new []byte, seg string) bool {
|
||||||
|
if len(org) == 0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
orgContent := GetFilterContent(string(org), seg)
|
||||||
|
newContent := GetFilterContent(string(new), seg)
|
||||||
|
orgMd5 := md5.Sum([]byte(orgContent))
|
||||||
|
newMd5 := md5.Sum([]byte(newContent))
|
||||||
|
if orgMd5 != newMd5 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
beeLogger.Log.Infof("File has no change in the content")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetFilterContent(content string, seg string) string {
|
||||||
|
res := ""
|
||||||
|
for _, s := range strings.Split(content, "\n") {
|
||||||
|
s = strings.TrimSpace(strings.TrimPrefix(s, seg))
|
||||||
|
var have bool
|
||||||
|
for _, except := range CompareExcept {
|
||||||
|
if strings.HasPrefix(s, except) {
|
||||||
|
have = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !have {
|
||||||
|
res += s
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetSeg(ext string) string {
|
||||||
|
switch ext {
|
||||||
|
case ".sql":
|
||||||
|
return "--"
|
||||||
|
default:
|
||||||
|
return "//"
|
||||||
|
}
|
||||||
|
}
|
65
internal/pkg/command/cmd.go
Normal file
65
internal/pkg/command/cmd.go
Normal file
|
@ -0,0 +1,65 @@
|
||||||
|
package command
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"os/exec"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ExecCmdDirBytes executes system command in given directory
|
||||||
|
// and return stdout, stderr in bytes type, along with possible error.
|
||||||
|
func ExecCmdDirBytes(dir, cmdName string, args ...string) ([]byte, []byte, error) {
|
||||||
|
bufOut := new(bytes.Buffer)
|
||||||
|
bufErr := new(bytes.Buffer)
|
||||||
|
|
||||||
|
cmd := exec.Command(cmdName, args...)
|
||||||
|
cmd.Dir = dir
|
||||||
|
cmd.Stdout = bufOut
|
||||||
|
cmd.Stderr = bufErr
|
||||||
|
|
||||||
|
err := cmd.Run()
|
||||||
|
return bufOut.Bytes(), bufErr.Bytes(), err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ExecCmdBytes executes system command
|
||||||
|
// and return stdout, stderr in bytes type, along with possible error.
|
||||||
|
func ExecCmdBytes(cmdName string, args ...string) ([]byte, []byte, error) {
|
||||||
|
return ExecCmdDirBytes("", cmdName, args...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ExecCmdDir executes system command in given directory
|
||||||
|
// and return stdout, stderr in string type, along with possible error.
|
||||||
|
func ExecCmdDir(dir, cmdName string, args ...string) (string, string, error) {
|
||||||
|
bufOut, bufErr, err := ExecCmdDirBytes(dir, cmdName, args...)
|
||||||
|
return string(bufOut), string(bufErr), err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ExecCmd executes system command
|
||||||
|
// and return stdout, stderr in string type, along with possible error.
|
||||||
|
func ExecCmd(cmdName string, args ...string) (string, string, error) {
|
||||||
|
return ExecCmdDir("", cmdName, args...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 版本对比 v1比v2大返回1,小于返回-1,等于返回0
|
||||||
|
func VerCompare(ver1, ver2 string) int {
|
||||||
|
ver1 = strings.TrimLeft(ver1, "ver") // 清除v,e,r
|
||||||
|
ver2 = strings.TrimLeft(ver2, "ver") // 清除v,e,r
|
||||||
|
p1 := strings.Split(ver1, ".")
|
||||||
|
p2 := strings.Split(ver2, ".")
|
||||||
|
|
||||||
|
ver1 = ""
|
||||||
|
for _, v := range p1 {
|
||||||
|
iv, _ := strconv.Atoi(v)
|
||||||
|
ver1 = fmt.Sprintf("%s%04d", ver1, iv)
|
||||||
|
}
|
||||||
|
|
||||||
|
ver2 = ""
|
||||||
|
for _, v := range p2 {
|
||||||
|
iv, _ := strconv.Atoi(v)
|
||||||
|
ver2 = fmt.Sprintf("%s%04d", ver2, iv)
|
||||||
|
}
|
||||||
|
|
||||||
|
return strings.Compare(ver1, ver2)
|
||||||
|
}
|
209
internal/pkg/git/repository.go
Normal file
209
internal/pkg/git/repository.go
Normal file
|
@ -0,0 +1,209 @@
|
||||||
|
package git
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"path/filepath"
|
||||||
|
"sort"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/internal/pkg/command"
|
||||||
|
"github.com/beego/bee/v2/internal/pkg/utils"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
// git tag
|
||||||
|
func GetTags(repoPath string, limit int) ([]string, error) {
|
||||||
|
repo, err := OpenRepository(repoPath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
err = repo.Pull()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
list, err := repo.GetTags()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if len(list) > limit {
|
||||||
|
list = list[0:limit]
|
||||||
|
}
|
||||||
|
return list, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// clone repo
|
||||||
|
func CloneRepo(url string, dst string) (err error) {
|
||||||
|
if utils.IsExist(dst) {
|
||||||
|
return errors.New("dst is not empty, dst is " + dst)
|
||||||
|
}
|
||||||
|
if !utils.Mkdir(dst) {
|
||||||
|
err = errors.New("make dir error, dst is " + dst)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
beeLogger.Log.Info("start git clone from " + url + ", to dst at " + dst)
|
||||||
|
_, stderr, err := command.ExecCmd("git", "clone", url, dst)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error("error git clone from " + url + ", to dst at " + dst)
|
||||||
|
return concatenateError(err, stderr)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// CloneORPullRepo
|
||||||
|
func CloneORPullRepo(url string, dst string) error {
|
||||||
|
if !utils.IsDir(dst) {
|
||||||
|
return CloneRepo(url, dst)
|
||||||
|
} else {
|
||||||
|
utils.Mkdir(dst)
|
||||||
|
|
||||||
|
repo, err := OpenRepository(dst)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return repo.Pull()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
func CloneRepoBranch(branch string, url string, dst string) error {
|
||||||
|
_, stderr, err := command.ExecCmd("git", "clone", "-b", branch, url, dst)
|
||||||
|
if err != nil {
|
||||||
|
return concatenateError(err, stderr)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// SortTag ...
|
||||||
|
type SortTag struct {
|
||||||
|
data []string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Len ...
|
||||||
|
func (t *SortTag) Len() int {
|
||||||
|
return len(t.data)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Swap ...
|
||||||
|
func (t *SortTag) Swap(i, j int) {
|
||||||
|
t.data[i], t.data[j] = t.data[j], t.data[i]
|
||||||
|
}
|
||||||
|
|
||||||
|
// Less ...
|
||||||
|
func (t *SortTag) Less(i, j int) bool {
|
||||||
|
return command.VerCompare(t.data[i], t.data[j]) == 1
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sort ...
|
||||||
|
func (t *SortTag) Sort() []string {
|
||||||
|
sort.Sort(t)
|
||||||
|
return t.data
|
||||||
|
}
|
||||||
|
|
||||||
|
// Repository ...
|
||||||
|
type Repository struct {
|
||||||
|
Path string
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenRepository ...
|
||||||
|
func OpenRepository(repoPath string) (*Repository, error) {
|
||||||
|
repoPath, err := filepath.Abs(repoPath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
} else if !utils.IsDir(repoPath) {
|
||||||
|
return nil, errors.New("no such file or directory")
|
||||||
|
}
|
||||||
|
|
||||||
|
return &Repository{Path: repoPath}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 拉取代码
|
||||||
|
func (repo *Repository) Pull() error {
|
||||||
|
beeLogger.Log.Info("git pull " + repo.Path)
|
||||||
|
_, stderr, err := command.ExecCmdDir(repo.Path, "git", "pull")
|
||||||
|
if err != nil {
|
||||||
|
return concatenateError(err, stderr)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取tag列表
|
||||||
|
func (repo *Repository) GetTags() ([]string, error) {
|
||||||
|
stdout, stderr, err := command.ExecCmdDir(repo.Path, "git", "tag", "-l")
|
||||||
|
if err != nil {
|
||||||
|
return nil, concatenateError(err, stderr)
|
||||||
|
}
|
||||||
|
tags := strings.Split(stdout, "\n")
|
||||||
|
tags = tags[:len(tags)-1]
|
||||||
|
|
||||||
|
so := &SortTag{data: tags}
|
||||||
|
return so.Sort(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取两个版本之间的修改日志
|
||||||
|
func (repo *Repository) GetChangeLogs(startVer, endVer string) ([]string, error) {
|
||||||
|
// git log --pretty=format:"%cd %cn: %s" --date=iso v1.8.0...v1.9.0
|
||||||
|
stdout, stderr, err := command.ExecCmdDir(repo.Path, "git", "log", "--pretty=format:%cd %cn: %s", "--date=iso", startVer+"..."+endVer)
|
||||||
|
if err != nil {
|
||||||
|
return nil, concatenateError(err, stderr)
|
||||||
|
}
|
||||||
|
|
||||||
|
logs := strings.Split(stdout, "\n")
|
||||||
|
return logs, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取两个版本之间的差异文件列表
|
||||||
|
func (repo *Repository) GetChangeFiles(startVer, endVer string, onlyFile bool) ([]string, error) {
|
||||||
|
// git diff --name-status -b v1.8.0 v1.9.0
|
||||||
|
param := "--name-status"
|
||||||
|
if onlyFile {
|
||||||
|
param = "--name-only"
|
||||||
|
}
|
||||||
|
stdout, stderr, err := command.ExecCmdDir(repo.Path, "git", "diff", param, "-b", startVer, endVer)
|
||||||
|
if err != nil {
|
||||||
|
return nil, concatenateError(err, stderr)
|
||||||
|
}
|
||||||
|
lines := strings.Split(stdout, "\n")
|
||||||
|
return lines[:len(lines)-1], nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取两个版本间的新增或修改的文件数量
|
||||||
|
func (repo *Repository) GetDiffFileCount(startVer, endVer string) (int, error) {
|
||||||
|
cmd := "git diff --name-status -b " + startVer + " " + endVer + " |grep -v ^D |wc -l"
|
||||||
|
stdout, stderr, err := command.ExecCmdDir(repo.Path, "/bin/bash", "-c", cmd)
|
||||||
|
if err != nil {
|
||||||
|
return 0, concatenateError(err, stderr)
|
||||||
|
}
|
||||||
|
count, _ := strconv.Atoi(strings.TrimSpace(stdout))
|
||||||
|
return count, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 导出版本到tar包
|
||||||
|
func (repo *Repository) Export(startVer, endVer string, filename string) error {
|
||||||
|
// git archive --format=tar.gz $endVer $(git diff --name-status -b $beginVer $endVer |grep -v ^D |grep -v Upgrade/ |awk '{print $2}') -o $tmpFile
|
||||||
|
|
||||||
|
cmd := ""
|
||||||
|
if startVer == "" {
|
||||||
|
cmd = "git archive --format=tar " + endVer + " | gzip > " + filename
|
||||||
|
} else {
|
||||||
|
cmd = "git archive --format=tar " + endVer + " $(dgit diff --name-status -b " + startVer + " " + endVer + "|grep -v ^D |awk '{print $2}') | gzip > " + filename
|
||||||
|
}
|
||||||
|
|
||||||
|
_, stderr, err := command.ExecCmdDir(repo.Path, "/bin/bash", "-c", cmd)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return concatenateError(err, stderr)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func concatenateError(err error, stderr string) error {
|
||||||
|
if len(stderr) == 0 {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return fmt.Errorf("%v: %s", err, stderr)
|
||||||
|
}
|
22
internal/pkg/system/system.go
Normal file
22
internal/pkg/system/system.go
Normal file
|
@ -0,0 +1,22 @@
|
||||||
|
package system
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"os/user"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Bee System Params ...
|
||||||
|
var (
|
||||||
|
Usr, _ = user.Current()
|
||||||
|
BeegoHome = filepath.Join(Usr.HomeDir, "/.beego")
|
||||||
|
CurrentDir = getCurrentDirectory()
|
||||||
|
GoPath = os.Getenv("GOPATH")
|
||||||
|
)
|
||||||
|
|
||||||
|
func getCurrentDirectory() string {
|
||||||
|
if dir, err := os.Getwd(); err == nil {
|
||||||
|
return dir
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
113
internal/pkg/utils/dsn.go
Normal file
113
internal/pkg/utils/dsn.go
Normal file
|
@ -0,0 +1,113 @@
|
||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// DSN ...
|
||||||
|
type DSN struct {
|
||||||
|
User string // Username
|
||||||
|
Passwd string // Password (requires User)
|
||||||
|
Net string // Network type
|
||||||
|
Addr string // Network address (requires Net)
|
||||||
|
DBName string // Database name
|
||||||
|
Params map[string]string // Connection parameters
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
errInvalidDSNUnescaped = errors.New("invalid DSN: did you forget to escape a param value?")
|
||||||
|
errInvalidDSNAddr = errors.New("invalid DSN: network address not terminated (missing closing brace)")
|
||||||
|
errInvalidDSNNoSlash = errors.New("invalid DSN: missing the slash separating the database name")
|
||||||
|
)
|
||||||
|
|
||||||
|
// ParseDSN parses the DSN string to a Config
|
||||||
|
func ParseDSN(dsn string) (cfg *DSN, err error) {
|
||||||
|
// New config with some default values
|
||||||
|
cfg = new(DSN)
|
||||||
|
|
||||||
|
// [user[:password]@][net[(addr)]]/dbname[?param1=value1¶mN=valueN]
|
||||||
|
// Find the last '/' (since the password or the net addr might contain a '/')
|
||||||
|
foundSlash := false
|
||||||
|
for i := len(dsn) - 1; i >= 0; i-- {
|
||||||
|
if dsn[i] == '/' {
|
||||||
|
foundSlash = true
|
||||||
|
var j, k int
|
||||||
|
|
||||||
|
// left part is empty if i <= 0
|
||||||
|
if i > 0 {
|
||||||
|
// [username[:password]@][protocol[(address)]]
|
||||||
|
// Find the last '@' in dsn[:i]
|
||||||
|
for j = i; j >= 0; j-- {
|
||||||
|
if dsn[j] == '@' {
|
||||||
|
// username[:password]
|
||||||
|
// Find the first ':' in dsn[:j]
|
||||||
|
for k = 0; k < j; k++ {
|
||||||
|
if dsn[k] == ':' {
|
||||||
|
cfg.Passwd = dsn[k+1 : j]
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
cfg.User = dsn[:k]
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// [protocol[(address)]]
|
||||||
|
// Find the first '(' in dsn[j+1:i]
|
||||||
|
for k = j + 1; k < i; k++ {
|
||||||
|
if dsn[k] == '(' {
|
||||||
|
// dsn[i-1] must be == ')' if an address is specified
|
||||||
|
if dsn[i-1] != ')' {
|
||||||
|
if strings.ContainsRune(dsn[k+1:i], ')') {
|
||||||
|
return nil, errInvalidDSNUnescaped
|
||||||
|
}
|
||||||
|
return nil, errInvalidDSNAddr
|
||||||
|
}
|
||||||
|
cfg.Addr = dsn[k+1 : i-1]
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
cfg.Net = dsn[j+1 : k]
|
||||||
|
}
|
||||||
|
|
||||||
|
// dbname[?param1=value1&...¶mN=valueN]
|
||||||
|
// Find the first '?' in dsn[i+1:]
|
||||||
|
for j = i + 1; j < len(dsn); j++ {
|
||||||
|
if dsn[j] == '?' {
|
||||||
|
if err = parseDSNParams(cfg, dsn[j+1:]); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
cfg.DBName = dsn[i+1 : j]
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !foundSlash && len(dsn) > 0 {
|
||||||
|
return nil, errInvalidDSNNoSlash
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseDSNParams(cfg *DSN, params string) (err error) {
|
||||||
|
for _, v := range strings.Split(params, "&") {
|
||||||
|
param := strings.SplitN(v, "=", 2)
|
||||||
|
if len(param) != 2 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// lazy init
|
||||||
|
if cfg.Params == nil {
|
||||||
|
cfg.Params = make(map[string]string)
|
||||||
|
}
|
||||||
|
value := param[1]
|
||||||
|
if cfg.Params[param[0]], err = url.QueryUnescape(value); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
38
internal/pkg/utils/file.go
Normal file
38
internal/pkg/utils/file.go
Normal file
|
@ -0,0 +1,38 @@
|
||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Mkdir ...
|
||||||
|
func Mkdir(dir string) bool {
|
||||||
|
if dir == "" {
|
||||||
|
beeLogger.Log.Fatalf("The directory is empty")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
err := os.MkdirAll(dir, 0755)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not create the directory: %s", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
beeLogger.Log.Infof("Create %s Success!", dir)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsDir ...
|
||||||
|
func IsDir(dir string) bool {
|
||||||
|
f, e := os.Stat(dir)
|
||||||
|
if e != nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return f.IsDir()
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsExist returns whether a file or directory exists.
|
||||||
|
func IsExist(path string) bool {
|
||||||
|
_, err := os.Stat(path)
|
||||||
|
return err == nil || os.IsExist(err)
|
||||||
|
}
|
|
@ -12,7 +12,7 @@
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
|
|
||||||
package main
|
package colors
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
@ -53,8 +53,8 @@ func NewModeColorWriter(w io.Writer, mode outputMode) io.Writer {
|
||||||
return w
|
return w
|
||||||
}
|
}
|
||||||
|
|
||||||
func bold(message string) string {
|
func Bold(message string) string {
|
||||||
return fmt.Sprintf("\x1b[1m%s\x1b[21m", message)
|
return fmt.Sprintf("\x1b[1m%s\x1b[0m", message)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Black returns a black string
|
// Black returns a black string
|
||||||
|
@ -102,47 +102,47 @@ func Magenta(message string) string {
|
||||||
return fmt.Sprintf("\x1b[35m%s\x1b[0m", message)
|
return fmt.Sprintf("\x1b[35m%s\x1b[0m", message)
|
||||||
}
|
}
|
||||||
|
|
||||||
// BlackBold returns a black bold string
|
// BlackBold returns a black Bold string
|
||||||
func BlackBold(message string) string {
|
func BlackBold(message string) string {
|
||||||
return fmt.Sprintf("\x1b[30m%s\x1b[0m", bold(message))
|
return fmt.Sprintf("\x1b[30m%s\x1b[0m", Bold(message))
|
||||||
}
|
}
|
||||||
|
|
||||||
// WhiteBold returns a white bold string
|
// WhiteBold returns a white Bold string
|
||||||
func WhiteBold(message string) string {
|
func WhiteBold(message string) string {
|
||||||
return fmt.Sprintf("\x1b[37m%s\x1b[0m", bold(message))
|
return fmt.Sprintf("\x1b[37m%s\x1b[0m", Bold(message))
|
||||||
}
|
}
|
||||||
|
|
||||||
// CyanBold returns a cyan bold string
|
// CyanBold returns a cyan Bold string
|
||||||
func CyanBold(message string) string {
|
func CyanBold(message string) string {
|
||||||
return fmt.Sprintf("\x1b[36m%s\x1b[0m", bold(message))
|
return fmt.Sprintf("\x1b[36m%s\x1b[0m", Bold(message))
|
||||||
}
|
}
|
||||||
|
|
||||||
// BlueBold returns a blue bold string
|
// BlueBold returns a blue Bold string
|
||||||
func BlueBold(message string) string {
|
func BlueBold(message string) string {
|
||||||
return fmt.Sprintf("\x1b[34m%s\x1b[0m", bold(message))
|
return fmt.Sprintf("\x1b[34m%s\x1b[0m", Bold(message))
|
||||||
}
|
}
|
||||||
|
|
||||||
// RedBold returns a red bold string
|
// RedBold returns a red Bold string
|
||||||
func RedBold(message string) string {
|
func RedBold(message string) string {
|
||||||
return fmt.Sprintf("\x1b[31m%s\x1b[0m", bold(message))
|
return fmt.Sprintf("\x1b[31m%s\x1b[0m", Bold(message))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GreenBold returns a green bold string
|
// GreenBold returns a green Bold string
|
||||||
func GreenBold(message string) string {
|
func GreenBold(message string) string {
|
||||||
return fmt.Sprintf("\x1b[32m%s\x1b[0m", bold(message))
|
return fmt.Sprintf("\x1b[32m%s\x1b[0m", Bold(message))
|
||||||
}
|
}
|
||||||
|
|
||||||
// YellowBold returns a yellow bold string
|
// YellowBold returns a yellow Bold string
|
||||||
func YellowBold(message string) string {
|
func YellowBold(message string) string {
|
||||||
return fmt.Sprintf("\x1b[33m%s\x1b[0m", bold(message))
|
return fmt.Sprintf("\x1b[33m%s\x1b[0m", Bold(message))
|
||||||
}
|
}
|
||||||
|
|
||||||
// GrayBold returns a gray bold string
|
// GrayBold returns a gray Bold string
|
||||||
func GrayBold(message string) string {
|
func GrayBold(message string) string {
|
||||||
return fmt.Sprintf("\x1b[37m%s\x1b[0m", bold(message))
|
return fmt.Sprintf("\x1b[37m%s\x1b[0m", Bold(message))
|
||||||
}
|
}
|
||||||
|
|
||||||
// MagentaBold returns a magenta bold string
|
// MagentaBold returns a magenta Bold string
|
||||||
func MagentaBold(message string) string {
|
func MagentaBold(message string) string {
|
||||||
return fmt.Sprintf("\x1b[35m%s\x1b[0m", bold(message))
|
return fmt.Sprintf("\x1b[35m%s\x1b[0m", Bold(message))
|
||||||
}
|
}
|
|
@ -14,7 +14,7 @@
|
||||||
|
|
||||||
// +build !windows
|
// +build !windows
|
||||||
|
|
||||||
package main
|
package colors
|
||||||
|
|
||||||
import "io"
|
import "io"
|
||||||
|
|
|
@ -14,7 +14,7 @@
|
||||||
|
|
||||||
// +build windows
|
// +build windows
|
||||||
|
|
||||||
package main
|
package colors
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
@ -360,7 +360,7 @@ func isParameterChar(b byte) bool {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cw *colorWriter) Write(p []byte) (int, error) {
|
func (cw *colorWriter) Write(p []byte) (int, error) {
|
||||||
r, nw, first, last := 0, 0, 0, 0
|
var r, nw, first, last int
|
||||||
if cw.mode != DiscardNonColorEscSeq {
|
if cw.mode != DiscardNonColorEscSeq {
|
||||||
cw.state = outsideCsiCode
|
cw.state = outsideCsiCode
|
||||||
cw.resetBuffer()
|
cw.resetBuffer()
|
||||||
|
@ -378,7 +378,6 @@ func (cw *colorWriter) Write(p []byte) (int, error) {
|
||||||
switch ch {
|
switch ch {
|
||||||
case firstCsiChar:
|
case firstCsiChar:
|
||||||
cw.paramStartBuf.WriteByte(ch)
|
cw.paramStartBuf.WriteByte(ch)
|
||||||
break
|
|
||||||
case secondeCsiChar:
|
case secondeCsiChar:
|
||||||
cw.paramStartBuf.WriteByte(ch)
|
cw.paramStartBuf.WriteByte(ch)
|
||||||
cw.state = secondCsiCode
|
cw.state = secondCsiCode
|
|
@ -11,7 +11,7 @@
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
// License for the specific language governing permissions and limitations
|
// License for the specific language governing permissions and limitations
|
||||||
// under the License.
|
// under the License.
|
||||||
package main
|
package beeLogger
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
@ -22,19 +22,22 @@ import (
|
||||||
"sync"
|
"sync"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
"text/template"
|
"text/template"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
)
|
)
|
||||||
|
|
||||||
var errInvalidLogLevel = errors.New("logger: invalid log level")
|
var errInvalidLogLevel = errors.New("logger: invalid log level")
|
||||||
|
|
||||||
const (
|
const (
|
||||||
levelCritical = iota
|
levelDebug = iota
|
||||||
levelFatal
|
|
||||||
levelSuccess
|
|
||||||
levelHint
|
|
||||||
levelDebug
|
|
||||||
levelInfo
|
|
||||||
levelWarn
|
|
||||||
levelError
|
levelError
|
||||||
|
levelFatal
|
||||||
|
levelCritical
|
||||||
|
levelSuccess
|
||||||
|
levelWarn
|
||||||
|
levelInfo
|
||||||
|
levelHint
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
|
@ -42,6 +45,9 @@ var (
|
||||||
instance *BeeLogger
|
instance *BeeLogger
|
||||||
once sync.Once
|
once sync.Once
|
||||||
)
|
)
|
||||||
|
var debugMode = os.Getenv("DEBUG_ENABLED") == "1"
|
||||||
|
|
||||||
|
var logLevel = levelInfo
|
||||||
|
|
||||||
// BeeLogger logs logging records to the specified io.Writer
|
// BeeLogger logs logging records to the specified io.Writer
|
||||||
type BeeLogger struct {
|
type BeeLogger struct {
|
||||||
|
@ -59,6 +65,8 @@ type LogRecord struct {
|
||||||
LineNo int
|
LineNo int
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var Log = GetBeeLogger(os.Stdout)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
logRecordTemplate *template.Template
|
logRecordTemplate *template.Template
|
||||||
debugLogRecordTemplate *template.Template
|
debugLogRecordTemplate *template.Template
|
||||||
|
@ -80,11 +88,15 @@ func GetBeeLogger(w io.Writer) *BeeLogger {
|
||||||
"EndLine": EndLine,
|
"EndLine": EndLine,
|
||||||
}
|
}
|
||||||
logRecordTemplate, err = template.New("simpleLogFormat").Funcs(funcs).Parse(simpleLogFormat)
|
logRecordTemplate, err = template.New("simpleLogFormat").Funcs(funcs).Parse(simpleLogFormat)
|
||||||
MustCheck(err)
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
debugLogRecordTemplate, err = template.New("debugLogFormat").Funcs(funcs).Parse(debugLogFormat)
|
debugLogRecordTemplate, err = template.New("debugLogFormat").Funcs(funcs).Parse(debugLogFormat)
|
||||||
MustCheck(err)
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
instance = &BeeLogger{output: NewColorWriter(w)}
|
instance = &BeeLogger{output: colors.NewColorWriter(w)}
|
||||||
})
|
})
|
||||||
return instance
|
return instance
|
||||||
}
|
}
|
||||||
|
@ -93,7 +105,17 @@ func GetBeeLogger(w io.Writer) *BeeLogger {
|
||||||
func (l *BeeLogger) SetOutput(w io.Writer) {
|
func (l *BeeLogger) SetOutput(w io.Writer) {
|
||||||
l.mu.Lock()
|
l.mu.Lock()
|
||||||
defer l.mu.Unlock()
|
defer l.mu.Unlock()
|
||||||
l.output = NewColorWriter(w)
|
l.output = colors.NewColorWriter(w)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Now returns the current local time in the specified layout
|
||||||
|
func Now(layout string) string {
|
||||||
|
return time.Now().Format(layout)
|
||||||
|
}
|
||||||
|
|
||||||
|
// EndLine returns the a newline escape character
|
||||||
|
func EndLine() string {
|
||||||
|
return "\n"
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *BeeLogger) getLevelTag(level int) string {
|
func (l *BeeLogger) getLevelTag(level int) string {
|
||||||
|
@ -122,21 +144,21 @@ func (l *BeeLogger) getLevelTag(level int) string {
|
||||||
func (l *BeeLogger) getColorLevel(level int) string {
|
func (l *BeeLogger) getColorLevel(level int) string {
|
||||||
switch level {
|
switch level {
|
||||||
case levelCritical:
|
case levelCritical:
|
||||||
return RedBold(l.getLevelTag(level))
|
return colors.RedBold(l.getLevelTag(level))
|
||||||
case levelFatal:
|
case levelFatal:
|
||||||
return RedBold(l.getLevelTag(level))
|
return colors.RedBold(l.getLevelTag(level))
|
||||||
case levelInfo:
|
case levelInfo:
|
||||||
return BlueBold(l.getLevelTag(level))
|
return colors.BlueBold(l.getLevelTag(level))
|
||||||
case levelHint:
|
case levelHint:
|
||||||
return CyanBold(l.getLevelTag(level))
|
return colors.CyanBold(l.getLevelTag(level))
|
||||||
case levelDebug:
|
case levelDebug:
|
||||||
return YellowBold(l.getLevelTag(level))
|
return colors.YellowBold(l.getLevelTag(level))
|
||||||
case levelError:
|
case levelError:
|
||||||
return RedBold(l.getLevelTag(level))
|
return colors.RedBold(l.getLevelTag(level))
|
||||||
case levelWarn:
|
case levelWarn:
|
||||||
return YellowBold(l.getLevelTag(level))
|
return colors.YellowBold(l.getLevelTag(level))
|
||||||
case levelSuccess:
|
case levelSuccess:
|
||||||
return GreenBold(l.getLevelTag(level))
|
return colors.GreenBold(l.getLevelTag(level))
|
||||||
default:
|
default:
|
||||||
panic(errInvalidLogLevel)
|
panic(errInvalidLogLevel)
|
||||||
}
|
}
|
||||||
|
@ -145,6 +167,9 @@ func (l *BeeLogger) getColorLevel(level int) string {
|
||||||
// mustLog logs the message according to the specified level and arguments.
|
// mustLog logs the message according to the specified level and arguments.
|
||||||
// It panics in case of an error.
|
// It panics in case of an error.
|
||||||
func (l *BeeLogger) mustLog(level int, message string, args ...interface{}) {
|
func (l *BeeLogger) mustLog(level int, message string, args ...interface{}) {
|
||||||
|
if level > logLevel {
|
||||||
|
return
|
||||||
|
}
|
||||||
// Acquire the lock
|
// Acquire the lock
|
||||||
l.mu.Lock()
|
l.mu.Lock()
|
||||||
defer l.mu.Unlock()
|
defer l.mu.Unlock()
|
||||||
|
@ -157,13 +182,15 @@ func (l *BeeLogger) mustLog(level int, message string, args ...interface{}) {
|
||||||
}
|
}
|
||||||
|
|
||||||
err := logRecordTemplate.Execute(l.output, record)
|
err := logRecordTemplate.Execute(l.output, record)
|
||||||
MustCheck(err)
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// mustLogDebug logs a debug message only if debug mode
|
// mustLogDebug logs a debug message only if debug mode
|
||||||
// is enabled. i.e. DEBUG_ENABLED="1"
|
// is enabled. i.e. DEBUG_ENABLED="1"
|
||||||
func (l *BeeLogger) mustLogDebug(message string, file string, line int, args ...interface{}) {
|
func (l *BeeLogger) mustLogDebug(message string, file string, line int, args ...interface{}) {
|
||||||
if !IsDebugEnabled() {
|
if !debugMode {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -179,7 +206,9 @@ func (l *BeeLogger) mustLogDebug(message string, file string, line int, args ...
|
||||||
Filename: filepath.Base(file),
|
Filename: filepath.Base(file),
|
||||||
}
|
}
|
||||||
err := debugLogRecordTemplate.Execute(l.output, record)
|
err := debugLogRecordTemplate.Execute(l.output, record)
|
||||||
MustCheck(err)
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Debug outputs a debug log message
|
// Debug outputs a debug log message
|
67
main.go
Normal file
67
main.go
Normal file
|
@ -0,0 +1,67 @@
|
||||||
|
// Copyright 2013 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/cmd"
|
||||||
|
"github.com/beego/bee/v2/cmd/commands"
|
||||||
|
"github.com/beego/bee/v2/config"
|
||||||
|
"github.com/beego/bee/v2/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
utils.NoticeUpdateBee()
|
||||||
|
flag.Usage = cmd.Usage
|
||||||
|
flag.Parse()
|
||||||
|
log.SetFlags(0)
|
||||||
|
|
||||||
|
args := flag.Args()
|
||||||
|
|
||||||
|
if len(args) < 1 {
|
||||||
|
cmd.Usage()
|
||||||
|
os.Exit(2)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if args[0] == "help" {
|
||||||
|
cmd.Help(args[1:])
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range commands.AvailableCommands {
|
||||||
|
if c.Name() == args[0] && c.Run != nil {
|
||||||
|
c.Flag.Usage = func() { c.Usage() }
|
||||||
|
if c.CustomFlags {
|
||||||
|
args = args[1:]
|
||||||
|
} else {
|
||||||
|
c.Flag.Parse(args[1:])
|
||||||
|
args = c.Flag.Args()
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.PreRun != nil {
|
||||||
|
c.PreRun(c, args)
|
||||||
|
}
|
||||||
|
|
||||||
|
config.LoadConfig()
|
||||||
|
os.Exit(c.Run(c, args))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
utils.PrintErrorAndExit("Unknown subcommand", cmd.ErrorTemplate)
|
||||||
|
}
|
215
run.go
215
run.go
|
@ -1,215 +0,0 @@
|
||||||
// Copyright 2013 bee authors
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io/ioutil"
|
|
||||||
"os"
|
|
||||||
path "path/filepath"
|
|
||||||
"runtime"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
var cmdRun = &Command{
|
|
||||||
UsageLine: "run [appname] [watchall] [-main=*.go] [-downdoc=true] [-gendoc=true] [-vendor=true] [-e=folderToExclude] [-tags=goBuildTags] [-runmode=BEEGO_RUNMODE]",
|
|
||||||
Short: "Run the application by starting a local development server",
|
|
||||||
Long: `
|
|
||||||
Run command will supervise the filesystem of the application for any changes, and recompile/restart it.
|
|
||||||
|
|
||||||
`,
|
|
||||||
PreRun: func(cmd *Command, args []string) { ShowShortVersionBanner() },
|
|
||||||
Run: runApp,
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
mainFiles ListOpts
|
|
||||||
downdoc docValue
|
|
||||||
gendoc docValue
|
|
||||||
// The flags list of the paths excluded from watching
|
|
||||||
excludedPaths strFlags
|
|
||||||
// Pass through to -tags arg of "go build"
|
|
||||||
buildTags string
|
|
||||||
// Application path
|
|
||||||
currpath string
|
|
||||||
// Application name
|
|
||||||
appname string
|
|
||||||
// Channel to signal an Exit
|
|
||||||
exit chan bool
|
|
||||||
// Flag to watch the vendor folder
|
|
||||||
vendorWatch bool
|
|
||||||
// Current user workspace
|
|
||||||
currentGoPath string
|
|
||||||
// Current runmode
|
|
||||||
runmode string
|
|
||||||
)
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
cmdRun.Flag.Var(&mainFiles, "main", "Specify main go files.")
|
|
||||||
cmdRun.Flag.Var(&gendoc, "gendoc", "Enable auto-generate the docs.")
|
|
||||||
cmdRun.Flag.Var(&downdoc, "downdoc", "Enable auto-download of the swagger file if it does not exist.")
|
|
||||||
cmdRun.Flag.Var(&excludedPaths, "e", "List of paths to exclude.")
|
|
||||||
cmdRun.Flag.BoolVar(&vendorWatch, "vendor", false, "Enable watch vendor folder.")
|
|
||||||
cmdRun.Flag.StringVar(&buildTags, "tags", "", "Set the build tags. See: https://golang.org/pkg/go/build/")
|
|
||||||
cmdRun.Flag.StringVar(&runmode, "runmode", "", "Set the Beego run mode.")
|
|
||||||
exit = make(chan bool)
|
|
||||||
}
|
|
||||||
|
|
||||||
func runApp(cmd *Command, args []string) int {
|
|
||||||
if len(args) == 0 || args[0] == "watchall" {
|
|
||||||
currpath, _ = os.Getwd()
|
|
||||||
|
|
||||||
if found, _gopath, _ := SearchGOPATHs(currpath); found {
|
|
||||||
appname = path.Base(currpath)
|
|
||||||
currentGoPath = _gopath
|
|
||||||
} else {
|
|
||||||
logger.Fatalf("No application '%s' found in your GOPATH", currpath)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// Check if passed Bee application path/name exists in the GOPATH(s)
|
|
||||||
if found, _gopath, _path := SearchGOPATHs(args[0]); found {
|
|
||||||
currpath = _path
|
|
||||||
currentGoPath = _gopath
|
|
||||||
appname = path.Base(currpath)
|
|
||||||
} else {
|
|
||||||
logger.Fatalf("No application '%s' found in your GOPATH", args[0])
|
|
||||||
}
|
|
||||||
|
|
||||||
if strings.HasSuffix(appname, ".go") && isExist(currpath) {
|
|
||||||
logger.Warnf("The appname is in conflict with file's current path. Do you want to build appname as '%s'", appname)
|
|
||||||
logger.Info("Do you want to overwrite it? [yes|no] ")
|
|
||||||
if !askForConfirmation() {
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
logger.Infof("Using '%s' as 'appname'", appname)
|
|
||||||
|
|
||||||
logger.Debugf("Current path: %s", __FILE__(), __LINE__(), currpath)
|
|
||||||
|
|
||||||
if runmode == "prod" || runmode == "dev" {
|
|
||||||
os.Setenv("BEEGO_RUNMODE", runmode)
|
|
||||||
logger.Infof("Using '%s' as 'runmode'", os.Getenv("BEEGO_RUNMODE"))
|
|
||||||
} else if runmode != "" {
|
|
||||||
os.Setenv("BEEGO_RUNMODE", runmode)
|
|
||||||
logger.Warnf("Using '%s' as 'runmode'", os.Getenv("BEEGO_RUNMODE"))
|
|
||||||
} else if os.Getenv("BEEGO_RUNMODE") != "" {
|
|
||||||
logger.Warnf("Using '%s' as 'runmode'", os.Getenv("BEEGO_RUNMODE"))
|
|
||||||
}
|
|
||||||
|
|
||||||
err := loadConfig()
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("Failed to load configuration: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var paths []string
|
|
||||||
readAppDirectories(currpath, &paths)
|
|
||||||
|
|
||||||
// Because monitor files has some issues, we watch current directory
|
|
||||||
// and ignore non-go files.
|
|
||||||
for _, p := range conf.DirStruct.Others {
|
|
||||||
paths = append(paths, strings.Replace(p, "$GOPATH", currentGoPath, -1))
|
|
||||||
}
|
|
||||||
|
|
||||||
files := []string{}
|
|
||||||
for _, arg := range mainFiles {
|
|
||||||
if len(arg) > 0 {
|
|
||||||
files = append(files, arg)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if downdoc == "true" {
|
|
||||||
if _, err := os.Stat(path.Join(currpath, "swagger", "index.html")); err != nil {
|
|
||||||
if os.IsNotExist(err) {
|
|
||||||
downloadFromURL(swaggerlink, "swagger.zip")
|
|
||||||
unzipAndDelete("swagger.zip")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if gendoc == "true" {
|
|
||||||
NewWatcher(paths, files, true)
|
|
||||||
AutoBuild(files, true)
|
|
||||||
} else {
|
|
||||||
NewWatcher(paths, files, false)
|
|
||||||
AutoBuild(files, false)
|
|
||||||
}
|
|
||||||
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-exit:
|
|
||||||
runtime.Goexit()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func readAppDirectories(directory string, paths *[]string) {
|
|
||||||
fileInfos, err := ioutil.ReadDir(directory)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
useDirectory := false
|
|
||||||
for _, fileInfo := range fileInfos {
|
|
||||||
if strings.HasSuffix(fileInfo.Name(), "docs") {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if strings.HasSuffix(fileInfo.Name(), "swagger") {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if !vendorWatch && strings.HasSuffix(fileInfo.Name(), "vendor") {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if isExcluded(path.Join(directory, fileInfo.Name())) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if fileInfo.IsDir() == true && fileInfo.Name()[0] != '.' {
|
|
||||||
readAppDirectories(directory+"/"+fileInfo.Name(), paths)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if useDirectory == true {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if path.Ext(fileInfo.Name()) == ".go" {
|
|
||||||
*paths = append(*paths, directory)
|
|
||||||
useDirectory = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// If a file is excluded
|
|
||||||
func isExcluded(filePath string) bool {
|
|
||||||
for _, p := range excludedPaths {
|
|
||||||
absP, err := path.Abs(p)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Cannot get absolute path of '%s'", p)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
absFilePath, err := path.Abs(filePath)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Cannot get absolute path of '%s'", filePath)
|
|
||||||
break
|
|
||||||
}
|
|
||||||
if strings.HasPrefix(absFilePath, absP) {
|
|
||||||
logger.Infof("'%s' is not being watched", filePath)
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
155
rundocs.go
155
rundocs.go
|
@ -1,155 +0,0 @@
|
||||||
// Copyright 2013 bee authors
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"archive/zip"
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"net/http"
|
|
||||||
"os"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
var cmdRundocs = &Command{
|
|
||||||
UsageLine: "rundocs [-isDownload=true] [-docport=8888]",
|
|
||||||
Short: "rundocs will run the docs server,default is 8089",
|
|
||||||
Long: `
|
|
||||||
-d meaning will download the docs file from github
|
|
||||||
-p meaning server the Server on which port, default is 8089
|
|
||||||
|
|
||||||
`,
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
swaggerVersion = "2"
|
|
||||||
swaggerlink = "https://github.com/beego/swagger/archive/v" + swaggerVersion + ".zip"
|
|
||||||
)
|
|
||||||
|
|
||||||
type docValue string
|
|
||||||
|
|
||||||
func (d *docValue) String() string {
|
|
||||||
return fmt.Sprint(*d)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *docValue) Set(value string) error {
|
|
||||||
*d = docValue(value)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
var isDownload docValue
|
|
||||||
var docport docValue
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
cmdRundocs.Run = runDocs
|
|
||||||
cmdRundocs.PreRun = func(cmd *Command, args []string) { ShowShortVersionBanner() }
|
|
||||||
cmdRundocs.Flag.Var(&isDownload, "isDownload", "weather download the Swagger Docs")
|
|
||||||
cmdRundocs.Flag.Var(&docport, "docport", "doc server port")
|
|
||||||
}
|
|
||||||
|
|
||||||
func runDocs(cmd *Command, args []string) int {
|
|
||||||
if isDownload == "true" {
|
|
||||||
downloadFromURL(swaggerlink, "swagger.zip")
|
|
||||||
err := unzipAndDelete("swagger.zip")
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Error while unzipping 'swagger.zip' file: %s", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if docport == "" {
|
|
||||||
docport = "8089"
|
|
||||||
}
|
|
||||||
if _, err := os.Stat("swagger"); err != nil && os.IsNotExist(err) {
|
|
||||||
logger.Fatal("No Swagger dist found. Run: bee rundocs -isDownload=true")
|
|
||||||
}
|
|
||||||
|
|
||||||
logger.Infof("Starting the docs server on: http://127.0.0.1:%s", docport)
|
|
||||||
|
|
||||||
err := http.ListenAndServe(":"+string(docport), http.FileServer(http.Dir("swagger")))
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("%s", err)
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func downloadFromURL(url, fileName string) {
|
|
||||||
var down bool
|
|
||||||
if fd, err := os.Stat(fileName); err != nil && os.IsNotExist(err) {
|
|
||||||
down = true
|
|
||||||
} else if fd.Size() == int64(0) {
|
|
||||||
down = true
|
|
||||||
} else {
|
|
||||||
logger.Infof("'%s' already exists", fileName)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if down {
|
|
||||||
logger.Infof("Downloading '%s' to '%s'...", url, fileName)
|
|
||||||
output, err := os.Create(fileName)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Error while creating '%s': %s", fileName, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
defer output.Close()
|
|
||||||
|
|
||||||
response, err := http.Get(url)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Error while downloading '%s': %s", url, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
defer response.Body.Close()
|
|
||||||
|
|
||||||
n, err := io.Copy(output, response.Body)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorf("Error while downloading '%s': %s", url, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
logger.Successf("%d bytes downloaded!", n)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func unzipAndDelete(src string) error {
|
|
||||||
logger.Infof("Unzipping '%s'...", src)
|
|
||||||
r, err := zip.OpenReader(src)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer r.Close()
|
|
||||||
|
|
||||||
rp := strings.NewReplacer("swagger-"+swaggerVersion, "swagger")
|
|
||||||
for _, f := range r.File {
|
|
||||||
rc, err := f.Open()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer rc.Close()
|
|
||||||
|
|
||||||
fname := rp.Replace(f.Name)
|
|
||||||
if f.FileInfo().IsDir() {
|
|
||||||
os.MkdirAll(fname, f.Mode())
|
|
||||||
} else {
|
|
||||||
f, err := os.OpenFile(
|
|
||||||
fname, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer f.Close()
|
|
||||||
|
|
||||||
_, err = io.Copy(f, rc)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
logger.Successf("Done! Deleting '%s'...", src)
|
|
||||||
return os.RemoveAll(src)
|
|
||||||
}
|
|
7
scripts/quickstart.sh
Normal file
7
scripts/quickstart.sh
Normal file
|
@ -0,0 +1,7 @@
|
||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
export GO111MODULE=on
|
||||||
|
go get github.com/beego/bee/v2@latest
|
||||||
|
bee new hello
|
||||||
|
cd hello
|
||||||
|
bee run
|
1
staticcheck.ignore
Normal file
1
staticcheck.ignore
Normal file
|
@ -0,0 +1 @@
|
||||||
|
github.com/beego/bee/v2/generate/swaggergen/*.go:SA1024
|
102
test.go
102
test.go
|
@ -1,102 +0,0 @@
|
||||||
// Copyright 2013 bee authors
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"os"
|
|
||||||
"os/exec"
|
|
||||||
path "path/filepath"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
_ "github.com/smartystreets/goconvey/convey"
|
|
||||||
)
|
|
||||||
|
|
||||||
var cmdTest = &Command{
|
|
||||||
UsageLine: "test [appname]",
|
|
||||||
Short: "test the app",
|
|
||||||
Long: ``,
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
cmdTest.Run = testApp
|
|
||||||
cmdTest.PreRun = func(cmd *Command, args []string) { ShowShortVersionBanner() }
|
|
||||||
}
|
|
||||||
|
|
||||||
func safePathAppend(arr []string, paths ...string) []string {
|
|
||||||
for _, path := range paths {
|
|
||||||
if pathExists(path) {
|
|
||||||
arr = append(arr, path)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return arr
|
|
||||||
}
|
|
||||||
|
|
||||||
func pathExists(path string) bool {
|
|
||||||
_, err := os.Stat(path)
|
|
||||||
return err == nil || os.IsExist(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var started = make(chan bool)
|
|
||||||
|
|
||||||
func testApp(cmd *Command, args []string) int {
|
|
||||||
if len(args) != 1 {
|
|
||||||
logger.Fatalf("Failed to start: %s", "argument 'appname' is missing")
|
|
||||||
}
|
|
||||||
|
|
||||||
currpath, _ := os.Getwd()
|
|
||||||
|
|
||||||
logger.Debugf("Current path: %s", __FILE__(), __LINE__(), currpath)
|
|
||||||
|
|
||||||
err := loadConfig()
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("Failed to load configuration: %s", err)
|
|
||||||
}
|
|
||||||
var paths []string
|
|
||||||
readAppDirectories(currpath, &paths)
|
|
||||||
|
|
||||||
NewWatcher(paths, nil, false)
|
|
||||||
appname = args[0]
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-started:
|
|
||||||
runTest()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func runTest() {
|
|
||||||
logger.Info("Start testing...")
|
|
||||||
time.Sleep(time.Second * 1)
|
|
||||||
crupwd, _ := os.Getwd()
|
|
||||||
testDir := path.Join(crupwd, "tests")
|
|
||||||
if pathExists(testDir) {
|
|
||||||
os.Chdir(testDir)
|
|
||||||
}
|
|
||||||
|
|
||||||
var err error
|
|
||||||
icmd := exec.Command("go", "test")
|
|
||||||
icmd.Stdout = os.Stdout
|
|
||||||
icmd.Stderr = os.Stderr
|
|
||||||
logger.Info("============== Test Begin ===================")
|
|
||||||
err = icmd.Run()
|
|
||||||
logger.Info("============== Test End =====================")
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
logger.Error("============== Test failed ===================")
|
|
||||||
logger.Errorf("Cause: %s", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
logger.Success("Test Completed")
|
|
||||||
}
|
|
28
testdata/router/router.go
vendored
28
testdata/router/router.go
vendored
|
@ -1,28 +0,0 @@
|
||||||
package router
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/astaxie/beego"
|
|
||||||
)
|
|
||||||
|
|
||||||
type Router struct {
|
|
||||||
beego.Controller
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Router) Get() {
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Router) Post() {
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
type Controller struct {
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Controller) Put() {
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Controller) Delete() {
|
|
||||||
|
|
||||||
}
|
|
309
util.go
309
util.go
|
@ -1,309 +0,0 @@
|
||||||
// Copyright 2013 bee authors
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
||||||
// not use this file except in compliance with the License. You may obtain
|
|
||||||
// a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
||||||
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
||||||
// License for the specific language governing permissions and limitations
|
|
||||||
// under the License.
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"fmt"
|
|
||||||
"io/ioutil"
|
|
||||||
"os"
|
|
||||||
"os/exec"
|
|
||||||
"path"
|
|
||||||
"path/filepath"
|
|
||||||
"regexp"
|
|
||||||
"runtime"
|
|
||||||
"strings"
|
|
||||||
"text/template"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Go is a basic promise implementation: it wraps calls a function in a goroutine
|
|
||||||
// and returns a channel which will later return the function's return value.
|
|
||||||
func Go(f func() error) chan error {
|
|
||||||
ch := make(chan error)
|
|
||||||
go func() {
|
|
||||||
ch <- f()
|
|
||||||
}()
|
|
||||||
return ch
|
|
||||||
}
|
|
||||||
|
|
||||||
// Now returns the current local time in the specified layout
|
|
||||||
func Now(layout string) string {
|
|
||||||
return time.Now().Format(layout)
|
|
||||||
}
|
|
||||||
|
|
||||||
// EndLine returns the a newline escape character
|
|
||||||
func EndLine() string {
|
|
||||||
return "\n"
|
|
||||||
}
|
|
||||||
|
|
||||||
// IsExist returns whether a file or directory exists.
|
|
||||||
func isExist(path string) bool {
|
|
||||||
_, err := os.Stat(path)
|
|
||||||
return err == nil || os.IsExist(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetGOPATHs returns all paths in GOPATH variable.
|
|
||||||
func GetGOPATHs() []string {
|
|
||||||
gopath := os.Getenv("GOPATH")
|
|
||||||
var paths []string
|
|
||||||
if runtime.GOOS == "windows" {
|
|
||||||
gopath = strings.Replace(gopath, "\\", "/", -1)
|
|
||||||
paths = strings.Split(gopath, ";")
|
|
||||||
} else {
|
|
||||||
paths = strings.Split(gopath, ":")
|
|
||||||
}
|
|
||||||
return paths
|
|
||||||
}
|
|
||||||
|
|
||||||
// IsBeegoProject checks whether the current path is a Beego application or not
|
|
||||||
func IsBeegoProject(thePath string) bool {
|
|
||||||
mainFiles := []string{}
|
|
||||||
hasBeegoRegex := regexp.MustCompile(`(?s)package main.*?import.*?\(.*?github.com/astaxie/beego".*?\).*func main()`)
|
|
||||||
c := make(chan error)
|
|
||||||
// Walk the application path tree to look for main files.
|
|
||||||
// Main files must satisfy the 'hasBeegoRegex' regular expression.
|
|
||||||
go func() {
|
|
||||||
filepath.Walk(thePath, func(fpath string, f os.FileInfo, err error) error {
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
// Skip sub-directories
|
|
||||||
if !f.IsDir() {
|
|
||||||
var data []byte
|
|
||||||
data, err = ioutil.ReadFile(fpath)
|
|
||||||
if err != nil {
|
|
||||||
c <- err
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(hasBeegoRegex.Find(data)) > 0 {
|
|
||||||
mainFiles = append(mainFiles, fpath)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
close(c)
|
|
||||||
}()
|
|
||||||
|
|
||||||
if err := <-c; err != nil {
|
|
||||||
logger.Fatalf("Unable to walk '%s' tree: %s", thePath, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(mainFiles) > 0 {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
// SearchGOPATHs searchs the user GOPATH(s) for the specified application name.
|
|
||||||
// It returns a boolean, the application's GOPATH and its full path.
|
|
||||||
func SearchGOPATHs(app string) (bool, string, string) {
|
|
||||||
gps := GetGOPATHs()
|
|
||||||
if len(gps) == 0 {
|
|
||||||
logger.Fatal("GOPATH environment variable is not set or empty")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Lookup the application inside the user workspace(s)
|
|
||||||
for _, gopath := range gps {
|
|
||||||
var currentPath string
|
|
||||||
|
|
||||||
if !strings.Contains(app, "src") {
|
|
||||||
gopathsrc := path.Join(gopath, "src")
|
|
||||||
currentPath = path.Join(gopathsrc, app)
|
|
||||||
} else {
|
|
||||||
currentPath = app
|
|
||||||
}
|
|
||||||
|
|
||||||
if isExist(currentPath) {
|
|
||||||
return true, gopath, currentPath
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false, "", ""
|
|
||||||
}
|
|
||||||
|
|
||||||
// askForConfirmation uses Scanln to parse user input. A user must type in "yes" or "no" and
|
|
||||||
// then press enter. It has fuzzy matching, so "y", "Y", "yes", "YES", and "Yes" all count as
|
|
||||||
// confirmations. If the input is not recognized, it will ask again. The function does not return
|
|
||||||
// until it gets a valid response from the user. Typically, you should use fmt to print out a question
|
|
||||||
// before calling askForConfirmation. E.g. fmt.Println("WARNING: Are you sure? (yes/no)")
|
|
||||||
func askForConfirmation() bool {
|
|
||||||
var response string
|
|
||||||
_, err := fmt.Scanln(&response)
|
|
||||||
if err != nil {
|
|
||||||
logger.Fatalf("%s", err)
|
|
||||||
}
|
|
||||||
okayResponses := []string{"y", "Y", "yes", "Yes", "YES"}
|
|
||||||
nokayResponses := []string{"n", "N", "no", "No", "NO"}
|
|
||||||
if containsString(okayResponses, response) {
|
|
||||||
return true
|
|
||||||
} else if containsString(nokayResponses, response) {
|
|
||||||
return false
|
|
||||||
} else {
|
|
||||||
fmt.Println("Please type yes or no and then press enter:")
|
|
||||||
return askForConfirmation()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func containsString(slice []string, element string) bool {
|
|
||||||
for _, elem := range slice {
|
|
||||||
if elem == element {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
// snake string, XxYy to xx_yy
|
|
||||||
func snakeString(s string) string {
|
|
||||||
data := make([]byte, 0, len(s)*2)
|
|
||||||
j := false
|
|
||||||
num := len(s)
|
|
||||||
for i := 0; i < num; i++ {
|
|
||||||
d := s[i]
|
|
||||||
if i > 0 && d >= 'A' && d <= 'Z' && j {
|
|
||||||
data = append(data, '_')
|
|
||||||
}
|
|
||||||
if d != '_' {
|
|
||||||
j = true
|
|
||||||
}
|
|
||||||
data = append(data, d)
|
|
||||||
}
|
|
||||||
return strings.ToLower(string(data[:]))
|
|
||||||
}
|
|
||||||
|
|
||||||
func camelString(s string) string {
|
|
||||||
data := make([]byte, 0, len(s))
|
|
||||||
j := false
|
|
||||||
k := false
|
|
||||||
num := len(s) - 1
|
|
||||||
for i := 0; i <= num; i++ {
|
|
||||||
d := s[i]
|
|
||||||
if k == false && d >= 'A' && d <= 'Z' {
|
|
||||||
k = true
|
|
||||||
}
|
|
||||||
if d >= 'a' && d <= 'z' && (j || k == false) {
|
|
||||||
d = d - 32
|
|
||||||
j = false
|
|
||||||
k = true
|
|
||||||
}
|
|
||||||
if k && d == '_' && num > i && s[i+1] >= 'a' && s[i+1] <= 'z' {
|
|
||||||
j = true
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
data = append(data, d)
|
|
||||||
}
|
|
||||||
return string(data[:])
|
|
||||||
}
|
|
||||||
|
|
||||||
// camelCase converts a _ delimited string to camel case
|
|
||||||
// e.g. very_important_person => VeryImportantPerson
|
|
||||||
func camelCase(in string) string {
|
|
||||||
tokens := strings.Split(in, "_")
|
|
||||||
for i := range tokens {
|
|
||||||
tokens[i] = strings.Title(strings.Trim(tokens[i], " "))
|
|
||||||
}
|
|
||||||
return strings.Join(tokens, "")
|
|
||||||
}
|
|
||||||
|
|
||||||
// formatSourceCode formats source files
|
|
||||||
func formatSourceCode(filename string) {
|
|
||||||
cmd := exec.Command("gofmt", "-w", filename)
|
|
||||||
if err := cmd.Run(); err != nil {
|
|
||||||
logger.Warnf("Error while running gofmt: %s", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// The string flag list, implemented flag.Value interface
|
|
||||||
type strFlags []string
|
|
||||||
|
|
||||||
func (s *strFlags) String() string {
|
|
||||||
return fmt.Sprintf("%s", *s)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *strFlags) Set(value string) error {
|
|
||||||
*s = append(*s, value)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// CloseFile attempts to close the passed file
|
|
||||||
// or panics with the actual error
|
|
||||||
func CloseFile(f *os.File) {
|
|
||||||
err := f.Close()
|
|
||||||
MustCheck(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// MustCheck panics when the error is not nil
|
|
||||||
func MustCheck(err error) {
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func exitPrint(con string) {
|
|
||||||
fmt.Fprintln(os.Stderr, con)
|
|
||||||
os.Exit(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
// WriteToFile creates a file and writes content to it
|
|
||||||
func WriteToFile(filename, content string) {
|
|
||||||
f, err := os.Create(filename)
|
|
||||||
MustCheck(err)
|
|
||||||
defer CloseFile(f)
|
|
||||||
_, err = f.WriteString(content)
|
|
||||||
MustCheck(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// IsDebugEnabled checks if DEBUG_ENABLED is set or not
|
|
||||||
func IsDebugEnabled() bool {
|
|
||||||
debugMode := os.Getenv("DEBUG_ENABLED")
|
|
||||||
return map[string]bool{"1": true, "0": false}[debugMode]
|
|
||||||
}
|
|
||||||
|
|
||||||
// __FILE__ returns the file name in which the function was invoked
|
|
||||||
func __FILE__() string {
|
|
||||||
_, file, _, _ := runtime.Caller(1)
|
|
||||||
return file
|
|
||||||
}
|
|
||||||
|
|
||||||
// __LINE__ returns the line number at which the function was invoked
|
|
||||||
func __LINE__() int {
|
|
||||||
_, _, line, _ := runtime.Caller(1)
|
|
||||||
return line
|
|
||||||
}
|
|
||||||
|
|
||||||
// BeeFuncMap returns a FuncMap of functions used in different templates.
|
|
||||||
func BeeFuncMap() template.FuncMap {
|
|
||||||
return template.FuncMap{
|
|
||||||
"trim": strings.TrimSpace,
|
|
||||||
"bold": bold,
|
|
||||||
"headline": MagentaBold,
|
|
||||||
"foldername": RedBold,
|
|
||||||
"endline": EndLine,
|
|
||||||
"tmpltostr": TmplToString,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// TmplToString parses a text template and return the result as a string.
|
|
||||||
func TmplToString(tmpl string, data interface{}) string {
|
|
||||||
t := template.New("tmpl").Funcs(BeeFuncMap())
|
|
||||||
template.Must(t.Parse(tmpl))
|
|
||||||
|
|
||||||
var doc bytes.Buffer
|
|
||||||
err := t.Execute(&doc, data)
|
|
||||||
MustCheck(err)
|
|
||||||
|
|
||||||
return doc.String()
|
|
||||||
}
|
|
14
utils/doc_value.go
Normal file
14
utils/doc_value.go
Normal file
|
@ -0,0 +1,14 @@
|
||||||
|
package utils
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
type DocValue string
|
||||||
|
|
||||||
|
func (d *DocValue) String() string {
|
||||||
|
return fmt.Sprint(*d)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DocValue) Set(value string) error {
|
||||||
|
*d = DocValue(value)
|
||||||
|
return nil
|
||||||
|
}
|
14
utils/list_opts.go
Normal file
14
utils/list_opts.go
Normal file
|
@ -0,0 +1,14 @@
|
||||||
|
package utils
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
type ListOpts []string
|
||||||
|
|
||||||
|
func (opts *ListOpts) String() string {
|
||||||
|
return fmt.Sprint(*opts)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (opts *ListOpts) Set(value string) error {
|
||||||
|
*opts = append(*opts, value)
|
||||||
|
return nil
|
||||||
|
}
|
84
utils/notification.go
Normal file
84
utils/notification.go
Normal file
|
@ -0,0 +1,84 @@
|
||||||
|
// Copyright 2017 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os/exec"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"runtime"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
const appName = "Beego"
|
||||||
|
|
||||||
|
func Notify(text, title string) {
|
||||||
|
if !config.Conf.EnableNotification {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
switch runtime.GOOS {
|
||||||
|
case "darwin":
|
||||||
|
osxNotify(text, title)
|
||||||
|
case "linux":
|
||||||
|
linuxNotify(text, title)
|
||||||
|
case "windows":
|
||||||
|
windowsNotify(text, title)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func osxNotify(text, title string) {
|
||||||
|
var cmd *exec.Cmd
|
||||||
|
if existTerminalNotifier() {
|
||||||
|
cmd = exec.Command("terminal-notifier", "-title", appName, "-message", text, "-subtitle", title)
|
||||||
|
} else if MacOSVersionSupport() {
|
||||||
|
notification := fmt.Sprintf("display notification \"%s\" with title \"%s\" subtitle \"%s\"", text, appName, title)
|
||||||
|
cmd = exec.Command("osascript", "-e", notification)
|
||||||
|
} else {
|
||||||
|
cmd = exec.Command("growlnotify", "-n", appName, "-m", title)
|
||||||
|
}
|
||||||
|
cmd.Run()
|
||||||
|
}
|
||||||
|
|
||||||
|
func windowsNotify(text, title string) {
|
||||||
|
exec.Command("growlnotify", "/i:", "", "/t:", title, text).Run()
|
||||||
|
}
|
||||||
|
|
||||||
|
func linuxNotify(text, title string) {
|
||||||
|
exec.Command("notify-send", "-i", "", title, text).Run()
|
||||||
|
}
|
||||||
|
|
||||||
|
func existTerminalNotifier() bool {
|
||||||
|
cmd := exec.Command("which", "terminal-notifier")
|
||||||
|
err := cmd.Start()
|
||||||
|
if err != nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
err = cmd.Wait()
|
||||||
|
return err != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func MacOSVersionSupport() bool {
|
||||||
|
cmd := exec.Command("sw_vers", "-productVersion")
|
||||||
|
check, _ := cmd.Output()
|
||||||
|
version := strings.Split(string(check), ".")
|
||||||
|
major, _ := strconv.Atoi(version[0])
|
||||||
|
minor, _ := strconv.Atoi(version[1])
|
||||||
|
if major < 10 || (major == 10 && minor < 9) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
15
utils/str_flag.go
Normal file
15
utils/str_flag.go
Normal file
|
@ -0,0 +1,15 @@
|
||||||
|
package utils
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
// The string flag list, implemented flag.Value interface
|
||||||
|
type StrFlags []string
|
||||||
|
|
||||||
|
func (s *StrFlags) String() string {
|
||||||
|
return fmt.Sprintf("%s", *s)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *StrFlags) Set(value string) error {
|
||||||
|
*s = append(*s, value)
|
||||||
|
return nil
|
||||||
|
}
|
652
utils/utils.go
Normal file
652
utils/utils.go
Normal file
|
@ -0,0 +1,652 @@
|
||||||
|
// Copyright 2013 bee authors
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"): you may
|
||||||
|
// not use this file except in compliance with the License. You may obtain
|
||||||
|
// a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
// License for the specific language governing permissions and limitations
|
||||||
|
// under the License.
|
||||||
|
|
||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"path"
|
||||||
|
"path/filepath"
|
||||||
|
"regexp"
|
||||||
|
"runtime"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"text/template"
|
||||||
|
"time"
|
||||||
|
"unicode"
|
||||||
|
|
||||||
|
"github.com/beego/bee/v2/config"
|
||||||
|
"github.com/beego/bee/v2/internal/pkg/system"
|
||||||
|
beeLogger "github.com/beego/bee/v2/logger"
|
||||||
|
"github.com/beego/bee/v2/logger/colors"
|
||||||
|
)
|
||||||
|
|
||||||
|
type tagName struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Repos struct {
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
PushedAt time.Time `json:"pushed_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Releases struct {
|
||||||
|
PublishedAt time.Time `json:"published_at"`
|
||||||
|
TagName string `json:"tag_name"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetBeeWorkPath() string {
|
||||||
|
curpath, err := os.Getwd()
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return curpath
|
||||||
|
}
|
||||||
|
|
||||||
|
// Go is a basic promise implementation: it wraps calls a function in a goroutine
|
||||||
|
// and returns a channel which will later return the function's return value.
|
||||||
|
func Go(f func() error) chan error {
|
||||||
|
ch := make(chan error)
|
||||||
|
go func() {
|
||||||
|
ch <- f()
|
||||||
|
}()
|
||||||
|
return ch
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsExist returns whether a file or directory exists.
|
||||||
|
func IsExist(path string) bool {
|
||||||
|
_, err := os.Stat(path)
|
||||||
|
return err == nil || os.IsExist(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetGOPATHs returns all paths in GOPATH variable.
|
||||||
|
func GetGOPATHs() []string {
|
||||||
|
gopath := os.Getenv("GOPATH")
|
||||||
|
if gopath == "" && strings.Compare(runtime.Version(), "go1.8") >= 0 {
|
||||||
|
gopath = defaultGOPATH()
|
||||||
|
}
|
||||||
|
return filepath.SplitList(gopath)
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsInGOPATH checks whether the path is inside of any GOPATH or not
|
||||||
|
func IsInGOPATH(thePath string) bool {
|
||||||
|
for _, gopath := range GetGOPATHs() {
|
||||||
|
if strings.Contains(thePath, filepath.Join(gopath, "src")) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsBeegoProject checks whether the current path is a Beego application or not
|
||||||
|
func IsBeegoProject(thePath string) bool {
|
||||||
|
mainFiles := []string{}
|
||||||
|
hasBeegoRegex := regexp.MustCompile(`(?s)package main.*?import.*?\(.*?github.com/beego/beego/v2".*?\).*func main()`)
|
||||||
|
c := make(chan error)
|
||||||
|
// Walk the application path tree to look for main files.
|
||||||
|
// Main files must satisfy the 'hasBeegoRegex' regular expression.
|
||||||
|
go func() {
|
||||||
|
filepath.Walk(thePath, func(fpath string, f os.FileInfo, err error) error {
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
// Skip sub-directories
|
||||||
|
if !f.IsDir() {
|
||||||
|
var data []byte
|
||||||
|
data, err = ioutil.ReadFile(fpath)
|
||||||
|
if err != nil {
|
||||||
|
c <- err
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(hasBeegoRegex.Find(data)) > 0 {
|
||||||
|
mainFiles = append(mainFiles, fpath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
close(c)
|
||||||
|
}()
|
||||||
|
|
||||||
|
if err := <-c; err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Unable to walk '%s' tree: %s", thePath, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(mainFiles) > 0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// SearchGOPATHs searchs the user GOPATH(s) for the specified application name.
|
||||||
|
// It returns a boolean, the application's GOPATH and its full path.
|
||||||
|
func SearchGOPATHs(app string) (bool, string, string) {
|
||||||
|
gps := GetGOPATHs()
|
||||||
|
if len(gps) == 0 {
|
||||||
|
beeLogger.Log.Fatal("GOPATH environment variable is not set or empty")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Lookup the application inside the user workspace(s)
|
||||||
|
for _, gopath := range gps {
|
||||||
|
var currentPath string
|
||||||
|
|
||||||
|
if !strings.Contains(app, "src") {
|
||||||
|
gopathsrc := path.Join(gopath, "src")
|
||||||
|
currentPath = path.Join(gopathsrc, app)
|
||||||
|
} else {
|
||||||
|
currentPath = app
|
||||||
|
}
|
||||||
|
|
||||||
|
if IsExist(currentPath) {
|
||||||
|
return true, gopath, currentPath
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false, "", ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// askForConfirmation uses Scanln to parse user input. A user must type in "yes" or "no" and
|
||||||
|
// then press enter. It has fuzzy matching, so "y", "Y", "yes", "YES", and "Yes" all count as
|
||||||
|
// confirmations. If the input is not recognized, it will ask again. The function does not return
|
||||||
|
// until it gets a valid response from the user. Typically, you should use fmt to print out a question
|
||||||
|
// before calling askForConfirmation. E.g. fmt.Println("WARNING: Are you sure? (yes/no)")
|
||||||
|
func AskForConfirmation() bool {
|
||||||
|
var response string
|
||||||
|
_, err := fmt.Scanln(&response)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Fatalf("%s", err)
|
||||||
|
}
|
||||||
|
okayResponses := []string{"y", "Y", "yes", "Yes", "YES"}
|
||||||
|
nokayResponses := []string{"n", "N", "no", "No", "NO"}
|
||||||
|
if containsString(okayResponses, response) {
|
||||||
|
return true
|
||||||
|
} else if containsString(nokayResponses, response) {
|
||||||
|
return false
|
||||||
|
} else {
|
||||||
|
fmt.Println("Please type yes or no and then press enter:")
|
||||||
|
return AskForConfirmation()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func containsString(slice []string, element string) bool {
|
||||||
|
for _, elem := range slice {
|
||||||
|
if elem == element {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// snake string, XxYy to xx_yy
|
||||||
|
func SnakeString(s string) string {
|
||||||
|
data := make([]byte, 0, len(s)*2)
|
||||||
|
j := false
|
||||||
|
num := len(s)
|
||||||
|
for i := 0; i < num; i++ {
|
||||||
|
d := s[i]
|
||||||
|
if i > 0 && d >= 'A' && d <= 'Z' && j {
|
||||||
|
data = append(data, '_')
|
||||||
|
}
|
||||||
|
if d != '_' {
|
||||||
|
j = true
|
||||||
|
}
|
||||||
|
data = append(data, d)
|
||||||
|
}
|
||||||
|
return strings.ToLower(string(data[:]))
|
||||||
|
}
|
||||||
|
|
||||||
|
func CamelString(s string) string {
|
||||||
|
data := make([]byte, 0, len(s))
|
||||||
|
j := false
|
||||||
|
k := false
|
||||||
|
num := len(s) - 1
|
||||||
|
for i := 0; i <= num; i++ {
|
||||||
|
d := s[i]
|
||||||
|
if !k && d >= 'A' && d <= 'Z' {
|
||||||
|
k = true
|
||||||
|
}
|
||||||
|
if d >= 'a' && d <= 'z' && (j || !k) {
|
||||||
|
d = d - 32
|
||||||
|
j = false
|
||||||
|
k = true
|
||||||
|
}
|
||||||
|
if k && d == '_' && num > i && s[i+1] >= 'a' && s[i+1] <= 'z' {
|
||||||
|
j = true
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
data = append(data, d)
|
||||||
|
}
|
||||||
|
return string(data[:])
|
||||||
|
}
|
||||||
|
|
||||||
|
// camelCase converts a _ delimited string to camel case
|
||||||
|
// e.g. very_important_person => VeryImportantPerson
|
||||||
|
func CamelCase(in string) string {
|
||||||
|
tokens := strings.Split(in, "_")
|
||||||
|
for i := range tokens {
|
||||||
|
tokens[i] = strings.Title(strings.Trim(tokens[i], " "))
|
||||||
|
}
|
||||||
|
return strings.Join(tokens, "")
|
||||||
|
}
|
||||||
|
|
||||||
|
// formatSourceCode formats source files
|
||||||
|
func FormatSourceCode(filename string) {
|
||||||
|
cmd := exec.Command("gofmt", "-w", filename)
|
||||||
|
if err := cmd.Run(); err != nil {
|
||||||
|
beeLogger.Log.Warnf("Error while running gofmt: %s", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// CloseFile attempts to close the passed file
|
||||||
|
// or panics with the actual error
|
||||||
|
func CloseFile(f *os.File) {
|
||||||
|
err := f.Close()
|
||||||
|
MustCheck(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MustCheck panics when the error is not nil
|
||||||
|
func MustCheck(err error) {
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// WriteToFile creates a file and writes content to it
|
||||||
|
func WriteToFile(filename, content string) {
|
||||||
|
f, err := os.Create(filename)
|
||||||
|
MustCheck(err)
|
||||||
|
defer CloseFile(f)
|
||||||
|
_, err = f.WriteString(content)
|
||||||
|
MustCheck(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// __FILE__ returns the file name in which the function was invoked
|
||||||
|
func FILE() string {
|
||||||
|
_, file, _, _ := runtime.Caller(1)
|
||||||
|
return file
|
||||||
|
}
|
||||||
|
|
||||||
|
// __LINE__ returns the line number at which the function was invoked
|
||||||
|
func LINE() int {
|
||||||
|
_, _, line, _ := runtime.Caller(1)
|
||||||
|
return line
|
||||||
|
}
|
||||||
|
|
||||||
|
// BeeFuncMap returns a FuncMap of functions used in different templates.
|
||||||
|
func BeeFuncMap() template.FuncMap {
|
||||||
|
return template.FuncMap{
|
||||||
|
"trim": strings.TrimSpace,
|
||||||
|
"bold": colors.Bold,
|
||||||
|
"headline": colors.MagentaBold,
|
||||||
|
"foldername": colors.RedBold,
|
||||||
|
"endline": EndLine,
|
||||||
|
"tmpltostr": TmplToString,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TmplToString parses a text template and return the result as a string.
|
||||||
|
func TmplToString(tmpl string, data interface{}) string {
|
||||||
|
t := template.New("tmpl").Funcs(BeeFuncMap())
|
||||||
|
template.Must(t.Parse(tmpl))
|
||||||
|
|
||||||
|
var doc bytes.Buffer
|
||||||
|
err := t.Execute(&doc, data)
|
||||||
|
MustCheck(err)
|
||||||
|
|
||||||
|
return doc.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// EndLine returns the a newline escape character
|
||||||
|
func EndLine() string {
|
||||||
|
return "\n"
|
||||||
|
}
|
||||||
|
|
||||||
|
func Tmpl(text string, data interface{}) {
|
||||||
|
output := colors.NewColorWriter(os.Stderr)
|
||||||
|
|
||||||
|
t := template.New("Usage").Funcs(BeeFuncMap())
|
||||||
|
template.Must(t.Parse(text))
|
||||||
|
|
||||||
|
err := t.Execute(output, data)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Error(err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func CheckEnv(appname string) (apppath, packpath string, err error) {
|
||||||
|
gps := GetGOPATHs()
|
||||||
|
if len(gps) == 0 {
|
||||||
|
beeLogger.Log.Error("if you want new a go module project,please add param `-gopath=false`.")
|
||||||
|
beeLogger.Log.Fatal("GOPATH environment variable is not set or empty")
|
||||||
|
}
|
||||||
|
currpath, _ := os.Getwd()
|
||||||
|
currpath = filepath.Join(currpath, appname)
|
||||||
|
for _, gpath := range gps {
|
||||||
|
gsrcpath := filepath.Join(gpath, "src")
|
||||||
|
if strings.HasPrefix(strings.ToLower(currpath), strings.ToLower(gsrcpath)) {
|
||||||
|
packpath = strings.Replace(currpath[len(gsrcpath)+1:], string(filepath.Separator), "/", -1)
|
||||||
|
return currpath, packpath, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// In case of multiple paths in the GOPATH, by default
|
||||||
|
// we use the first path
|
||||||
|
gopath := gps[0]
|
||||||
|
|
||||||
|
beeLogger.Log.Warn("You current workdir is not inside $GOPATH/src.")
|
||||||
|
beeLogger.Log.Debugf("GOPATH: %s", FILE(), LINE(), gopath)
|
||||||
|
|
||||||
|
gosrcpath := filepath.Join(gopath, "src")
|
||||||
|
apppath = filepath.Join(gosrcpath, appname)
|
||||||
|
|
||||||
|
if _, e := os.Stat(apppath); !os.IsNotExist(e) {
|
||||||
|
err = fmt.Errorf("cannot create application without removing '%s' first", apppath)
|
||||||
|
beeLogger.Log.Errorf("Path '%s' already exists", apppath)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
packpath = strings.Join(strings.Split(apppath[len(gosrcpath)+1:], string(filepath.Separator)), "/")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func PrintErrorAndExit(message, errorTemplate string) {
|
||||||
|
Tmpl(fmt.Sprintf(errorTemplate, message), nil)
|
||||||
|
os.Exit(2)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GoCommand executes the passed command using Go tool
|
||||||
|
func GoCommand(command string, args ...string) error {
|
||||||
|
allargs := []string{command}
|
||||||
|
allargs = append(allargs, args...)
|
||||||
|
goBuild := exec.Command("go", allargs...)
|
||||||
|
goBuild.Stderr = os.Stderr
|
||||||
|
return goBuild.Run()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SplitQuotedFields is like strings.Fields but ignores spaces
|
||||||
|
// inside areas surrounded by single quotes.
|
||||||
|
// To specify a single quote use backslash to escape it: '\''
|
||||||
|
func SplitQuotedFields(in string) []string {
|
||||||
|
type stateEnum int
|
||||||
|
const (
|
||||||
|
inSpace stateEnum = iota
|
||||||
|
inField
|
||||||
|
inQuote
|
||||||
|
inQuoteEscaped
|
||||||
|
)
|
||||||
|
state := inSpace
|
||||||
|
r := []string{}
|
||||||
|
var buf bytes.Buffer
|
||||||
|
|
||||||
|
for _, ch := range in {
|
||||||
|
switch state {
|
||||||
|
case inSpace:
|
||||||
|
if ch == '\'' {
|
||||||
|
state = inQuote
|
||||||
|
} else if !unicode.IsSpace(ch) {
|
||||||
|
buf.WriteRune(ch)
|
||||||
|
state = inField
|
||||||
|
}
|
||||||
|
|
||||||
|
case inField:
|
||||||
|
if ch == '\'' {
|
||||||
|
state = inQuote
|
||||||
|
} else if unicode.IsSpace(ch) {
|
||||||
|
r = append(r, buf.String())
|
||||||
|
buf.Reset()
|
||||||
|
} else {
|
||||||
|
buf.WriteRune(ch)
|
||||||
|
}
|
||||||
|
|
||||||
|
case inQuote:
|
||||||
|
if ch == '\'' {
|
||||||
|
state = inField
|
||||||
|
} else if ch == '\\' {
|
||||||
|
state = inQuoteEscaped
|
||||||
|
} else {
|
||||||
|
buf.WriteRune(ch)
|
||||||
|
}
|
||||||
|
|
||||||
|
case inQuoteEscaped:
|
||||||
|
buf.WriteRune(ch)
|
||||||
|
state = inQuote
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if buf.Len() != 0 {
|
||||||
|
r = append(r, buf.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetFileModTime returns unix timestamp of `os.File.ModTime` for the given path.
|
||||||
|
func GetFileModTime(path string) int64 {
|
||||||
|
path = strings.Replace(path, "\\", "/", -1)
|
||||||
|
f, err := os.Open(path)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Failed to open file on '%s': %s", path, err)
|
||||||
|
return time.Now().Unix()
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
fi, err := f.Stat()
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Errorf("Failed to get file stats: %s", err)
|
||||||
|
return time.Now().Unix()
|
||||||
|
}
|
||||||
|
|
||||||
|
return fi.ModTime().Unix()
|
||||||
|
}
|
||||||
|
|
||||||
|
func defaultGOPATH() string {
|
||||||
|
env := "HOME"
|
||||||
|
if runtime.GOOS == "windows" {
|
||||||
|
env = "USERPROFILE"
|
||||||
|
} else if runtime.GOOS == "plan9" {
|
||||||
|
env = "home"
|
||||||
|
}
|
||||||
|
if home := os.Getenv(env); home != "" {
|
||||||
|
return filepath.Join(home, "go")
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetGoVersionSkipMinor() string {
|
||||||
|
strArray := strings.Split(runtime.Version()[2:], `.`)
|
||||||
|
return strArray[0] + `.` + strArray[1]
|
||||||
|
}
|
||||||
|
|
||||||
|
func IsGOMODULE() bool {
|
||||||
|
if combinedOutput, e := exec.Command(`go`, `env`).CombinedOutput(); e != nil {
|
||||||
|
beeLogger.Log.Errorf("i cann't find go.")
|
||||||
|
} else {
|
||||||
|
regex := regexp.MustCompile(`GOMOD="?(.+go.mod)"?`)
|
||||||
|
stringSubmatch := regex.FindStringSubmatch(string(combinedOutput))
|
||||||
|
return len(stringSubmatch) == 2
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func NoticeUpdateBee() {
|
||||||
|
cmd := exec.Command("go", "version")
|
||||||
|
cmd.Output()
|
||||||
|
if cmd.Process == nil || cmd.Process.Pid <= 0 {
|
||||||
|
beeLogger.Log.Warn("There is no go environment")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
beeHome := system.BeegoHome
|
||||||
|
if !IsExist(beeHome) {
|
||||||
|
if err := os.MkdirAll(beeHome, 0755); err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not create the directory: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fp := beeHome + "/.noticeUpdateBee"
|
||||||
|
timeNow := time.Now().Unix()
|
||||||
|
var timeOld int64
|
||||||
|
if !IsExist(fp) {
|
||||||
|
f, err := os.Create(fp)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Warnf("Create noticeUpdateBee file err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
}
|
||||||
|
oldContent, err := ioutil.ReadFile(fp)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Warnf("Read noticeUpdateBee file err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
timeOld, _ = strconv.ParseInt(string(oldContent), 10, 64)
|
||||||
|
if timeNow-timeOld < 24*60*60 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
w, err := os.OpenFile(fp, os.O_WRONLY|os.O_TRUNC, 0644)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Warnf("Open noticeUpdateBee file err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer w.Close()
|
||||||
|
timeNowStr := strconv.FormatInt(timeNow, 10)
|
||||||
|
if _, err := w.WriteString(timeNowStr); err != nil {
|
||||||
|
beeLogger.Log.Warnf("Update noticeUpdateBee file err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
beeLogger.Log.Info("Getting bee latest version...")
|
||||||
|
versionLast := BeeLastVersion()
|
||||||
|
versionNow := config.Version
|
||||||
|
if versionLast == "" {
|
||||||
|
beeLogger.Log.Warn("Get latest version err")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if versionNow != versionLast {
|
||||||
|
beeLogger.Log.Warnf("Update available %s ==> %s", versionNow, versionLast)
|
||||||
|
beeLogger.Log.Warn("Run `bee update` to update")
|
||||||
|
}
|
||||||
|
beeLogger.Log.Info("Your bee are up to date")
|
||||||
|
}
|
||||||
|
|
||||||
|
func BeeLastVersion() (version string) {
|
||||||
|
var url = "https://api.github.com/repos/beego/bee/tags"
|
||||||
|
resp, err := http.Get(url)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Warnf("Get bee tags from github error: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
bodyContent, _ := ioutil.ReadAll(resp.Body)
|
||||||
|
var tags []tagName
|
||||||
|
if err = json.Unmarshal(bodyContent, &tags); err != nil {
|
||||||
|
beeLogger.Log.Warnf("Unmarshal tags body error: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if len(tags) < 1 {
|
||||||
|
beeLogger.Log.Warn("There is no tags!")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
last := tags[0]
|
||||||
|
re, _ := regexp.Compile(`[0-9.]+`)
|
||||||
|
versionList := re.FindStringSubmatch(last.Name)
|
||||||
|
if len(versionList) > 0 {
|
||||||
|
return versionList[0]
|
||||||
|
}
|
||||||
|
beeLogger.Log.Warn("There is no tags!")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// get info of bee repos
|
||||||
|
func BeeReposInfo() (repos Repos) {
|
||||||
|
var url = "https://api.github.com/repos/beego/bee"
|
||||||
|
resp, err := http.Get(url)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Warnf("Get bee repos from github error: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
bodyContent, _ := ioutil.ReadAll(resp.Body)
|
||||||
|
if err = json.Unmarshal(bodyContent, &repos); err != nil {
|
||||||
|
beeLogger.Log.Warnf("Unmarshal repos body error: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// get info of bee releases
|
||||||
|
func BeeReleasesInfo() (repos []Releases) {
|
||||||
|
var url = "https://api.github.com/repos/beego/bee/releases"
|
||||||
|
resp, err := http.Get(url)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Warnf("Get bee releases from github error: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
bodyContent, _ := ioutil.ReadAll(resp.Body)
|
||||||
|
if err = json.Unmarshal(bodyContent, &repos); err != nil {
|
||||||
|
beeLogger.Log.Warnf("Unmarshal releases body error: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
//TODO merge UpdateLastPublishedTime and NoticeUpdateBee
|
||||||
|
func UpdateLastPublishedTime() {
|
||||||
|
info := BeeReleasesInfo()
|
||||||
|
if len(info) == 0 {
|
||||||
|
beeLogger.Log.Warn("Has no releases")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
createdAt := info[0].PublishedAt.Format("2006-01-02")
|
||||||
|
beeHome := system.BeegoHome
|
||||||
|
if !IsExist(beeHome) {
|
||||||
|
if err := os.MkdirAll(beeHome, 0755); err != nil {
|
||||||
|
beeLogger.Log.Fatalf("Could not create the directory: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fp := beeHome + "/.lastPublishedAt"
|
||||||
|
w, err := os.OpenFile(fp, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Warnf("Open .lastPublishedAt file err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer w.Close()
|
||||||
|
if _, err := w.WriteString(createdAt); err != nil {
|
||||||
|
beeLogger.Log.Warnf("Update .lastPublishedAt file err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetLastPublishedTime() string {
|
||||||
|
fp := system.BeegoHome + "/.lastPublishedAt"
|
||||||
|
if !IsExist(fp) {
|
||||||
|
UpdateLastPublishedTime()
|
||||||
|
}
|
||||||
|
w, err := os.OpenFile(fp, os.O_RDONLY, 0644)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Warnf("Open .lastPublishedAt file err: %s", err)
|
||||||
|
return "unknown"
|
||||||
|
}
|
||||||
|
t := make([]byte, 1024)
|
||||||
|
read, err := w.Read(t)
|
||||||
|
if err != nil {
|
||||||
|
beeLogger.Log.Warnf("read .lastPublishedAt file err: %s", err)
|
||||||
|
return "unknown"
|
||||||
|
}
|
||||||
|
return string(t[:read])
|
||||||
|
}
|
|
@ -1,10 +1,10 @@
|
||||||
// Copyright 2014 beego Author. All Rights Reserved.
|
// Copyright 2020
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
// You may obtain a copy of the License at
|
// You may obtain a copy of the License at
|
||||||
//
|
//
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
//
|
//
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
@ -14,12 +14,4 @@
|
||||||
|
|
||||||
package utils
|
package utils
|
||||||
|
|
||||||
import (
|
const BEEGO_VERSION = "v2.1.0"
|
||||||
"reflect"
|
|
||||||
"runtime"
|
|
||||||
)
|
|
||||||
|
|
||||||
// GetFuncName get function name
|
|
||||||
func GetFuncName(i interface{}) string {
|
|
||||||
return runtime.FuncForPC(reflect.ValueOf(i).Pointer()).Name()
|
|
||||||
}
|
|
13
vendor/github.com/astaxie/beego/LICENSE
generated
vendored
13
vendor/github.com/astaxie/beego/LICENSE
generated
vendored
|
@ -1,13 +0,0 @@
|
||||||
Copyright 2014 astaxie
|
|
||||||
|
|
||||||
Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
you may not use this file except in compliance with the License.
|
|
||||||
You may obtain a copy of the License at
|
|
||||||
|
|
||||||
http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
|
|
||||||
Unless required by applicable law or agreed to in writing, software
|
|
||||||
distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
See the License for the specific language governing permissions and
|
|
||||||
limitations under the License.
|
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user