FilesExpand file tree

 master

/

controller.js

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

189 lines (172 loc) · 8.32 KB

 master

/

controller.js

Copy path

Top

File metadata and controls

• Code
• Blame

189 lines (172 loc) · 8.32 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

const baseController = require('controllers/base.js');

const interfaceModel = require('models/interface.js');

const projectModel = require('models/project.js');

// const wikiModel = require('../yapi-plugin-wiki/wikiModel.js');

const interfaceCatModel = require('models/interfaceCat.js');

const yapi = require('yapi.js');

const markdownIt = require('markdown-it');

const markdownItAnchor = require('markdown-it-anchor');

const markdownItTableOfContents = require('markdown-it-table-of-contents');

const defaultTheme = require('./defaultTheme.js');

const md = require('../../common/markdown');

// const htmlToPdf = require("html-pdf");

class exportController extends baseController {

constructor(ctx) {

super(ctx);

this.catModel = yapi.getInst(interfaceCatModel);

this.interModel = yapi.getInst(interfaceModel);

this.projectModel = yapi.getInst(projectModel);

}

async handleListClass(pid, status) {

let result = await this.catModel.list(pid),

newResult = [];

for (let i = 0, item, list; i < result.length; i++) {

item = result[i].toObject();

list = await this.interModel.listByInterStatus(item._id, status);

list = list.sort((a, b) => {

return a.index - b.index;

});

if (list.length > 0) {

item.list = list;

newResult.push(item);

}

}

return newResult;

}

handleExistId(data) {

function delArrId(arr, fn) {

if (!Array.isArray(arr)) return;

arr.forEach(item => {

delete item._id;

delete item.__v;

delete item.uid;

delete item.edit_uid;

delete item.catid;

delete item.project_id;

if (typeof fn === 'function') fn(item);

});

}

delArrId(data, function(item) {

delArrId(item.list, function(api) {

delArrId(api.req_body_form);

delArrId(api.req_params);

delArrId(api.req_query);

delArrId(api.req_headers);

if (api.query_path && typeof api.query_path === 'object') {

delArrId(api.query_path.params);

}

});

});

return data;

}

async exportData(ctx) {

let pid = ctx.request.query.pid;

let type = ctx.request.query.type;

let status = ctx.request.query.status;

let isWiki = ctx.request.query.isWiki;

if (!pid) {

ctx.body = yapi.commons.resReturn(null, 200, 'pid 不为空');

}

let curProject, wikiData;

let tp = '';

try {

curProject = await this.projectModel.get(pid);

if (isWiki === 'true') {

const wikiModel = require('../yapi-plugin-wiki/wikiModel.js');

wikiData = await yapi.getInst(wikiModel).get(pid);

}

ctx.set('Content-Type', 'application/octet-stream');

const list = await this.handleListClass(pid, status);

switch (type) {

case 'markdown': {

tp = await createMarkdown.bind(this)(list, false);

ctx.set('Content-Disposition', `attachment; filename=api.md`);

return (ctx.body = tp);

}

case 'json': {

let data = this.handleExistId(list);

tp = JSON.stringify(data, null, 2);

ctx.set('Content-Disposition', `attachment; filename=api.json`);

return (ctx.body = tp);

}

default: {

//默认为html

tp = await createHtml.bind(this)(list);

ctx.set('Content-Disposition', `attachment; filename=api.html`);

return (ctx.body = tp);

}

}

} catch (error) {

yapi.commons.log(error, 'error');

ctx.body = yapi.commons.resReturn(null, 502, '下载出错');

}

async function createHtml(list) {

let md = await createMarkdown.bind(this)(list, true);

let markdown = markdownIt({ html: true, breaks: true });

markdown.use(markdownItAnchor); // Optional, but makes sense as you really want to link to something

markdown.use(markdownItTableOfContents, {

markerPattern: /^\[toc\]/im

});

// require('fs').writeFileSync('./a.markdown', md);

let tp = unescape(markdown.render(md));

// require('fs').writeFileSync('./a.html', tp);

let left;

// console.log('tp',tp);

let content = tp.replace(

/<div\s+?class="table-of-contents"\s*>[\s\S]*?<\/ul>\s*<\/div>/gi,

function(match) {

left = match;

return '';

}

);

return createHtml5(left || '', content);

}

function createHtml5(left, tp) {

//html5模板

let html = `<!DOCTYPE html>

<html>

<head>

<title>${curProject.name}</title>

<meta charset="utf-8" />

${defaultTheme}

</head>

<body>

<div class="m-header">

<a href="#" style="display: inherit;"><svg class="svg" width="32px" height="32px" viewBox="0 0 64 64" version="1.1"><title>Icon</title><desc>Created with Sketch.</desc><defs><linearGradient x1="50%" y1="0%" x2="50%" y2="100%" id="linearGradient-1"><stop stop-color="#FFFFFF" offset="0%"></stop><stop stop-color="#F2F2F2" offset="100%"></stop></linearGradient><circle id="path-2" cx="31.9988602" cy="31.9988602" r="2.92886048"></circle><filter x="-85.4%" y="-68.3%" width="270.7%" height="270.7%" filterUnits="objectBoundingBox" id="filter-3"><feOffset dx="0" dy="1" in="SourceAlpha" result="shadowOffsetOuter1"></feOffset><feGaussianBlur stdDeviation="1.5" in="shadowOffsetOuter1" result="shadowBlurOuter1"></feGaussianBlur><feColorMatrix values="0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0.159703351 0" type="matrix" in="shadowBlurOuter1"></feColorMatrix></filter></defs><g id="首页" stroke="none" stroke-width="1" fill="none" fill-rule="evenodd"><g id="大屏幕"><g id="Icon"><circle id="Oval-1" fill="url(#linearGradient-1)" cx="32" cy="32" r="32"></circle><path d="M36.7078009,31.8054514 L36.7078009,51.7110548 C36.7078009,54.2844537 34.6258634,56.3695395 32.0579205,56.3695395 C29.4899777,56.3695395 27.4099998,54.0704461 27.4099998,51.7941246 L27.4099998,31.8061972 C27.4099998,29.528395 29.4909575,27.218453 32.0589004,27.230043 C34.6268432,27.241633 36.7078009,29.528395 36.7078009,31.8054514 Z" id="blue" fill="#2359F1" fill-rule="nonzero"></path><path d="M45.2586091,17.1026914 C45.2586091,17.1026914 45.5657231,34.0524383 45.2345291,37.01141 C44.9033351,39.9703817 43.1767091,41.6667796 40.6088126,41.6667796 C38.040916,41.6667796 35.9609757,39.3676862 35.9609757,37.0913646 L35.9609757,17.1034372 C35.9609757,14.825635 38.0418959,12.515693 40.6097924,12.527283 C43.177689,12.538873 45.2586091,14.825635 45.2586091,17.1026914 Z" id="green" fill="#57CF27" fill-rule="nonzero" transform="translate(40.674608, 27.097010) rotate(60.000000) translate(-40.674608, -27.097010) "></path><path d="M28.0410158,17.0465598 L28.0410158,36.9521632 C28.0410158,39.525562 25.9591158,41.6106479 23.3912193,41.6106479 C20.8233227,41.6106479 18.7433824,39.3115545 18.7433824,37.035233 L18.7433824,17.0473055 C18.7433824,14.7695034 20.8243026,12.4595614 23.3921991,12.4711513 C25.9600956,12.4827413 28.0410158,14.7695034 28.0410158,17.0465598 Z" id="red" fill="#FF561B" fill-rule="nonzero" transform="translate(23.392199, 27.040878) rotate(-60.000000) translate(-23.392199, -27.040878) "></path><g id="inner-round"><use fill="black" fill-opacity="1" filter="url(#filter-3)" xmlns:xlink="http://www.w3.org/1999/xlink" xlink:href="#path-2"></use><use fill="#F7F7F7" fill-rule="evenodd" xmlns:xlink="http://www.w3.org/1999/xlink" xlink:href="#path-2"></use></g></g></g></g></svg></a>

<a href="#"><h1 class="title">YAPI 接口文档</h1></a>

<div class="nav">

<a href="https://yapi.ymfe.org/">YApi</a>

</div>

</div>

<div class="g-doc">

${left}

<div id="right" class="content-right">

${tp}

<footer class="m-footer">

<p>Build by <a href="https://ymfe.org/">YMFE</a>.</p>

</footer>

</div>

</div>

</body>

</html>

`;

return html;

}

function createMarkdown(list, isToc) {

//拼接markdown

//模板

let mdTemplate = ``;

try {

// 项目名称信息

mdTemplate += md.createProjectMarkdown(curProject, wikiData);

// 分类信息

mdTemplate += md.createClassMarkdown(curProject, list, isToc);

return mdTemplate;

} catch (e) {

yapi.commons.log(e, 'error');

ctx.body = yapi.commons.resReturn(null, 502, '下载出错');

}

}

}

}

module.exports = exportController;