package cmd

import (
	"fmt"
	"github.com/ghodss/yaml"
	"github.com/pkg/errors"
	"github.com/spf13/cobra"
	"html/template"
	"io/ioutil"
	"os"
	"path/filepath"
	"strings"
)

const extensionJson = ".json"
const extensionYaml = ".yaml"
const extensionHtml = ".html"

const swaggerHtmlTemplate = `
<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Swagger UI</title>
    <link href="https:
    <link rel="stylesheet" type="text/css" href="https:
    <link rel="icon" type="image/png" href="./favicon-32x32.png" sizes="32x32" />
    <link rel="icon" type="image/png" href="./favicon-16x16.png" sizes="16x16" />
    <style>
        html
        {
            box-sizing: border-box;
            overflow: -moz-scrollbars-vertical;
            overflow-y: scroll;
        }
        *,
        *:before,
        *:after
        {
            box-sizing: inherit;
        }

        body {
            margin:0;
            background: #fafafa;
        }
    </style>
</head>

<body>

<svg xmlns="http:
    <defs>
        <symbol viewBox="0 0 20 20" id="unlocked">
            <path d="M15.8 8H14V5.6C14 2.703 12.665 1 10 1 7.334 1 6 2.703 6 5.6V6h2v-.801C8 3.754 8.797 3 10 3c1.203 0 2 .754 2 2.199V8H4c-.553 0-1 .646-1 1.199V17c0 .549.428 1.139.951 1.307l1.197.387C5.672 18.861 6.55 19 7.1 19h5.8c.549 0 1.428-.139 1.951-.307l1.196-.387c.524-.167.953-.757.953-1.306V9.199C17 8.646 16.352 8 15.8 8z"></path>
        </symbol>

        <symbol viewBox="0 0 20 20" id="locked">
            <path d="M15.8 8H14V5.6C14 2.703 12.665 1 10 1 7.334 1 6 2.703 6 5.6V8H4c-.553 0-1 .646-1 1.199V17c0 .549.428 1.139.951 1.307l1.197.387C5.672 18.861 6.55 19 7.1 19h5.8c.549 0 1.428-.139 1.951-.307l1.196-.387c.524-.167.953-.757.953-1.306V9.199C17 8.646 16.352 8 15.8 8zM12 8H8V5.199C8 3.754 8.797 3 10 3c1.203 0 2 .754 2 2.199V8z"/>
        </symbol>

        <symbol viewBox="0 0 20 20" id="close">
            <path d="M14.348 14.849c-.469.469-1.229.469-1.697 0L10 11.819l-2.651 3.029c-.469.469-1.229.469-1.697 0-.469-.469-.469-1.229 0-1.697l2.758-3.15-2.759-3.152c-.469-.469-.469-1.228 0-1.697.469-.469 1.228-.469 1.697 0L10 8.183l2.651-3.031c.469-.469 1.228-.469 1.697 0 .469.469.469 1.229 0 1.697l-2.758 3.152 2.758 3.15c.469.469.469 1.229 0 1.698z"/>
        </symbol>

        <symbol viewBox="0 0 20 20" id="large-arrow">
            <path d="M13.25 10L6.109 2.58c-.268-.27-.268-.707 0-.979.268-.27.701-.27.969 0l7.83 7.908c.268.271.268.709 0 .979l-7.83 7.908c-.268.271-.701.27-.969 0-.268-.269-.268-.707 0-.979L13.25 10z"/>
        </symbol>

        <symbol viewBox="0 0 20 20" id="large-arrow-down">
            <path d="M17.418 6.109c.272-.268.709-.268.979 0s.271.701 0 .969l-7.908 7.83c-.27.268-.707.268-.979 0l-7.908-7.83c-.27-.268-.27-.701 0-.969.271-.268.709-.268.979 0L10 13.25l7.418-7.141z"/>
        </symbol>

        <symbol viewBox="0 0 24 24" id="jump-to">
            <path d="M19 7v4H5.83l3.58-3.59L8 6l-6 6 6 6 1.41-1.41L5.83 13H21V7z"/>
        </symbol>

        <symbol viewBox="0 0 24 24" id="expand">
            <path d="M10 18h4v-2h-4v2zM3 6v2h18V6H3zm3 7h12v-2H6v2z"/>
        </symbol>

    </defs>
</svg>

<div id="swagger-ui"></div>

<script src="https:
<script src="https:
<script>
    window.onload = function() {
        const ui = SwaggerUIBundle({
            spec: {{.Spec}},
            dom_id: '#swagger-ui',
            validatorUrl: null,
            presets: [
                SwaggerUIBundle.presets.apis,
                SwaggerUIStandalonePreset
            ],
            plugins: [
                SwaggerUIBundle.plugins.DownloadUrl
            ],
            layout: "StandaloneLayout"
        })

        window.ui = ui
    }
</script>
</body>

</html>`

