mirror of
https://github.com/beego/bee.git
synced 2024-11-05 06:50:53 +00:00
152 lines
3.6 KiB
Go
152 lines
3.6 KiB
Go
// 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"
|
|
"log"
|
|
"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.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 {
|
|
fmt.Println("has err exet unzipAndDelete", err)
|
|
}
|
|
}
|
|
if docport == "" {
|
|
docport = "8089"
|
|
}
|
|
if _, err := os.Stat("swagger"); err != nil && os.IsNotExist(err) {
|
|
fmt.Println("there's no swagger, please use bee rundocs -isDownload=true downlaod first")
|
|
os.Exit(2)
|
|
}
|
|
fmt.Println("start the docs server on: http://127.0.0.1:" + docport)
|
|
log.Fatal(http.ListenAndServe(":"+string(docport), http.FileServer(http.Dir("swagger"))))
|
|
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 {
|
|
ColorLog("[%s] Filename %s already exist\n", INFO, fileName)
|
|
return
|
|
}
|
|
if down {
|
|
ColorLog("[%s]Downloading %s to %s\n", SUCC, url, fileName)
|
|
output, err := os.Create(fileName)
|
|
if err != nil {
|
|
ColorLog("[%s]Error while creating %s: %s\n", ERRO, fileName, err)
|
|
return
|
|
}
|
|
defer output.Close()
|
|
|
|
response, err := http.Get(url)
|
|
if err != nil {
|
|
ColorLog("[%s]Error while downloading %s:%s\n", ERRO, url, err)
|
|
return
|
|
}
|
|
defer response.Body.Close()
|
|
|
|
n, err := io.Copy(output, response.Body)
|
|
if err != nil {
|
|
ColorLog("[%s]Error while downloading %s:%s\n", ERRO, url, err)
|
|
return
|
|
}
|
|
ColorLog("[%s] %d bytes downloaded.\n", SUCC, n)
|
|
}
|
|
}
|
|
|
|
func unzipAndDelete(src string) error {
|
|
ColorLog("[%s]start to unzip file from %s\n", INFO, 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
|
|
}
|
|
}
|
|
}
|
|
ColorLog("[%s]Start delete src file %s\n", INFO, src)
|
|
return os.RemoveAll(src)
|
|
}
|