FilesExpand file tree

 v0.6.14

/

code.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

189 lines (172 loc) · 6.69 KB

 v0.6.14

/

code.java

Top

File metadata and controls

• Code
• Blame

executable file

·

189 lines (172 loc) · 6.69 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

/*

* Copyright 2023 Red Hat

*

* 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

*

* https://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.

*/

/// usr/bin/env jbang "$0" "$@" ; exit $?

//JAVA 25

static class SourceSpec {

final String globSuffix;

final Path relativeDir;

SourceSpec(String globSuffix, Path relativeDir) {

this.globSuffix = globSuffix;

this.relativeDir = relativeDir;

}

}

static final String PACKAGE_NAME = "org.patternfly.showcase";

static final String CLASS_NAME = "Code";

static final String TARGET = "src/main/java";

static final String START_COMMENT = "// @code-start:";

static final String END_COMMENT = "// @code-end:";

static final List<SourceSpec> SOURCES = Arrays.asList(

new SourceSpec("*Component.java", Paths.get("src/main/java/org/patternfly/showcase/component")),

new SourceSpec("*Extension.java", Paths.get("src/main/java/org/patternfly/showcase/extension")),

new SourceSpec("*Layout.java", Paths.get("src/main/java/org/patternfly/showcase/layout")),

new SourceSpec("*Chart.java", Paths.get("src/main/java/org/patternfly/showcase/chart"))

);

void main(String... args) throws IOException {

if (args.length == 0 || args[0] == null || args[0].isEmpty()) {

System.err.println("Missing base directory!");

System.exit(1);

}

Path base = Paths.get(args[0]).toAbsolutePath().normalize();

Path targetPath = base.resolve(TARGET).resolve(PACKAGE_NAME.replace('.', '/'));

Files.createDirectories(targetPath);

File javaSource = targetPath.resolve(CLASS_NAME + ".java").toFile();

if (javaSource.exists() && !javaSource.delete()) {

throw new IOException("Unable to delete existing file: " + javaSource);

}

StringBuilder out = new StringBuilder();

startClass(out);

int counter = processSnippets(base, out);

endClass(out);

Files.write(javaSource.toPath(), out.toString().getBytes(StandardCharsets.UTF_8));

IO.println("Processed " + counter + " code snippets");

}

void startClass(StringBuilder out) {

out.append("package ").append(PACKAGE_NAME).append(";\n\n")

.append("import java.util.Map;\n")

.append("import java.util.HashMap;\n\n")

.append("import javax.annotation.processing.Generated;\n\n")

.append("/*\n")

.append(" * WARNING! This class is generated. Do not modify.\n")

.append(" */\n")

.append("@Generated(\"code.java\")\n")

.append("public final class ").append(CLASS_NAME).append(" {\n\n")

.append(" private static final Map<String, String> snippets = new HashMap<>();\n\n")

.append(" static {");

}

int processSnippets(Path base, StringBuilder out) throws IOException {

int counter = 0;

for (SourceSpec spec : SOURCES) {

Path dir = base.resolve(spec.relativeDir);

if (!Files.isDirectory(dir)) {

continue;

}

final Predicate<String> nameMatches = globPredicate(spec.globSuffix);

List<Path> files = new ArrayList<>();

try (Stream<Path> stream = Files.list(dir)) {

stream.filter(p -> Files.isRegularFile(p) && nameMatches.test(p.getFileName().toString()))

.forEach(files::add);

}

files.sort(java.util.Comparator.naturalOrder());

for (Path p : files) {

counter += extractFromFile(p, out);

}

}

return counter;

}

int extractFromFile(Path file, StringBuilder out) throws IOException {

String name = "";

boolean collectCode = false;

int leadingWhitespace = 0;

List<String> code = new ArrayList<>();

int counter = 0;

try (BufferedReader reader = new BufferedReader(

new InputStreamReader(new FileInputStream(file.toFile()), StandardCharsets.UTF_8))) {

String line;

while ((line = reader.readLine()) != null) {

if (line.contains(START_COMMENT)) {

leadingWhitespace = line.indexOf(START_COMMENT);

name = line.substring(line.indexOf(START_COMMENT) + START_COMMENT.length());

collectCode = true;

} else if (collectCode) {

if (line.contains(END_COMMENT + name)) {

addSnippet(out, name, code);

counter++;

// reset

name = "";

collectCode = false;

leadingWhitespace = 0;

code.clear();

} else {

if (line.length() < leadingWhitespace) {

code.add(line);

} else {

code.add(line.substring(leadingWhitespace));

}

}

}

}

}

return counter;

}

void addSnippet(StringBuilder out, String name, List<String> code) {

out.append("\n snippets.put(\"")

.append(escapeJava(name))

.append("\", \"")

.append(escapeJava(String.join("\n", code)))

.append("\");");

}

void endClass(StringBuilder out) {

out.append("\n }\n\n")

.append(" public static String code(String name) {\n")

.append(" return snippets.getOrDefault(name, \"n/a\");\n")

.append(" }\n")

.append("}\n");

}

String escapeJava(String in) {

if (in == null) {return "";}

StringBuilder sb = new StringBuilder(in.length());

for (int i = 0; i < in.length(); i++) {

char c = in.charAt(i);

switch (c) {

case '"':

sb.append("\\\"");

break;

case '\\':

sb.append("\\\\");

break;

case '\r':

sb.append("\\r");

break;

case '\n':

sb.append("\\n");

break;

case '\t':

sb.append("\\t");

break;

default:

sb.append(c);

}

}

return sb.toString();

}

Predicate<String> globPredicate(String globSuffix) {

// globSuffix like "*Component.java" -> endsWith("Component.java")

if (globSuffix.startsWith("*")) {

String suffix = globSuffix.substring(1);

return name -> name.endsWith(suffix);

}

return name -> name.equals(globSuffix);

}