fork download 
  1. /*
  2.  * hosts ファイルの整理
  3.  *
  4.  * author: Leonardone @ NEETSDKASU
  5.  */
  6. //package myapp.tool;
  7.  
  8. import java.lang.Comparable;
  9. import java.lang.String;
  10. import java.lang.System;
  11. import java.util.HashMap;
  12. import java.util.Map;
  13. import java.util.PriorityQueue;
  14. import java.util.Queue;
  15. import java.util.SortedMap;
  16. import java.util.TreeMap;
  17. import java.io.BufferedReader;
  18. import java.io.FileReader;
  19. import java.io.InputStreamReader;
  20. import java.io.PrintStream;
  21.  
  22. class Hosts
  23. {
  24. 	private static final String ARG_INPUT = "-i";
  25. 	private static final String ARG_TEMPLATE = "-t";
  26. 	private static final String ARG_OUTPUT = "-o";
  27.  
  28. 	public static void main (String[] args) throws java.lang.Exception
  29. 	{
  30. 		Map<String, String> argmap = parseArgs(args);
  31. 		BufferedReader in = null;
  32. 		PrintStream out = null;
  33.  
  34. 		try {
  35. 			if (argmap.containsKey(ARG_INPUT)) {
  36. 				in = new BufferedReader(new FileReader(argmap.get(ARG_INPUT)));
  37. 			} else {
  38. 				in = new BufferedReader(new InputStreamReader(System.in));
  39. 			}
  40.  
  41. 			String line, ip;
  42. 			Map<String, String> ips = new HashMap<>();
  43. 			Tree tree = new Tree();
  44.  
  45. 			while ((line = in.readLine()) != null) {
  46. 				if (line.matches("^\\s*(\\d{1,3}\\.){3}\\d{1,3}(\\s+([^\\.]+\\.)+[^\\.]+)+") == false) {
  47. 					continue;
  48. 				}
  49. 				String[] tokens = line.split("\\s");
  50. 				ip = null;
  51. 				for (String s : tokens) {
  52. 					if (s.length() > 0) {
  53. 						if (ip == null) {
  54. 							ip = s;
  55. 							if (ips.containsKey(ip)) {
  56. 								ip = ips.get(ip);
  57. 							} else {
  58. 								ips.put(ip, ip);
  59. 							}
  60. 						} else {
  61. 							tree.add(s, ip);
  62. 						}
  63. 					}
  64. 				}
  65. 			}
  66.  
  67. 			if (argmap.containsKey(ARG_OUTPUT)) {
  68. 				out = new PrintStream(argmap.get(ARG_OUTPUT));
  69. 			} else {
  70. 				out = System.out;
  71. 			}
  72.  
  73. 			printHeader(out, argmap.get(ARG_TEMPLATE));
  74.  
  75. 			tree.print(out);
  76.  
  77. 		} catch (java.lang.Exception ex) {
  78. 			throw ex;
  79. 		} finally {
  80. 			if (in != null && argmap.containsKey(ARG_INPUT)) {
  81. 				in.close();
  82. 			}
  83. 			if (out != null && argmap.containsKey(ARG_OUTPUT)) {
  84. 				out.flush();
  85. 				out.close();
  86. 			}
  87. 		}
  88. 	}
  89.  
  90. 	private static void printHeader(PrintStream out, String filepath) throws java.lang.Exception
  91. 	{
  92. 		if (filepath == null) {
  93. 			out.println("# hosts");
  94. 			out.println();
  95. 			return;
  96. 		}
  97. 		BufferedReader in = null;
  98. 		try {
  99. 			in = new BufferedReader(new FileReader(filepath));
  100. 			String line;
  101. 			while ((line = in.readLine()) != null) {
  102. 				out.println(line);
  103. 			}
  104. 		} catch (java.lang.Exception ex) {
  105. 			throw ex;
  106. 		} finally {
  107. 			if (in != null) {
  108. 				in.close();
  109. 			}
  110. 		}
  111. 	}
  112.  
  113. 	private static enum Flag
  114. 	{
  115. 		ANY, INPUT, OUPUT, TEMPLATE
  116. 	}
  117.  
  118. 	private static Map<String, String> parseArgs(String args[])
  119. 	{
  120. 		Map<String, String> argmap = new HashMap<>();
  121. 		if (args == null) {
  122. 			return argmap;
  123. 		}
  124. 		Flag flag = Flag.ANY;
  125. 		for (String arg : args) {
  126. 			switch (flag) {
  127. 			case ANY:
  128. 				switch (arg) {
  129. 				case ARG_TEMPLATE: flag = Flag.TEMPLATE; break;
  130. 				case ARG_OUTPUT: flag = Flag.OUPUT; break;
  131. 				case ARG_INPUT: flag = Flag.INPUT; break;
  132. 				default: 
  133. 					argmap.put(ARG_INPUT, arg); break;
  134. 				}
  135. 				break;
  136. 			case TEMPLATE:
  137. 				argmap.put(ARG_TEMPLATE, arg);
  138. 				flag = Flag.ANY;
  139. 				break;
  140. 			case OUPUT:
  141. 				argmap.put(ARG_OUTPUT, arg);
  142. 				flag = Flag.ANY;
  143. 				break;
  144. 			case INPUT:
  145. 				argmap.put(ARG_INPUT, arg);
  146. 				flag = Flag.ANY;
  147. 				break;
  148. 			default:
  149. 				break;
  150. 			}
  151. 		}
  152. 		return argmap;
  153. 	}
  154. }
  155.  
  156. class Tree
  157. {
  158. 	private class Node implements Comparable<Node>
  159. 	{
  160. 		SortedMap<String, Node> map = null;
  161. 		final String name;
  162. 		String ip = null;
  163. 		int depth = 0;
  164.  
  165. 		Node(String name)
  166. 		{
  167. 			this.name = name;
  168. 		}
  169.  
  170. 		void setIP(String ip)
  171. 		{
  172. 			this.ip = ip;
  173. 		}
  174.  
  175. 		int calcDepth()
  176. 		{
  177. 			if (depth > 0) {
  178. 				return depth;
  179. 			}
  180. 			depth = 1;
  181. 			if (map != null) {
  182. 				int d = 1;
  183. 				for (Node node : map.values()) {
  184. 					int t = node.calcDepth();
  185. 					if (t > d) {
  186. 						d = t;
  187. 					}
  188. 				}
  189. 				depth = d + 1;
  190. 			}
  191. 			return depth;
  192. 		}
  193.  
  194. 		public int compareTo(Node o)
  195. 		{
  196. 			return depth - o.depth;
  197. 		}
  198.  
  199. 		Node add(String name)
  200. 		{
  201. 			depth = 0;
  202. 			if (map == null) {
  203. 				map = new TreeMap<>();
  204. 			}
  205. 			Node node = map.get(name);
  206. 			if (node == null) {
  207. 				map.put(name, node = new Node(name));
  208. 			}
  209. 			return node;
  210. 		}
  211.  
  212. 		public boolean checkChildIP() {
  213. 			if (map != null) {
  214. 				for (Node node : map.values()) {
  215. 					if (node.ip != null) {
  216. 						return true;
  217. 					}
  218. 				}
  219. 			}
  220. 			return false;
  221. 		}
  222.  
  223. 		int print(PrintStream out, String host) {
  224. 			String temp;
  225. 			if (host.length() > 0) {
  226. 				temp =  name + "." + host;
  227. 			} else {
  228. 				out.println();
  229. 				temp = name;
  230. 			}
  231.  			if (map == null) {
  232. 				if (ip != null) {
  233. 					out.println("\t" + ip + "\t" + temp);
  234. 				}
  235. 				return 1;
  236. 			} else {
  237. 				if (temp.length() > 0) {
  238. 					if (host.length() == 0) {
  239. 						out.println("# ***********************************************");
  240. 						out.println("# ." + temp);
  241. 						out.println("# ***********************************************");
  242. 						out.println();
  243. 					} else  {
  244. 						out.println("# ." + temp);
  245. 					}
  246. 				}
  247. 				if (ip != null) {
  248. 					out.println("\t" + ip + "\t" + temp);
  249. 				}
  250. 				if (checkChildIP()) {
  251. 					Queue<Node> que = new PriorityQueue<>();
  252. 					for (Node node : map.values()) {
  253. 						que.add(node);
  254. 					}
  255. 					int n = 0;
  256. 					while (que.isEmpty() == false) {
  257. 						n += que.poll().print(out, temp);
  258. 					}
  259. 					if (n > 0) {
  260. 						out.println();
  261. 					}
  262. 				} else {
  263. 					int n = 0;
  264. 					for (Node node : map.values()) {
  265. 						n += node.print(out, temp);
  266. 					}
  267. 					if (n > 0) {
  268. 						out.println();
  269. 					}
  270. 				}
  271.  
  272. 				return 0;
  273. 			}
  274. 		}
  275. 	}
  276.  
  277. 	Node root;
  278.  
  279. 	Tree()
  280. 	{
  281. 		root = new Node("");
  282. 	}
  283.  
  284. 	void add(String hostname, String ip)
  285. 	{
  286. 		String[] tokens = hostname.split("\\.");
  287. 		Node node = root;
  288.  
  289. 		for (int i = tokens.length - 1; i >= 0; i--) {
  290. 			node = node.add(tokens[i]);
  291. 		}
  292. 		node.setIP(ip);
  293.  
  294. 	}
  295.  
  296. 	void print(PrintStream out)
  297. 	{
  298. 		root.calcDepth();
  299. 		out.println();
  300. 		root.print(out, "");
  301. 	}
  302. }
  303.  
