Skip to content

Commit c96fb65

Browse files
code-ashercode-asher
committed
Split some entry methods into main
This is so they can be unit tested.
1 parent 083400b commit c96fb65

File tree

2 files changed

+166
-165
lines changed

2 files changed

+166
-165
lines changed

‎src/node/entry.ts‎

Lines changed: 3 additions & 165 deletions
Original file line numberDiff line numberDiff line change
@@ -1,179 +1,17 @@
1-
import { field, logger } from "@coder/logger"
2-
import * as cp from "child_process"
3-
import http from "http"
4-
import * as path from "path"
5-
import { CliMessage, OpenCommandPipeArgs } from "../../typings/ipc"
6-
import { plural } from "../common/util"
7-
import { createApp, ensureAddress } from "./app"
1+
import { logger } from "@coder/logger"
82
import {
9-
AuthType,
10-
DefaultedArgs,
11-
Feature,
123
optionDescriptions,
134
parse,
145
readConfigFile,
156
setDefaults,
167
shouldOpenInExistingInstance,
178
shouldRunVsCodeCli,
189
} from "./cli"
19-
import { coderCloudBind } from "./coder_cloud"
2010
import { commit, version } from "./constants"
11+
import { openInExistingInstance, runCodeServer, runVsCodeCli } from "./main"
2112
import * as proxyAgent from "./proxy_agent"
22-
import { register } from "./routes"
23-
import { humanPath, isFile, open } from "./util"
2413
import { isChild, wrapper } from "./wrapper"
2514

