1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.mybatis.generator.internal;
17
18 import static org.mybatis.generator.internal.util.messages.Messages.getString;
19
20 import java.io.File;
21 import java.io.IOException;
22 import java.nio.file.Files;
23 import java.nio.file.Path;
24 import java.util.StringTokenizer;
25
26 import org.mybatis.generator.api.ShellCallback;
27 import org.mybatis.generator.exception.ShellException;
28
29 public class DefaultShellCallback implements ShellCallback {
30
31 public DefaultShellCallback() {
32 super();
33 }
34
35 @Override
36 public File getDirectory(String targetProject, String targetPackage) throws ShellException {
37
38
39
40
41
42
43 Path targetProjectDirectory = Path.of(targetProject);
44 if (!Files.isDirectory(targetProjectDirectory)) {
45 throw new ShellException(getString("Warning.9",
46 targetProject));
47 }
48
49 StringBuilder sb = new StringBuilder();
50 StringTokenizer st = new StringTokenizer(targetPackage, ".");
51 while (st.hasMoreTokens()) {
52 sb.append(st.nextToken());
53 sb.append(File.separatorChar);
54 }
55
56 Path directory = targetProjectDirectory.resolve(sb.toString());
57 if (!Files.isDirectory(directory)) {
58 try {
59 Files.createDirectories(directory);
60 } catch (IOException e) {
61 throw new ShellException(getString("Warning.10",
62 directory.toFile().getAbsolutePath()));
63 }
64 }
65
66 return directory.toFile();
67 }
68 }