0
0
mirror of https://github.com/mongodb/mongo.git synced 2024-11-29 16:47:28 +01:00
mongodb/site_scons/site_tools/jsheader.py
Tad Marshall 2d2a2ce570 SERVER-7892 Do not strip indentation in JavaScript source code
V8 displays source code "as-is", so if we remove indentation then it
won't be there when users display "built-in" shell helper source code.
2013-03-18 13:14:06 -04:00

54 lines
1.3 KiB
Python

import os
from SCons.Builder import Builder
def jsToH(target, source, env):
outFile = str( target[0] )
h = ['#include "mongo/base/string_data.h"'
,'namespace mongo {'
,'struct JSFile{ const char* name; const StringData& source; };'
,'namespace JSFiles{'
]
def cppEscape(s):
s = s.rstrip()
s = s.replace( '\\', '\\\\' )
s = s.replace( '"', r'\"' )
return s
for s in source:
filename = str(s)
objname = os.path.split(filename)[1].split('.')[0]
stringname = '_jscode_raw_' + objname
h.append('const StringData ' + stringname + " = ")
for l in open( filename, 'r' ):
h.append( '"' + cppEscape(l) + r'\n" ' )
h.append(";")
h.append('extern const JSFile %s;'%objname) #symbols aren't exported w/o this
h.append('const JSFile %s = { "%s", %s };'%(objname, filename.replace('\\', '/'), stringname))
h.append("} // namespace JSFiles")
h.append("} // namespace mongo")
h.append("")
text = '\n'.join(h);
out = open( outFile, 'wb' )
try:
out.write( text )
finally:
out.close()
jshBuilder = Builder( action=jsToH )
def generate(env, **kw):
env.Append( BUILDERS=dict( JSHeader=jshBuilder ) )
def exists(env):
return True