26-
export const runVsCodeCli = (args: DefaultedArgs): void => {
27-
logger.debug("forking vs code cli...")
28-
const vscode = cp.fork(path.resolve(__dirname, "../../lib/vscode/out/vs/server/fork"), [], {
29-
env: {
30-
...process.env,
31-
CODE_SERVER_PARENT_PID: process.pid.toString(),
32-
},
33-
})
34-
vscode.once("message", (message: any) => {
35-
logger.debug("got message from VS Code", field("message", message))
36-
if (message.type !== "ready") {
37-
logger.error("Unexpected response waiting for ready response", field("type", message.type))
38-
process.exit(1)
39-
}
40-
const send: CliMessage = { type: "cli", args }
41-
vscode.send(send)
42-
})
43-
vscode.once("error", (error) => {
44-
logger.error("Got error from VS Code", field("error", error))
45-
process.exit(1)
46-
})
47-
vscode.on("exit", (code) => process.exit(code || 0))
48-
}
49-
50-
export const openInExistingInstance = async (args: DefaultedArgs, socketPath: string): Promise<void> => {
51-
const pipeArgs: OpenCommandPipeArgs & { fileURIs: string[] } = {
52-
type: "open",
53-
folderURIs: [],
54-
fileURIs: [],
55-
forceReuseWindow: args["reuse-window"],
56-
forceNewWindow: args["new-window"],
57-
}
58-
59-
for (let i = 0; i < args._.length; i++) {
60-
const fp = path.resolve(args._[i])
61-
if (await isFile(fp)) {
62-
pipeArgs.fileURIs.push(fp)
63-
} else {
64-
pipeArgs.folderURIs.push(fp)
65-
}
66-
}
67-
68-
if (pipeArgs.forceNewWindow && pipeArgs.fileURIs.length > 0) {
69-
logger.error("--new-window can only be used with folder paths")
70-
process.exit(1)
71-
}
72-
73-
if (pipeArgs.folderURIs.length === 0 && pipeArgs.fileURIs.length === 0) {
74-
logger.error("Please specify at least one file or folder")
75-
process.exit(1)
76-
}
77-
78-
const vscode = http.request(
79-
{
80-
path: "/",
81-
method: "POST",
82-
socketPath,
83-
},
84-
(response) => {
85-
response.on("data", (message) => {
86-
logger.debug("got message from VS Code", field("message", message.toString()))
87-
})
88-
},
89-
)
90-
vscode.on("error", (error: unknown) => {
91-
logger.error("got error from VS Code", field("error", error))
92-
})
93-
vscode.write(JSON.stringify(pipeArgs))
94-
vscode.end()
95-
}
96-
97-
const main = async (args: DefaultedArgs): Promise<void> => {
98-
logger.info(`code-server ${version} ${commit}`)
99-
100-
logger.info(`Using user-data-dir ${humanPath(args["user-data-dir"])}`)
101-
logger.trace(`Using extensions-dir ${humanPath(args["extensions-dir"])}`)
102-
103-
if (args.auth === AuthType.Password && !args.password && !args["hashed-password"]) {
104-
throw new Error(
105-
"Please pass in a password via the config file or environment variable ($PASSWORD or $HASHED_PASSWORD)",
106-
)
107-
}
108-
109-
const [app, wsApp, server] = await createApp(args)
110-
const serverAddress = ensureAddress(server)
111-
await register(app, wsApp, server, args)
112-
113-
logger.info(`Using config file ${humanPath(args.config)}`)
114-
logger.info(`HTTP server listening on ${serverAddress} ${args.link ? "(randomized by --link)" : ""}`)
115-
116-
if (args.auth === AuthType.Password) {
117-
logger.info(" - Authentication is enabled")
118-
if (args.usingEnvPassword) {
119-
logger.info(" - Using password from $PASSWORD")
120-
} else if (args.usingEnvHashedPassword) {
121-
logger.info(" - Using password from $HASHED_PASSWORD")
122-
} else {
123-
logger.info(` - Using password from ${humanPath(args.config)}`)
124-
}
125-
} else {
126-
logger.info(` - Authentication is disabled ${args.link ? "(disabled by --link)" : ""}`)
127-
}
128-
129-
if (args.cert) {
130-
logger.info(` - Using certificate for HTTPS: ${humanPath(args.cert.value)}`)
131-
} else {
132-
logger.info(` - Not serving HTTPS ${args.link ? "(disabled by --link)" : ""}`)
133-
}
134-
135-
if (args["proxy-domain"].length > 0) {
136-
logger.info(` - ${plural(args["proxy-domain"].length, "Proxying the following domain")}:`)
137-
args["proxy-domain"].forEach((domain) => logger.info(` - *.${domain}`))
138-
}
139-
140-
if (args.link) {
141-
try {
142-
await coderCloudBind(serverAddress.replace(/^https?:\/\//, ""), args.link.value)
143-
logger.info(" - Connected to cloud agent")
144-
} catch (err) {
145-
logger.error(err.message)
146-
wrapper.exit(1)
147-
}
148-
}
149-
150-
if (args.enable && args.enable.length > 0) {
151-
logger.info("Enabling the following experimental features:")
152-
args.enable.forEach((feature) => {
153-
if (Object.values(Feature).includes(feature as Feature)) {
154-
logger.info(` - "${feature}"`)
155-
} else {
156-
logger.error(` X "${feature}" (unknown feature)`)
157-
}
158-
})
159-
// TODO: Could be nice to add wrapping to the logger?
160-
logger.info(
161-
" The code-server project does not provide stability guarantees or commit to fixing bugs relating to these experimental features. When filing bug reports, please ensure that you can reproduce the bug with all experimental features turned off.",
162-
)
163-
}
164-
165-
if (!args.socket && args.open) {
166-
// The web socket doesn't seem to work if browsing with 0.0.0.0.
167-
const openAddress = serverAddress.replace("://0.0.0.0", "://localhost")
168-
try {
169-
await open(openAddress)
170-
logger.info(`Opened ${openAddress}`)
171-
} catch (error) {
172-
logger.error("Failed to open", field("address", openAddress), field("error", error))
173-
}
174-
}
175-
}
176-
17715
async function entry(): Promise<void> {
17816
proxyAgent.monkeyPatch(false)
17917

@@ -186,7 +24,7 @@ async function entry(): Promise<void> {
18624
if (isChild(wrapper)) {
18725
const args = await wrapper.handshake()
18826
wrapper.preventExit()
189-
return main(args)
27+
return runCodeServer(args)
19028
}
19129

19230
const cliArgs = parse(process.argv.slice(2))

‎src/node/main.ts‎

Lines changed: 163 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,163 @@
1+
import { field, logger } from "@coder/logger"
2+
import * as cp from "child_process"
3+
import http from "http"
4+
import * as path from "path"
5+
import { CliMessage, OpenCommandPipeArgs } from "../../typings/ipc"
6+
import { plural } from "../common/util"
7+
import { createApp, ensureAddress } from "./app"
8+
import { AuthType, DefaultedArgs, Feature } from "./cli"
9+
import { coderCloudBind } from "./coder_cloud"
10+
import { commit, version } from "./constants"
11+
import { register } from "./routes"
12+
import { humanPath, isFile, open } from "./util"
13+
import { wrapper } from "./wrapper"
14+
15+
export const runVsCodeCli = (args: DefaultedArgs): void => {
16+
logger.debug("forking vs code cli...")
17+
const vscode = cp.fork(path.resolve(__dirname, "../../lib/vscode/out/vs/server/fork"), [], {
18+
env: {
19+
...process.env,
20+
CODE_SERVER_PARENT_PID: process.pid.toString(),
21+
},
22+
})
23+
vscode.once("message", (message: any) => {
24+
logger.debug("got message from VS Code", field("message", message))
25+
if (message.type !== "ready") {
26+
logger.error("Unexpected response waiting for ready response", field("type", message.type))
27+
process.exit(1)
28+
}
29+
const send: CliMessage = { type: "cli", args }
30+
vscode.send(send)
31+
})
32+
vscode.once("error", (error) => {
33+
logger.error("Got error from VS Code", field("error", error))
34+
process.exit(1)
35+
})
36+
vscode.on("exit", (code) => process.exit(code || 0))
37+
}
38+
39+
export const openInExistingInstance = async (args: DefaultedArgs, socketPath: string): Promise<void> => {
40+
const pipeArgs: OpenCommandPipeArgs & { fileURIs: string[] } = {
41+
type: "open",
42+
folderURIs: [],
43+
fileURIs: [],
44+
forceReuseWindow: args["reuse-window"],
45+
forceNewWindow: args["new-window"],
46+
}
47+
48+
for (let i = 0; i < args._.length; i++) {
49+
const fp = path.resolve(args._[i])
50+
if (await isFile(fp)) {
51+
pipeArgs.fileURIs.push(fp)
52+
} else {
53+
pipeArgs.folderURIs.push(fp)
54+
}
55+
}
56+
57+
if (pipeArgs.forceNewWindow && pipeArgs.fileURIs.length > 0) {
58+
logger.error("--new-window can only be used with folder paths")
59+
process.exit(1)
60+
}
61+
62+
if (pipeArgs.folderURIs.length === 0 && pipeArgs.fileURIs.length === 0) {
63+
logger.error("Please specify at least one file or folder")
64+
process.exit(1)
65+
}
66+
67+
const vscode = http.request(
68+
{
69+
path: "/",
70+
method: "POST",
71+
socketPath,
72+
},
73+
(response) => {
74+
response.on("data", (message) => {
75+
logger.debug("got message from VS Code", field("message", message.toString()))
76+
})
77+
},
78+
)
79+
vscode.on("error", (error: unknown) => {
80+
logger.error("got error from VS Code", field("error", error))
81+
})
82+
vscode.write(JSON.stringify(pipeArgs))
83+
vscode.end()
84+
}
85+
86+
export const runCodeServer = async (args: DefaultedArgs): Promise<void> => {
87+
logger.info(`code-server ${version} ${commit}`)
88+
89+
logger.info(`Using user-data-dir ${humanPath(args["user-data-dir"])}`)
90+
logger.trace(`Using extensions-dir ${humanPath(args["extensions-dir"])}`)
91+
92+
if (args.auth === AuthType.Password && !args.password && !args["hashed-password"]) {
93+
throw new Error(
94+
"Please pass in a password via the config file or environment variable ($PASSWORD or $HASHED_PASSWORD)",
95+
)
96+
}
97+
98+
const [app, wsApp, server] = await createApp(args)
99+
const serverAddress = ensureAddress(server)
100+
await register(app, wsApp, server, args)
101+
102+
logger.info(`Using config file ${humanPath(args.config)}`)
103+
logger.info(`HTTP server listening on ${serverAddress} ${args.link ? "(randomized by --link)" : ""}`)
104+
if (args.auth === AuthType.Password) {
105+
logger.info(" - Authentication is enabled")
106+
if (args.usingEnvPassword) {
107+
logger.info(" - Using password from $PASSWORD")
108+
} else if (args.usingEnvHashedPassword) {
109+
logger.info(" - Using password from $HASHED_PASSWORD")
110+
} else {
111+
logger.info(` - Using password from ${humanPath(args.config)}`)
112+
}
113+
} else {
114+
logger.info(` - Authentication is disabled ${args.link ? "(disabled by --link)" : ""}`)
115+
}
116+
117+
if (args.cert) {
118+
logger.info(` - Using certificate for HTTPS: ${humanPath(args.cert.value)}`)
119+
} else {
120+
logger.info(` - Not serving HTTPS ${args.link ? "(disabled by --link)" : ""}`)
121+
}
122+
123+
if (args["proxy-domain"].length > 0) {
124+
logger.info(` - ${plural(args["proxy-domain"].length, "Proxying the following domain")}:`)
125+
args["proxy-domain"].forEach((domain) => logger.info(` - *.${domain}`))
126+
}
127+
128+
if (args.link) {
129+
try {
130+
await coderCloudBind(serverAddress.replace(/^https?:\/\//, ""), args.link.value)
131+
logger.info(" - Connected to cloud agent")
132+
} catch (err) {
133+
logger.error(err.message)
134+
wrapper.exit(1)
135+
}
136+
}
137+
138+
if (args.enable && args.enable.length > 0) {
139+
logger.info("Enabling the following experimental features:")
140+
args.enable.forEach((feature) => {
141+
if (Object.values(Feature).includes(feature as Feature)) {
142+
logger.info(` - "${feature}"`)
143+
} else {
144+
logger.error(` X "${feature}" (unknown feature)`)
145+
}
146+
})
147+
// TODO: Could be nice to add wrapping to the logger?
148+
logger.info(
149+
" The code-server project does not provide stability guarantees or commit to fixing bugs relating to these experimental features. When filing bug reports, please ensure that you can reproduce the bug with all experimental features turned off.",
150+
)
151+
}
152+
153+
if (!args.socket && args.open) {
154+
// The web socket doesn't seem to work if browsing with 0.0.0.0.
155+
const openAddress = serverAddress.replace("://0.0.0.0", "://localhost")
156+
try {
157+
await open(openAddress)
158+
logger.info(`Opened ${openAddress}`)
159+
} catch (error) {
160+
logger.error("Failed to open", field("address", openAddress), field("error", error))
161+
}
162+
}
163+
}

0 commit comments

Comments
 (0)