149 lines
2.7 KiB
C++
149 lines
2.7 KiB
C++
#include <iostream>
|
|
|
|
#include <string.h>
|
|
|
|
#include <ztd/options.hpp>
|
|
#include <ztd/shell.hpp>
|
|
|
|
#include <unistd.h>
|
|
|
|
#include "util.hpp"
|
|
#include "struc.hpp"
|
|
#include "parse.hpp"
|
|
#include "options.hpp"
|
|
|
|
int execute(block& sh, std::vector<std::string>& args)
|
|
{
|
|
std::string data=sh.generate();
|
|
|
|
std::string filename=ztd::exec("basename", args[0]).first;
|
|
filename.pop_back();
|
|
|
|
// generate path
|
|
std::string tmpdir = (getenv("TMPDIR") != NULL) ? getenv("TMPDIR") : "/tmp" ;
|
|
std::string dirpath = tmpdir + "/lxsh_" + ztd::sh("tr -dc '[:alnum:]' < /dev/urandom | head -c10");
|
|
std::string filepath = dirpath+'/'+filename;
|
|
|
|
// create dir
|
|
if(ztd::exec("mkdir", "-p", dirpath).second)
|
|
{
|
|
throw std::runtime_error("Failed to create directory '"+dirpath+'\'');
|
|
}
|
|
|
|
// create stream
|
|
std::ofstream stream(filepath);
|
|
if(!stream)
|
|
{
|
|
ztd::exec("rm", "-rf", dirpath);
|
|
throw std::runtime_error("Failed to write to '"+filepath+'\'');
|
|
}
|
|
|
|
// output
|
|
stream << data;
|
|
stream.close();
|
|
if(ztd::exec("chmod", "+x", filepath).second != 0)
|
|
{
|
|
ztd::exec("rm", "-rf", dirpath);
|
|
throw std::runtime_error("Failed to make '"+filepath+"' executable");
|
|
}
|
|
|
|
// exec
|
|
int retval=_exec(filepath, args);
|
|
ztd::exec("rm", "-rf", dirpath);
|
|
|
|
return retval;
|
|
}
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
auto args=options.process(argc, argv, false, true);
|
|
|
|
if(options['m'])
|
|
opt_minimize=true;
|
|
|
|
piped=false;
|
|
|
|
if(options['h'])
|
|
{
|
|
print_help(argv[0]);
|
|
return 1;
|
|
}
|
|
if(options["help-commands"])
|
|
{
|
|
print_include_help();
|
|
printf("\n\n");
|
|
print_resolve_help();
|
|
return 1;
|
|
}
|
|
|
|
std::string file;
|
|
if(args.size() > 0)
|
|
{
|
|
if(args[0] == "-")
|
|
{
|
|
piped=true;
|
|
file = "/dev/stdin";
|
|
}
|
|
else
|
|
file=args[0];
|
|
}
|
|
else
|
|
{
|
|
if(isatty(fileno(stdin)))
|
|
{
|
|
print_help(argv[0]);
|
|
return 1;
|
|
}
|
|
else
|
|
{
|
|
piped=true;
|
|
file = "/dev/stdin";
|
|
}
|
|
}
|
|
|
|
|
|
g_origin=file;
|
|
add_include(file);
|
|
|
|
try
|
|
{
|
|
block sh(parse(import_file(file)));
|
|
std::string curbin, binshebang;
|
|
curbin=ztd::exec("basename", argv[0]).first;
|
|
binshebang=ztd::exec("basename", sh.shebang).first;
|
|
if(binshebang==curbin)
|
|
sh.shebang="#!/bin/sh";
|
|
if(options['e'])
|
|
{
|
|
return execute(sh, args);
|
|
}
|
|
else if(options['o'])
|
|
{
|
|
std::cout << sh.generate();
|
|
}
|
|
else
|
|
{
|
|
if(binshebang == curbin)
|
|
{
|
|
return execute(sh, args);
|
|
}
|
|
else
|
|
{
|
|
std::cout << sh.generate();
|
|
}
|
|
}
|
|
}
|
|
catch(ztd::format_error& e)
|
|
{
|
|
printFormatError(e);
|
|
return 100;
|
|
}
|
|
catch(std::exception& e)
|
|
{
|
|
std::cerr << e.what() << std::endl;
|
|
return 2;
|
|
}
|
|
|
|
|
|
return 0;
|
|
}
|