mirror of
https://github.com/ethauvin/kobalt.git
synced 2025-04-26 08:27:12 -07:00
Refactor.
This commit is contained in:
Cedric Beust
parent
f0ea553f55
commit
e5d03956f9
2 changed files with 2 additions and 391 deletions
|
|
@ -1,389 +0,0 @@
|
||||||
package com.beust.kobalt.wrapper;
|
|
||||||
|
|
||||||
import java.io.*;
|
|
||||||
import java.net.HttpURLConnection;
|
|
||||||
import java.net.URL;
|
|
||||||
import java.nio.file.*;
|
|
||||||
import java.util.*;
|
|
||||||
import java.util.zip.ZipEntry;
|
|
||||||
import java.util.zip.ZipException;
|
|
||||||
import java.util.zip.ZipFile;
|
|
||||||
|
|
||||||
//@com.beust.apt.processor.Version("1.3")
|
|
||||||
public class Main {
|
|
||||||
public static void main(String[] argv) throws IOException, InterruptedException {
|
|
||||||
new Main().installAndLaunchMain(argv);
|
|
||||||
}
|
|
||||||
|
|
||||||
private static final String KOBALT_PROPERTIES = "kobalt.properties";
|
|
||||||
private static final String KOBALTW = "kobaltw";
|
|
||||||
private static final String KOBALT_WRAPPER_PROPERTIES = "kobalt-wrapper.properties";
|
|
||||||
private static final String PROPERTY_VERSION = "kobalt.version";
|
|
||||||
private static final String PROPERTY_DOWNLOAD_URL = "kobalt.downloadUrl";
|
|
||||||
private static final String FILE_NAME = "kobalt";
|
|
||||||
private static final String DISTRIBUTIONS_DIR =
|
|
||||||
System.getProperty("user.home") + "/.kobalt/wrapper/dist";
|
|
||||||
|
|
||||||
private final Properties wrapperProperties = new Properties();
|
|
||||||
|
|
||||||
private static int logLevel = 1;
|
|
||||||
private boolean noOverwrite = false;
|
|
||||||
|
|
||||||
private void installAndLaunchMain(String[] argv) throws IOException, InterruptedException {
|
|
||||||
List<String> kobaltArgv = new ArrayList<>();
|
|
||||||
boolean noLaunch = false;
|
|
||||||
for (int i = 0; i < argv.length; i++) {
|
|
||||||
boolean passToKobalt = true;
|
|
||||||
switch(argv[i]) {
|
|
||||||
case "--noOverwrite":
|
|
||||||
noOverwrite = true;
|
|
||||||
passToKobalt = false;
|
|
||||||
break;
|
|
||||||
case "--noLaunch":
|
|
||||||
noLaunch = true;
|
|
||||||
break;
|
|
||||||
case "--log":
|
|
||||||
logLevel = Integer.parseInt(argv[i + 1]);
|
|
||||||
kobaltArgv.add(argv[i]);
|
|
||||||
i++;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
if (passToKobalt) {
|
|
||||||
kobaltArgv.add(argv[i]);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Path kobaltJarFile = installDistribution();
|
|
||||||
if (! noLaunch) {
|
|
||||||
launchMain(kobaltJarFile, kobaltArgv.toArray(new String[kobaltArgv.size()]));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void readProperties(Properties properties, InputStream ins) throws IOException {
|
|
||||||
properties.load(ins);
|
|
||||||
ins.close();
|
|
||||||
}
|
|
||||||
|
|
||||||
private Properties maybeCreateProperties() throws IOException {
|
|
||||||
Properties result = new Properties();
|
|
||||||
URL url = getClass().getClassLoader().getResource(KOBALT_PROPERTIES);
|
|
||||||
if (url != null) {
|
|
||||||
readProperties(result, url.openConnection().getInputStream());
|
|
||||||
} else {
|
|
||||||
File file = new File("src/main/resources/kobalt.properties");
|
|
||||||
if (file.exists()) {
|
|
||||||
FileInputStream fis = null;
|
|
||||||
try {
|
|
||||||
fis = new FileInputStream(file);
|
|
||||||
readProperties(result, fis);
|
|
||||||
} finally {
|
|
||||||
if (fis != null) fis.close();
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
throw new IllegalArgumentException("Couldn't find " + KOBALT_PROPERTIES);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
private File getWrapperDir() {
|
|
||||||
return new File("kobalt", "wrapper");
|
|
||||||
}
|
|
||||||
|
|
||||||
private static String downloadUrl(String version) {
|
|
||||||
return "http://beust.com/kobalt/kobalt-" + version + ".zip";
|
|
||||||
}
|
|
||||||
|
|
||||||
private void initWrapperFile(String version) throws IOException {
|
|
||||||
File config = new File(getWrapperDir(), KOBALT_WRAPPER_PROPERTIES);
|
|
||||||
if (! config.exists()) {
|
|
||||||
saveFile(config,
|
|
||||||
PROPERTY_VERSION + "=" + version + "\n"
|
|
||||||
// + PROPERTY_DOWNLOAD_URL + "=" + downloadUrl(version) + "\n"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
wrapperProperties.load(new FileReader(config));
|
|
||||||
}
|
|
||||||
|
|
||||||
private String getWrapperVersion() {
|
|
||||||
return wrapperProperties.getProperty(PROPERTY_VERSION);
|
|
||||||
}
|
|
||||||
|
|
||||||
private String getWrapperDownloadUrl(String version) {
|
|
||||||
String result = wrapperProperties.getProperty(PROPERTY_DOWNLOAD_URL);
|
|
||||||
if (result == null) {
|
|
||||||
result = downloadUrl(version);
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
private boolean isWindows() {
|
|
||||||
return System.getProperty("os.name").contains("Windows");
|
|
||||||
}
|
|
||||||
|
|
||||||
private Path installDistribution() throws IOException {
|
|
||||||
Properties properties = maybeCreateProperties();
|
|
||||||
|
|
||||||
String version = properties.getProperty(PROPERTY_VERSION);
|
|
||||||
initWrapperFile(version);
|
|
||||||
String wrapperVersion = getWrapperVersion();
|
|
||||||
|
|
||||||
log(2, "Wrapper version: " + wrapperVersion);
|
|
||||||
|
|
||||||
String fileName = FILE_NAME + "-" + wrapperVersion + ".zip";
|
|
||||||
Files.createDirectories(Paths.get(DISTRIBUTIONS_DIR));
|
|
||||||
Path localZipFile = Paths.get(DISTRIBUTIONS_DIR, fileName);
|
|
||||||
String zipOutputDir = DISTRIBUTIONS_DIR + "/" + wrapperVersion;
|
|
||||||
Path kobaltJarFile = Paths.get(zipOutputDir,
|
|
||||||
getWrapperDir().getPath() + "/" + FILE_NAME + "-" + wrapperVersion + ".jar");
|
|
||||||
if (! Files.exists(localZipFile) || ! Files.exists(kobaltJarFile)) {
|
|
||||||
download(localZipFile.toFile(), wrapperVersion);
|
|
||||||
}
|
|
||||||
|
|
||||||
//
|
|
||||||
// Extract all the zip files
|
|
||||||
//
|
|
||||||
if (! noOverwrite) {
|
|
||||||
int retries = 0;
|
|
||||||
while (retries < 2) {
|
|
||||||
try {
|
|
||||||
extractZipFile(localZipFile, zipOutputDir);
|
|
||||||
break;
|
|
||||||
} catch (ZipException e) {
|
|
||||||
retries++;
|
|
||||||
error("Couldn't open zip file " + localZipFile + ": " + e.getMessage());
|
|
||||||
error("The file is probably corrupt, downloading it again");
|
|
||||||
Files.delete(localZipFile);
|
|
||||||
download(localZipFile.toFile(), wrapperVersion);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//
|
|
||||||
// Copy the wrapper files in the current kobalt/wrapper directory
|
|
||||||
//
|
|
||||||
if (! noOverwrite) {
|
|
||||||
log(2, "Copying the wrapper files");
|
|
||||||
for (String file : FILES) {
|
|
||||||
Path to = Paths.get(new File(".").getAbsolutePath(), file);
|
|
||||||
if (Files.exists(to)) {
|
|
||||||
log(1, to + " already exists, not overwriting it");
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
boolean extractFile = true;
|
|
||||||
if (file.equals(KOBALTW)) {
|
|
||||||
//
|
|
||||||
// For kobaltw: try to generate it with the correct env shebang. If this fails,
|
|
||||||
// we'll extract the one from the zip file
|
|
||||||
//
|
|
||||||
File envFile = new File("/bin/env");
|
|
||||||
if (!envFile.exists()) {
|
|
||||||
envFile = new File("/usr/bin/env");
|
|
||||||
}
|
|
||||||
if (envFile.exists()) {
|
|
||||||
log(2, "Located " + envFile.getAbsolutePath() + ", generating " + file);
|
|
||||||
String content = "#!" + envFile.getAbsolutePath() + " bash\n"
|
|
||||||
+ "java -jar $(dirname $0)/kobalt/wrapper/kobalt-wrapper.jar $*\n";
|
|
||||||
Files.write(to, content.getBytes());
|
|
||||||
extractFile = false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (extractFile) {
|
|
||||||
Path from = Paths.get(zipOutputDir, file);
|
|
||||||
try {
|
|
||||||
if (isWindows() && to.toFile().exists()) {
|
|
||||||
log(1, "Windows detected, not overwriting " + to);
|
|
||||||
} else {
|
|
||||||
Files.copy(from, to, StandardCopyOption.REPLACE_EXISTING);
|
|
||||||
}
|
|
||||||
} catch (IOException ex) {
|
|
||||||
log(1, "Couldn't copy " + from + " to " + to + ": " + ex.getMessage());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!new File(KOBALTW).setExecutable(true)) {
|
|
||||||
if (!isWindows()) {
|
|
||||||
log(1, "Couldn't make " + KOBALTW + " executable");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return kobaltJarFile;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Extract the zip file in ~/.kobalt/wrapper/dist/$version
|
|
||||||
*/
|
|
||||||
private void extractZipFile(Path localZipFile, String zipOutputDir) throws IOException {
|
|
||||||
log(2, "Extracting " + localZipFile);
|
|
||||||
try (ZipFile zipFile = new ZipFile(localZipFile.toFile())) {
|
|
||||||
Enumeration<? extends ZipEntry> entries = zipFile.entries();
|
|
||||||
while (entries.hasMoreElements()) {
|
|
||||||
ZipEntry entry = entries.nextElement();
|
|
||||||
Path entryPath = Paths.get(zipOutputDir, entry.getName());
|
|
||||||
|
|
||||||
if (entry.isDirectory()) {
|
|
||||||
Files.createDirectories(entryPath);
|
|
||||||
} else {
|
|
||||||
log(2, " Writing " + entry.getName() + " to " + entryPath);
|
|
||||||
try {
|
|
||||||
Files.createDirectories(entryPath.getParent());
|
|
||||||
Files.copy(zipFile.getInputStream(entry), entryPath, StandardCopyOption.REPLACE_EXISTING);
|
|
||||||
} catch (FileSystemException ex) {
|
|
||||||
log(2, "Couldn't copy to " + entryPath);
|
|
||||||
throw new IOException(ex);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private static final String[] FILES = new String[] { KOBALTW, "kobalt/wrapper/" + FILE_NAME + "-wrapper.jar" };
|
|
||||||
|
|
||||||
private void download(File file, String version) throws IOException {
|
|
||||||
for (int attempt = 0; attempt < 3; ++attempt) {
|
|
||||||
try {
|
|
||||||
downloadImpl(file, version);
|
|
||||||
} catch (IOException e) {
|
|
||||||
error("Failed to download file " + file + " due to I/O issue: " + e.getMessage());
|
|
||||||
Files.deleteIfExists(file.toPath());
|
|
||||||
|
|
||||||
if (attempt == 2) {
|
|
||||||
throw e;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (file.exists()) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void downloadImpl(File file, String version) throws IOException {
|
|
||||||
String fileUrl = getWrapperDownloadUrl(version);
|
|
||||||
|
|
||||||
log(2, "Downloading " + fileUrl);
|
|
||||||
|
|
||||||
boolean done = false;
|
|
||||||
HttpURLConnection httpConn = null;
|
|
||||||
try {
|
|
||||||
int responseCode = 0;
|
|
||||||
URL url = null;
|
|
||||||
while (!done) {
|
|
||||||
url = new URL(fileUrl);
|
|
||||||
httpConn = (HttpURLConnection) url.openConnection();
|
|
||||||
responseCode = httpConn.getResponseCode();
|
|
||||||
if (responseCode == HttpURLConnection.HTTP_MOVED_TEMP ||
|
|
||||||
responseCode == HttpURLConnection.HTTP_MOVED_PERM) {
|
|
||||||
fileUrl = httpConn.getHeaderField("Location");
|
|
||||||
} else {
|
|
||||||
done = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// always check HTTP response code first
|
|
||||||
if (responseCode == HttpURLConnection.HTTP_OK) {
|
|
||||||
String fileName = "";
|
|
||||||
String disposition = httpConn.getHeaderField("Content-Disposition");
|
|
||||||
String contentType = httpConn.getContentType();
|
|
||||||
int contentLength = httpConn.getContentLength();
|
|
||||||
|
|
||||||
if (disposition != null) {
|
|
||||||
// extracts file name from header field
|
|
||||||
int index = disposition.indexOf("filename=");
|
|
||||||
if (index > 0) {
|
|
||||||
fileName = disposition.substring(index + 9, disposition.length());
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// extracts file name from URL
|
|
||||||
fileName = fileUrl.substring(fileUrl.lastIndexOf("/") + 1,
|
|
||||||
fileUrl.length());
|
|
||||||
}
|
|
||||||
|
|
||||||
log(2, "Content-Type = " + contentType);
|
|
||||||
log(2, "Content-Disposition = " + disposition);
|
|
||||||
log(2, "Content-Length = " + contentLength);
|
|
||||||
log(2, "fileName = " + fileName);
|
|
||||||
|
|
||||||
// opens input stream from the HTTP connection
|
|
||||||
try (InputStream inputStream = httpConn.getInputStream()) {
|
|
||||||
// opens an output stream to save into file
|
|
||||||
try (FileOutputStream outputStream = new FileOutputStream(file)) {
|
|
||||||
copyToStreamWithProgress(inputStream, outputStream, contentLength, url.toString());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
log(1, "Downloaded " + fileUrl);
|
|
||||||
} else {
|
|
||||||
error("No file to download. Server replied HTTP code: " + responseCode);
|
|
||||||
}
|
|
||||||
} finally {
|
|
||||||
if (httpConn != null) {
|
|
||||||
httpConn.disconnect();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void copyToStreamWithProgress(InputStream inputStream, OutputStream outputStream, long contentLength,
|
|
||||||
String url) throws IOException {
|
|
||||||
int bytesRead;
|
|
||||||
long bytesSoFar = 0;
|
|
||||||
byte[] buffer = new byte[100_000];
|
|
||||||
while ((bytesRead = inputStream.read(buffer)) != -1) {
|
|
||||||
outputStream.write(buffer, 0, bytesRead);
|
|
||||||
bytesSoFar += bytesRead;
|
|
||||||
if (bytesRead > 0) {
|
|
||||||
if (contentLength > 0) {
|
|
||||||
float percent = bytesSoFar * 100 / contentLength;
|
|
||||||
log2(1, "\rDownloading " + url + " " + percent + "%");
|
|
||||||
} else {
|
|
||||||
log2(1, ".");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
log2(1, "\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
private void saveFile(File file, String text) throws IOException {
|
|
||||||
Files.createDirectories(file.getAbsoluteFile().toPath().getParent());
|
|
||||||
Files.deleteIfExists(file.toPath());
|
|
||||||
log(2, "Wrote " + file);
|
|
||||||
Files.write(Paths.get(file.toURI()), text.getBytes());
|
|
||||||
}
|
|
||||||
|
|
||||||
static void log2(int level, String s) {
|
|
||||||
p(level, s, false);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void log(int level, String s) {
|
|
||||||
p(level, "[Wrapper] " + s, true);
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void p(int level, String s, boolean newLine) {
|
|
||||||
if (level <= logLevel) {
|
|
||||||
if (newLine) System.out.println(s);
|
|
||||||
else System.out.print(s);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void error(String s) {
|
|
||||||
System.out.println("[Wrapper error] *** " + s);
|
|
||||||
}
|
|
||||||
|
|
||||||
private void launchMain(Path kobaltJarFile, String[] argv) throws IOException, InterruptedException {
|
|
||||||
List<String> args = new ArrayList<>();
|
|
||||||
args.add("java");
|
|
||||||
args.add("-jar");
|
|
||||||
args.add(kobaltJarFile.toFile().getAbsolutePath());
|
|
||||||
Collections.addAll(args, argv);
|
|
||||||
|
|
||||||
ProcessBuilder pb = new ProcessBuilder(args);
|
|
||||||
pb.inheritIO();
|
|
||||||
log(2, "Launching " + args);
|
|
||||||
Process process = pb.start();
|
|
||||||
process.waitFor();
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
@ -3,6 +3,7 @@ package com.beust.kobalt
|
||||||
import com.beust.kobalt.api.*
|
import com.beust.kobalt.api.*
|
||||||
import com.beust.kobalt.misc.KFiles
|
import com.beust.kobalt.misc.KFiles
|
||||||
import com.beust.kobalt.misc.log
|
import com.beust.kobalt.misc.log
|
||||||
|
import com.beust.kobalt.plugin.android.AndroidFiles
|
||||||
import java.io.File
|
import java.io.File
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
@ -79,8 +80,7 @@ class Variant(val initialProductFlavor: ProductFlavorConfig? = null,
|
||||||
* If either the Project or the current variant has a build config defined, generate BuildConfig.java
|
* If either the Project or the current variant has a build config defined, generate BuildConfig.java
|
||||||
*/
|
*/
|
||||||
fun maybeGenerateBuildConfig(project: Project, context: KobaltContext) {
|
fun maybeGenerateBuildConfig(project: Project, context: KobaltContext) {
|
||||||
fun generated(project: Project) = KFiles.joinDir(project.directory, project.buildDirectory, "generated",
|
fun generated(project: Project) = KFiles.joinDir(AndroidFiles.generated(project), "source")
|
||||||
"source")
|
|
||||||
|
|
||||||
if (project.buildConfig != null || context.variant.hasBuildConfig) {
|
if (project.buildConfig != null || context.variant.hasBuildConfig) {
|
||||||
val buildConfigs = arrayListOf<BuildConfig>()
|
val buildConfigs = arrayListOf<BuildConfig>()
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue