package com.qianwen.smart.core.auto.service;


import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

class ServicesFiles {
  private static final Charset UTF_8 = StandardCharsets.UTF_8;
  
  static Set<String> readServiceFile(InputStream input) throws IOException {
    HashSet<String> serviceClasses = new HashSet<>();
    try(InputStreamReader isr = new InputStreamReader(input, UTF_8); 
        BufferedReader r = new BufferedReader(isr)) {
      String line;
      while ((line = r.readLine()) != null) {
        int commentStart = line.indexOf('#');
        if (commentStart >= 0)
          line = line.substring(0, commentStart); 
        line = line.trim();
        if (!line.isEmpty())
          serviceClasses.add(line); 
      } 
      return serviceClasses;
    } 
  }
  
  static void writeServiceFile(Collection<String> services, OutputStream output) throws IOException {
    BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(output, UTF_8));
    for (String service : services) {
      writer.write(service);
      writer.newLine();
    } 
    writer.flush();
  }
  
}