1 /* 2 * Copyright (C) 2021 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.csuite.core; 18 19 import com.android.tradefed.config.Option; 20 import com.android.tradefed.config.Option.Importance; 21 22 import com.google.common.annotations.VisibleForTesting; 23 import com.google.common.base.Preconditions; 24 25 import java.io.File; 26 import java.io.IOException; 27 import java.nio.file.Files; 28 import java.util.AbstractMap; 29 import java.util.ArrayList; 30 import java.util.HashSet; 31 import java.util.List; 32 import java.util.Map; 33 import java.util.Set; 34 import java.util.stream.Collectors; 35 import java.util.stream.Stream; 36 37 /** Accepts files that contains template mapping entries. */ 38 public class FileBasedTemplateMappingProvider implements TemplateMappingProvider { 39 @VisibleForTesting static final String TEMPLATE_MAPPING_FILE_OPTION = "template-mapping-file"; 40 41 @VisibleForTesting static final String COMMENT_LINE_PREFIX = "#"; 42 @VisibleForTesting static final String MODULE_TEMPLATE_SEPARATOR = " "; 43 44 @Option( 45 name = TEMPLATE_MAPPING_FILE_OPTION, 46 description = "Template mapping file paths.", 47 importance = Importance.NEVER) 48 private Set<File> mTemplateMappingFiles = new HashSet<>(); 49 50 @Override get()51 public Stream<Map.Entry<String, String>> get() throws IOException { 52 List<Map.Entry<String, String>> entries = new ArrayList<>(); 53 54 // Using for loop instead of stream here so that exceptions can be caught early. 55 for (File file : mTemplateMappingFiles) { 56 List<String> lines = 57 Files.readAllLines(file.toPath()).stream() 58 .map(String::trim) 59 .filter(FileBasedTemplateMappingProvider::isNotCommentLine) 60 .distinct() 61 .collect(Collectors.toList()); 62 for (String line : lines) { 63 String[] pair = line.split(MODULE_TEMPLATE_SEPARATOR); 64 Preconditions.checkArgument( 65 pair.length == 2, "Unrecognized template map format " + line); 66 entries.add(new AbstractMap.SimpleEntry<>(pair[0].trim(), pair[1].trim())); 67 } 68 } 69 70 return entries.stream(); 71 } 72 isNotCommentLine(String text)73 private static boolean isNotCommentLine(String text) { 74 // Check the text is not an empty string and not a comment line. 75 return !text.isEmpty() && !text.startsWith(COMMENT_LINE_PREFIX); 76 } 77 } 78