summaryrefslogtreecommitdiffstats
path: root/jill/src/com/android/jill/Jill.java
blob: f32dd182a3f62c8ee96b0de89ab8560c7d92f525 (plain)
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
/*
 * Copyright (C) 2013 The Android Open Source Project
 *
 * 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
 *
 *      http://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.
 */

package com.android.jill;

import com.android.jill.frontend.java.JavaTransformer;
import com.android.jill.utils.FileUtils;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.jar.JarFile;

import javax.annotation.Nonnull;

/**
 * Main class of Jill.
 */
public class Jill {

  @Nonnull
  private final Options options;

  @Nonnull
  private final String version;

  public Jill(@Nonnull Options options, @Nonnull String version) {
    this.options = options;
    this.version = version;
  }

  public void process(@Nonnull File binaryFile) {
    if (binaryFile.isFile()) {
      if (FileUtils.isJavaBinaryFile(binaryFile)) {
          processJavaBinary(binaryFile);
      } else if (FileUtils.isJarFile(binaryFile)) {
          try {
            processJarFile(new JarFile(binaryFile));
          } catch (IOException e) {
            throw new JillException("Fails to create jar file " + binaryFile.getName(), e);
          }
      } else {
        throw new JillException("Unsupported file type: " + binaryFile.getName());
      }
    } else {
      processFolder(binaryFile);
    }
  }

  private void processJavaBinary(@Nonnull File javaBinaryFile) {
    assert javaBinaryFile.isFile();
    List<File> javaBinaryFiles = new ArrayList<File>();
    javaBinaryFiles.add(javaBinaryFile);
    new JavaTransformer(version, options).transform(javaBinaryFiles);
  }

  private void processJarFile(@Nonnull JarFile jarFile) {
    new JavaTransformer(version, options).transform(jarFile);
  }

  private void processFolder(@Nonnull File folder) {
    assert folder.isDirectory();
    List<File> javaBinaryFiles = new ArrayList<File>();
    FileUtils.getJavaBinaryFiles(folder, javaBinaryFiles);
    new JavaTransformer(version, options).transform(javaBinaryFiles);
  }

}