☰

knife4j接口导出

try to take over the world!

// ==UserScript==
// @name         knife4j接口导出
// @namespace    http://tampermonkey.net/
// @version      0.0.11
// @description  try to take over the world!
// @author       DoveAz
// @match        http://172.16.10.*:*/*/doc.html
// @grant        GM_setClipboard
// @grant        GM_getResourceText
// @grant        GM_addStyle
// @require      https://cdn.jsdelivr.net/npm/[email protected]/dist/jquery.min.js
// @require      https://cdn.jsdelivr.net/npm/[email protected]/lodash.min.js
// @resource     IMPORTED_CSS  https://cdn.jsdelivr.net/npm/[email protected]/dist/theme/default/layer.css
// @require      https://cdn.jsdelivr.net/npm/[email protected]/src/layer.min.js
// ==/UserScript==

(function() {
    'use strict';
    const my_css = GM_getResourceText("IMPORTED_CSS");
    GM_addStyle(my_css);
    knife4jGeneratate()



    function knife4jGeneratate() {
        if(!localStorage.getItem('moduleReplace')){
            localStorage.setItem('moduleReplace','/.*cim6d-(.*)-/,,system')
        }

        $('.knife4j-header-default .right').prepend(
            `
        <button type="button" class="ant-btn ant-btn-primary generate">复制接口</button>
        <button type="button" class="ant-btn ant-btn-primary ant-btn-background-ghost  generateModule">复制本模块</button>
        <button type="button" class="ant-btn ant-btn-info settings">设置</button>
        `
    )
        $('body').append({})

        $('.generate').click(function () {
            const documentEl = document.querySelector('.ant-tabs-tabpane-active .knife4j-body-content .document')
            const vm = documentEl.__vue__
            const api = vm.api
            const result = generateAxios(api)
            GM_setClipboard(result)
        })

        $('.generateModule').click(function () {
            const documentEl = document.querySelector('.ant-tabs-tabpane-active .knife4j-body-content .document')
            const vm = documentEl.__vue__
            const targetMenuList = _.find(vm.swaggerInstance.tags, { name: vm.$route.params.controller }).childrens
            const result = targetMenuList.map(generateAxios).join('\n\n')
            GM_setClipboard(result)
        })

        $('.settings').click(function(){
            const values = localStorage.getItem('moduleReplace')?localStorage.getItem('moduleReplace').split(',,') : []

            const settingsLayer = layer.open({
                title:'设置模块名 (正则替换)',
                type: 1,
                area: ['600px', '280px'], //宽高
                btn: ['确定'],
                shadeClose :true,
                yes: function(index, layero){
                    //按钮【按钮一】的回调
                    const values = $('.module-name-settings input').map(function(){
                        return $(this).val()
                    }).get()
                    localStorage.setItem('moduleReplace',values.join(',,'))
                    layer.close(index)
                }

                ,cancel: function(){
                    //右上角关闭回调

                    //return false 开启该代码可禁止点击该按钮关闭
                },
                content: `
<div class="module-name-settings">
<div style="display:flex;align-items: center;  flex-direction: column;;padding:20px"> <input class="ant-input" placeholder="正则" value="${values[0]}" /><div style="margin:10px 0;">替换为</div><input class="ant-input" value="${values[1]}" />  </div>
</div>
`
            });
            $('.settings-layer-close').click(function(){
                settingsLayer.close()
            })

        })

        function generateAxios(api) {
            const params = api.parameters
            const hasBody = !!_.find(params, { in: 'body' })
            const bodyParams = _.filter(params, { in: 'body' })
            const bodyName = hasBody && bodyParams[0].type === 'array' ? bodyParams[0].name :'body'
            return `/**
* ${api.tags[0]}-${api.summary}
${calculateDocParams(api.parameters)}
*/
export function ${calculateFunctionName(api.showUrl,api.parameters)}(${calculateFunctionParam(api.parameters)}) {
    return ${calculateRequestModule(api.showUrl)}.${_.toLower(api.methodType)}(\`${calculateRequestUrl(api.showUrl)}\`${hasBody?', '+bodyName:''})
}`.replace(/\n[\n]+/g, "\n")

        }

        function calculateFunctionName(url, params) {
            const actionHash = {
                get: 'get',
                post: 'create',
                add: 'create',
                create: 'create',
                save: 'save',

                update: 'update',

                delete: 'remove',
                batchDelete: 'batchDelete',
                del: 'remove',

                list: 'list',
                listAll: 'listAll'
            }
            const lastWithoutPathParams = _.last(url.split('/').filter(string => !/\{.*\}/.test(string)))
            const secondToLst = _.nth(url.split('/').filter(string => !/\{.*\}/.test(string)),-2)
            const action = actionHash[lastWithoutPathParams]
            let result = ''
            if (_.includes(action, 'list')) {
                result = `get${_.upperFirst(secondToLst)}${_.upperFirst(action)}`
            } else if (action) {
                result = action + _.upperFirst(secondToLst)
            } else {
                result = lastWithoutPathParams
            }

            const pathParams = _.filter(params, { in: 'path' })
            if(pathParams.length===1){
                result += 'By'+ _.upperFirst(pathParams[0].name)
            }
            return result
        }

        function calculateFunctionParam(params) {
            const result = []
            const pathParams = _.filter(params, { in: 'path' })
            const bodyParams = _.filter(params, { in: 'body' })

            pathParams.forEach(param => {
                result.push(param.name)
            })

            if (bodyParams.length) {
                if(bodyParams[0].type==='array'){
                    result.push(bodyParams[0].name +  ' = []')
                }else{
                    result.push('body = {}')
                }
            }

            return result.join(', ')
        }

        function calculateRequestModule(url) {
            const [value1,value2] = localStorage.getItem('moduleReplace').split(',,')
            return value2
        }

        function calculateRequestUrl(url) {
            const [value1,value2] = localStorage.getItem('moduleReplace').split(',,')
            url = url.replace(new RegExp(value1),'')
            return url.replaceAll('{', '${')
        }

        function calculateDocParams(params) {
            const result = []
            const pathParams = _.filter(params, { in: 'path' })
            const bodyParams = _.filter(params, { in: 'body' })

            pathParams.forEach(param => {
                result.push(`* @param {${paramsTypeConvert(param.type)}} ${param.name} ${param.description}`)
            })

            if (bodyParams.length) {
                if(bodyParams[0].type==='array'){
                    result.push(`* @param {Array} ${bodyParams[0].name}`)
                }else{
                    result.push('* @param {Object} body')
                }

            }

            return result.join('\n')
        }

        function paramsTypeConvert(type){
            const hash = {
                string: 'String',
                'integer(int64)': 'Number',
                'integer(int32)': 'Number',
            }
            if(hash[type]){
                return hash[type]
            }else{
                return type
            }
        }

    }
})();