func init() {
	rootCmd.AddCommand(genCmd)

	genCmd.Flags().StringP("input", "i", "swagger.json", "Path to input file with swagger spec. Can be json or yaml.")
	genCmd.Flags().StringP("output", "o", "swagger.html", "Path to output html file.")
}

var genCmd = &cobra.Command{
	Use:   "gen",
	Short: "Generate html file from json/yaml spec.",
	RunE:  gen,
}

func gen(cmd *cobra.Command, args []string) error {
	pathToInputFile, pathToOutputFile, err := getFilePaths(cmd)
	if err != nil {
		return err
	}

	specData, err  := getJsonSpecData(pathToInputFile)
	if err != nil {
		return err
	}

	err = generateHtmlFileFromSpecData(pathToOutputFile, specData)
	if err != nil {
		return err
	}

	return nil
}

func getFilePaths(cmd *cobra.Command) (string, string, error) {
	pathToInputFile, err := cmd.Flags().GetString("input")
	if err != nil {
		return "", "", err
	}

	pathToOutputFile, err := cmd.Flags().GetString("output")
	if err != nil {
		return "", "", err
	}

	inputFileExt := filepath.Ext(pathToInputFile)
	if inputFileExt != extensionJson && inputFileExt != extensionYaml {
		return "", "", fmt.Errorf("input file extenshion must be %s", strings.Join([]string{extensionJson, extensionYaml}, " or "))
	}

	outputFileExt := filepath.Ext(pathToOutputFile)
	if outputFileExt != extensionHtml {
		return "", "", fmt.Errorf("output file extenshion must be %s", extensionHtml)
	}

	return pathToInputFile, pathToOutputFile, nil
}

func getJsonSpecData(pathToFile string) (template.JS, error) {
	fileExt := filepath.Ext(pathToFile)

	switch fileExt {
	case extensionYaml:
		return getJsonSpecDataFromYamlFile(pathToFile)
	default:
		return getJsonSpecDataFromJsonFile(pathToFile)
	}
}

func getJsonSpecDataFromJsonFile(pathToFile string) (template.JS, error) {
	fileData, err := ioutil.ReadFile(pathToFile)
	if err != nil {
		return "", errors.Wrap(err, "failed to read file")
	}

	return template.JS(fileData), nil
}

func getJsonSpecDataFromYamlFile(pathToFile string) (template.JS, error) {
	fileData, err := ioutil.ReadFile(pathToFile)
	if err != nil {
		return "", errors.Wrap(err, "failed to read file")
	}

	jsonSpecData, err := yaml.YAMLToJSON(fileData)
	if err != nil {
		return "", errors.Wrap(err, "failed to convert yaml to json")
	}

	return template.JS(jsonSpecData), nil
}

func generateHtmlFileFromSpecData(pathToOutputFile string, specData template.JS) error {
	t, err := template.New("swagger_html").Parse(swaggerHtmlTemplate)
	if err != nil {
		return err
	}

	fd, err := os.Create(pathToOutputFile)
	if err != nil {
		return err
	}
	defer fd.Close()

	err = t.Execute(fd, map[string]template.JS{"Spec": specData})
	if err != nil {
		return nil
	}

	return nil
}
package cmd

import (
	"fmt"
	"github.com/ghodss/yaml"
	"github.com/pkg/errors"
	"github.com/spf13/cobra"
	"html/template"
	"io/ioutil"
	"os"
	"path/filepath"
	"strings"
)

const extensionJson = ".json"
const extensionYaml = ".yaml"
const extensionHtml = ".html"

const swaggerHtmlTemplate = `
<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Swagger UI</title>
    <link href="https://fonts.googleapis.com/css?family=Open+Sans:400,700|Source+Code+Pro:300,600|Titillium+Web:400,600,700" rel="stylesheet">
    <link rel="stylesheet" type="text/css" href="https://cdnjs.cloudflare.com/ajax/libs/swagger-ui/3.24.2/swagger-ui.css" >
    <link rel="icon" type="image/png" href="./favicon-32x32.png" sizes="32x32" />
    <link rel="icon" type="image/png" href="./favicon-16x16.png" sizes="16x16" />
    <style>
        html
        {
            box-sizing: border-box;
            overflow: -moz-scrollbars-vertical;
            overflow-y: scroll;
        }
        *,
        *:before,
        *:after
        {
            box-sizing: inherit;
        }

        body {
            margin:0;
            background: #fafafa;
        }
    </style>
</head>

<body>

<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" style="position:absolute;width:0;height:0">
    <defs>
        <symbol viewBox="0 0 20 20" id="unlocked">
            <path d="M15.8 8H14V5.6C14 2.703 12.665 1 10 1 7.334 1 6 2.703 6 5.6V6h2v-.801C8 3.754 8.797 3 10 3c1.203 0 2 .754 2 2.199V8H4c-.553 0-1 .646-1 1.199V17c0 .549.428 1.139.951 1.307l1.197.387C5.672 18.861 6.55 19 7.1 19h5.8c.549 0 1.428-.139 1.951-.307l1.196-.387c.524-.167.953-.757.953-1.306V9.199C17 8.646 16.352 8 15.8 8z"></path>
        </symbol>

        <symbol viewBox="0 0 20 20" id="locked">
            <path d="M15.8 8H14V5.6C14 2.703 12.665 1 10 1 7.334 1 6 2.703 6 5.6V8H4c-.553 0-1 .646-1 1.199V17c0 .549.428 1.139.951 1.307l1.197.387C5.672 18.861 6.55 19 7.1 19h5.8c.549 0 1.428-.139 1.951-.307l1.196-.387c.524-.167.953-.757.953-1.306V9.199C17 8.646 16.352 8 15.8 8zM12 8H8V5.199C8 3.754 8.797 3 10 3c1.203 0 2 .754 2 2.199V8z"/>
        </symbol>

        <symbol viewBox="0 0 20 20" id="close">
            <path d="M14.348 14.849c-.469.469-1.229.469-1.697 0L10 11.819l-2.651 3.029c-.469.469-1.229.469-1.697 0-.469-.469-.469-1.229 0-1.697l2.758-3.15-2.759-3.152c-.469-.469-.469-1.228 0-1.697.469-.469 1.228-.469 1.697 0L10 8.183l2.651-3.031c.469-.469 1.228-.469 1.697 0 .469.469.469 1.229 0 1.697l-2.758 3.152 2.758 3.15c.469.469.469 1.229 0 1.698z"/>
        </symbol>

        <symbol viewBox="0 0 20 20" id="large-arrow">
            <path d="M13.25 10L6.109 2.58c-.268-.27-.268-.707 0-.979.268-.27.701-.27.969 0l7.83 7.908c.268.271.268.709 0 .979l-7.83 7.908c-.268.271-.701.27-.969 0-.268-.269-.268-.707 0-.979L13.25 10z"/>
        </symbol>

        <symbol viewBox="0 0 20 20" id="large-arrow-down">
            <path d="M17.418 6.109c.272-.268.709-.268.979 0s.271.701 0 .969l-7.908 7.83c-.27.268-.707.268-.979 0l-7.908-7.83c-.27-.268-.27-.701 0-.969.271-.268.709-.268.979 0L10 13.25l7.418-7.141z"/>
        </symbol>


        <symbol viewBox="0 0 24 24" id="jump-to">
            <path d="M19 7v4H5.83l3.58-3.59L8 6l-6 6 6 6 1.41-1.41L5.83 13H21V7z"/>
        </symbol>

        <symbol viewBox="0 0 24 24" id="expand">
            <path d="M10 18h4v-2h-4v2zM3 6v2h18V6H3zm3 7h12v-2H6v2z"/>
        </symbol>

    </defs>
</svg>

<div id="swagger-ui"></div>

<script src="https://cdnjs.cloudflare.com/ajax/libs/swagger-ui/3.24.2/swagger-ui-bundle.js"> </script>
<script src="https://cdnjs.cloudflare.com/ajax/libs/swagger-ui/3.24.2/swagger-ui-standalone-preset.js"> </script>
<script>
    window.onload = function() {
        const ui = SwaggerUIBundle({
            spec: {{.Spec}},
            dom_id: '#swagger-ui',
            validatorUrl: null,
            presets: [
                SwaggerUIBundle.presets.apis,
                SwaggerUIStandalonePreset
            ],
            plugins: [
                SwaggerUIBundle.plugins.DownloadUrl
            ],
            layout: "StandaloneLayout"
        })

        window.ui = ui
    }
</script>
</body>

</html>`

