View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  You may obtain a copy of the License at
9    *
10   *   http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing,
13   * software distributed under the License is distributed on an
14   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   * KIND, either express or implied.  See the License for the
16   * specific language governing permissions and limitations
17   * under the License.
18   */
19  package org.apache.maven.surefire.api.util;
20  
21  import java.io.File;
22  import java.util.Objects;
23  import java.util.stream.Stream;
24  
25  /**
26   * Centralized file management of temporary files in surefire.<br>
27   * Files are deleted on VM exit.
28   *
29   * @author Markus Spann
30   */
31  public final class SureFireFileManager {
32  
33      private static TempFileManager instance = create();
34  
35      private static TempFileManager create() {
36          String subDirName = "surefire";
37  
38          // create directory name suffix from legal chars in the current user name
39          // or a millisecond timestamp as fallback
40          String userSuffix = Stream.of("user.name", "USER", "USERNAME")
41                  .map(System::getProperty)
42                  .filter(Objects::nonNull)
43                  .findFirst()
44                  .map(u -> u.replaceAll("[^A-Za-z0-9\\-_]", ""))
45                  .map(u -> u.isEmpty() ? null : u)
46                  .orElse(Long.toString(System.currentTimeMillis()));
47  
48          if (userSuffix != null) {
49              subDirName += "-" + userSuffix;
50          }
51  
52          TempFileManager tfm = TempFileManager.instance(subDirName);
53          tfm.setDeleteOnExit(true);
54          return tfm;
55      }
56  
57      public static File createTempFile(String prefix, String suffix) {
58          return instance.createTempFile(prefix, suffix);
59      }
60  }