UNPKG

@open-editor/server

Version:

internal utils shared across @open-editor packages

2 lines (1 loc) 3.99 kB
import http from"node:http";import https from"node:https";import{existsSync,statSync,readFileSync}from"node:fs";import connect from"connect";import corsMiddleware from"cors";import{ServerApis}from"@open-editor/shared";import{resolve}from"node:path";import{parse}from"node:url";import openEditor from"launch-editor";import net from"node:net";const DEFAULE_OPEN_DDITOR=(file,errorCallback)=>{openEditor(file,(_,errorMessage)=>errorCallback(errorMessage))};function openEditorMiddleware(options={}){const{rootDir=process.cwd(),onOpenEditor=DEFAULE_OPEN_DDITOR}=options;return(req,res)=>{var _a;try{const{query}=parse((_a=req.url)!=null?_a:"/",!0),{f:file="unknown",l:line="1",c:column="1"}=query;if(!file){sendErrorResponse(res,400,"\u7F3A\u5C11\u6587\u4EF6\u8DEF\u5F84\u53C2\u6570");return}const filename=resolve(rootDir,decodeURIComponent(file));if(!validateFile(filename,res))return;req.headers.referer&&onOpenEditor(`${filename}:${line}:${column}`,errorMessage=>{throw new Error(errorMessage||"\u53EF\u80FD\u539F\u56E0\u6709\u7F16\u8F91\u5668\u672A\u542F\u52A8/\u7F16\u8F91\u5668\u672A\u54CD\u5E94")}),sendFileContent(res,filename)}catch(error){const errorMessage=error instanceof Error?error.message:"\u672A\u77E5\u9519\u8BEF";sendErrorResponse(res,500,`\u670D\u52A1\u5668\u5185\u90E8\u9519\u8BEF: ${errorMessage}`)}}}function validateFile(filename,res){return existsSync(filename)?statSync(filename).isFile()?!0:(sendErrorResponse(res,400,`'${filename}' \u4E0D\u662F\u6709\u6548\u6587\u4EF6`),!1):(sendErrorResponse(res,404,`\u6587\u4EF6 '${filename}' \u4E0D\u5B58\u5728`),!1)}function sendFileContent(res,filename){res.setHeader("Content-Type","application/javascript;charset=UTF-8"),res.end(readFileSync(filename,"utf-8"))}function sendErrorResponse(res,code,message){res.statusCode=code,res.setHeader("Content-Type","text/plain;charset=UTF-8"),res.end(`[@open-editor/server] ${message}`)}function createApp(options){const{rootDir,onOpenEditor}=options,app=connect();return app.use(corsMiddleware({methods:"GET"})),app.use(ServerApis.OPEN_EDITOR,openEditorMiddleware({rootDir,onOpenEditor})),app}var __async=(__this,__arguments,generator)=>new Promise((resolve2,reject)=>{var fulfilled=value=>{try{step(generator.next(value))}catch(e){reject(e)}},rejected=value=>{try{step(generator.throw(value))}catch(e){reject(e)}},step=x=>x.done?resolve2(x.value):Promise.resolve(x.value).then(fulfilled,rejected);step((generator=generator.apply(__this,__arguments)).next())});const MIN_PORT_NUMBER=3e3,MAX_PORT_NUMBER=9e3;function getAvailablePort(){return __async(this,arguments,function*({concurrency=5,retries=10}={}){for(let i=0;i<retries;i++){const promises=Array.from({length:concurrency},generatePort).map(port=>checkPortNumber(port).then(available=>available?port:null)),result=yield Promise.race([...promises,new Promise(resolve2=>{setTimeout(()=>resolve2(null),100)})]);if(result)return result}throw new Error(`\u7AEF\u53E3\u63A2\u6D4B\u5931\u8D25\uFF0C\u8BF7\u68C0\u67E5\u7CFB\u7EDF\u8D44\u6E90\u3002\u5C1D\u8BD5\u6B21\u6570\uFF1A${retries}`)})}function checkPortNumber(port){return new Promise(resolve2=>{const server=net.createServer();server.unref(),server.on("error",()=>{resolve2(!1)}),server.listen(port,()=>{server.close(()=>{resolve2(!0)})})})}function generatePort(){return Math.floor(Math.random()*(MAX_PORT_NUMBER-MIN_PORT_NUMBER)+MIN_PORT_NUMBER)}function setupServer(options={}){const{rootDir,https:httpsConfig}=options,app=createApp({rootDir}),server=createHttpServer(app,httpsConfig);return startServer(server)}function createHttpServer(app,httpsConfig){if(!httpsConfig)return http.createServer(app);const sslOptions={key:readFileSync(httpsConfig.key),cert:readFileSync(httpsConfig.cert)};return https.createServer(sslOptions,app)}function startServer(server){return new Promise((resolve2,reject)=>{getAvailablePort().then(port=>{server.listen(port).once("listening",()=>resolve2(port)).once("error",reject)}).catch(reject)})}export{openEditorMiddleware,setupServer};