func init() {
	rootCmd.AddCommand(genCmd)

	genCmd.Flags().StringP("input", "i", "swagger.json", "Path to input file with swagger spec. Can be json or yaml.")
	genCmd.Flags().StringP("output", "o", "swagger.html", "Path to output html file.")
}

var genCmd = &cobra.Command{
	Use:   "gen",
	Short: "Generate html file from json/yaml spec.",
	RunE:  gen,
}

func gen(cmd *cobra.Command, args []string) error {
	pathToInputFile, pathToOutputFile, err := getFilePaths(cmd)
	if err != nil {
		return err
	}

	specData, err  := getJsonSpecData(pathToInputFile)
	if err != nil {
		return err
	}

	err = generateHtmlFileFromSpecData(pathToOutputFile, specData)
	if err != nil {
		return err
	}

	return nil
}

func getFilePaths(cmd *cobra.Command) (string, string, error) {
	pathToInputFile, err := cmd.Flags().GetString("input")
	if err != nil {
		return "", "", err
	}

	pathToOutputFile, err := cmd.Flags().GetString("output")
	if err != nil {
		return "", "", err
	}

	inputFileExt := filepath.Ext(pathToInputFile)
	if inputFileExt != extensionJson && inputFileExt != extensionYaml {
		return "", "", fmt.Errorf("input file extenshion must be %s", strings.Join([]string{extensionJson, extensionYaml}, " or "))
	}

	outputFileExt := filepath.Ext(pathToOutputFile)
	if outputFileExt != extensionHtml {
		return "", "", fmt.Errorf("output file extenshion must be %s", extensionHtml)
	}

	return pathToInputFile, pathToOutputFile, nil
}

func getJsonSpecData(pathToFile string) (template.JS, error) {
	fileExt := filepath.Ext(pathToFile)

	switch fileExt {
	case extensionYaml:
		return getJsonSpecDataFromYamlFile(pathToFile)
	default:
		return getJsonSpecDataFromJsonFile(pathToFile)
	}
}

func getJsonSpecDataFromJsonFile(pathToFile string) (template.JS, error) {
	fileData, err := ioutil.ReadFile(pathToFile)
	if err != nil {
		return "", errors.Wrap(err, "failed to read file")
	}

	return template.JS(fileData), nil
}

func getJsonSpecDataFromYamlFile(pathToFile string) (template.JS, error) {
	fileData, err := ioutil.ReadFile(pathToFile)
	if err != nil {
		return "", errors.Wrap(err, "failed to read file")
	}

	jsonSpecData, err := yaml.YAMLToJSON(fileData)
	if err != nil {
		return "", errors.Wrap(err, "failed to convert yaml to json")
	}

	return template.JS(jsonSpecData), nil
}

func generateHtmlFileFromSpecData(pathToOutputFile string, specData template.JS) error {
	t, err := template.New("swagger_html").Parse(swaggerHtmlTemplate)
	if err != nil {
		return err
	}

	fd, err := os.Create(pathToOutputFile)
	if err != nil {
		return err
	}
	defer fd.Close()

	err = t.Execute(fd, map[string]template.JS{"Spec": specData})
	if err != nil {
		return nil
	}

	return nil
}