Success #stdin #stdout 0.08s 380224KB
comments (?)
stdin 
# hosts


# example.com
	127.0.0.1	www.example.com
	127.0.0.1	abc.example.com
	127.0.0.1	xyz.example.com
	127.0.0.1	img.example.com
# yahoo.co.jp
	127.0.0.1	www.yahoo.co.jp
	127.0.0.1	abc.yahoo.co.jp
	127.0.0.1	xyz.yahoo.co.jp
	127.0.0.1	img.yahoo.co.jp
# google.co.jp
	127.0.0.1	www.google.co.jp
	127.0.0.1	abc.google.co.jp
	127.0.0.1	xyz.google.co.jp
	127.0.0.1	img.google.co.jp
stdout 
# hosts




# ***********************************************
# .com
# ***********************************************

# .example.com
	127.0.0.1	abc.example.com
	127.0.0.1	xyz.example.com
	127.0.0.1	www.example.com
	127.0.0.1	img.example.com


# ***********************************************
# .jp
# ***********************************************

# .co.jp
# .google.co.jp
	127.0.0.1	abc.google.co.jp
	127.0.0.1	xyz.google.co.jp
	127.0.0.1	www.google.co.jp
	127.0.0.1	img.google.co.jp

# .yahoo.co.jp
	127.0.0.1	abc.yahoo.co.jp
	127.0.0.1	xyz.yahoo.co.jp
	127.0.0.1	www.yahoo.co.jp
	127.0.0.1	img.yahoo.co.jp
https://ideone.com/oWk4XE
language:
Java (HotSpot 12)
created:
10 years ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!