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 package groovy.ui;
47
48 import groovy.lang.GroovyShell;
49 import groovy.lang.Script;
50
51 import java.io.BufferedReader;
52 import java.io.FileInputStream;
53 import java.io.IOException;
54 import java.io.InputStreamReader;
55 import java.io.PrintWriter;
56 import java.net.InetAddress;
57 import java.net.ServerSocket;
58 import java.net.Socket;
59 import java.net.URL;
60
61 /***
62 * Simple server that executes supplied script against a socket
63 * @author Jeremy Rayner
64 */
65
66 public class GroovySocketServer implements Runnable {
67 private URL url;
68 private GroovyShell groovy;
69 private boolean isScriptFile;
70 private String scriptFilenameOrText;
71 private boolean autoOutput;
72
73 public GroovySocketServer(GroovyShell groovy, boolean isScriptFile, String scriptFilenameOrText, boolean autoOutput, int port) {
74 this.groovy = groovy;
75 this.isScriptFile = isScriptFile;
76 this.scriptFilenameOrText = scriptFilenameOrText;
77 this.autoOutput = autoOutput;
78 try {
79 url = new URL("http", InetAddress.getLocalHost().getHostAddress(), port, "/");
80 System.out.println("groovy is listening on port " + port);
81 } catch (IOException e) {
82 e.printStackTrace();
83 }
84 new Thread(this).start();
85 }
86
87 public void run() {
88 try {
89 ServerSocket serverSocket = new ServerSocket(url.getPort());
90 while (true) {
91
92
93
94
95 Script script;
96 if (isScriptFile) {
97 GroovyMain gm = new GroovyMain();
98 script = groovy.parse(new FileInputStream(gm.huntForTheScriptFile(scriptFilenameOrText)));
99 } else {
100 script = groovy.parse(scriptFilenameOrText);
101 }
102 new GroovyClientConnection(script, autoOutput, serverSocket.accept());
103 }
104 } catch (Exception e) {
105 e.printStackTrace();
106 }
107 }
108
109 class GroovyClientConnection implements Runnable {
110 private Script script;
111 private Socket socket;
112 private BufferedReader reader;
113 private PrintWriter writer;
114 private boolean autoOutputFlag;
115
116 GroovyClientConnection(Script script, boolean autoOutput,Socket socket) throws IOException {
117 this.script = script;
118 this.autoOutputFlag = autoOutput;
119 this.socket = socket;
120 reader = new BufferedReader(new InputStreamReader(socket.getInputStream()));
121 writer = new PrintWriter(socket.getOutputStream());
122 new Thread(this, "Groovy client connection - " + socket.getInetAddress().getHostAddress()).start();
123 }
124 public void run() {
125 try {
126 String line = null;
127 script.setProperty("out", writer);
128 script.setProperty("socket", socket);
129 script.setProperty("init", Boolean.TRUE);
130 while ((line = reader.readLine()) != null) {
131
132 script.setProperty("line", line);
133 Object o = script.run();
134 script.setProperty("init", Boolean.FALSE);
135 if (o != null) {
136 if ("success".equals(o)) {
137 break;
138 } else {
139 if (autoOutputFlag) {
140 writer.println(o);
141 }
142 }
143 }
144 writer.flush();
145 }
146 } catch (IOException e) {
147 e.printStackTrace();
148 } finally {
149 try {
150 writer.flush();
151 writer.close();
152 } finally {
153 try {
154 socket.close();
155 } catch (IOException e3) {
156 e3.printStackTrace();
157 }
158 }
159 }
160 }
161 }
162 }