make.jl

module MakeLatexLambda

using JSON
using AWSCore
using AWSS3
using AWSIAM
using AWSLambda: create_lambda, update_lambda, lambda_configuration,
                 invoke_lambda, create_lambda_role
using InfoZIP

function all()
    download()
    build()
    zip()
    deploy()
    test()
end


# Download TeXLive installer.
function download()
    run(`wget
         http://mirror.ctan.org/systems/texlive/tlnet/install-tl-unx.tar.gz`)
end


# Build docker image from Dockerfile.
function build()
    run(`docker build -t octech/lambdalatex .`)
end


# Create Lambda deployment .ZIP file from docker image.
function zip()
    rm("latexlambda.zip", force=true)
    run(`docker run --rm -it -v $(pwd()):/var/host octech/lambdalatex zip --symlinks -r -9 /var/host/latexlambda.zip .`)
end


# Deploy .ZIP file to Lambda.
function deploy()
    AWSCore.set_debug_level(2)
    aws = AWSCore.default_aws_config()
    n = AWSCore.aws_account_number(aws)
    aws[:lambda_bucket] = "octech.latexlambda.deploy.$n"
    s3_create_bucket(aws[:lambda_bucket])
    s3_put(aws[:lambda_bucket], "latexlambda.zip", read("latexlambda.zip"))

    role = create_lambda_role(aws, "latex", """{
        "Version": "2012-10-17",
        "Statement": [
            {
                "Effect": "Allow",
                "Action": ["s3:GetObject", "s3:PutObject"],
                "Resource": "*"
            }
        ]
    }""")

    if lambda_configuration("latex") == nothing
        create_lambda(aws, "latex"; Runtime="python3.6",
                                    S3Key="latexlambda.zip",
                                    MemorySize=1536,
                                    Role=role)
    else
        update_lambda(aws, "latex"; S3Key="latexlambda.zip", Role=role)
    end
end


# Docker image interactive shell.
function shell()
    run(`docker run --rm -it -v $(pwd()):/var/host octech/lambdalatex bash`)
end

test_zip = base64encode(create_zip("document.tex" =>
                                   readstring("test_input.tex")))

# Test latex in local docker image.
function localtest()
    write("test_input.zip.base64", test_zip)
    pycmd = """
        import lambda_function
        import json
        import base64
        out = lambda_function.lambda_handler(
              {'input': open('/var/host/test_input.zip.base64').read()}, {})
        with open('/var/host/test_output.json', 'w') as f:
            f.write(json.dumps(out))
        """
    run(`docker run --security-opt seccomp:unconfined --rm -v $(pwd()):/var/host octech/lambdalatex strace -f -t -e trace=file -o /var/host/strace.out python3 -c $pycmd`)
    out = JSON.parse(readstring("test_output.json"))
    write("test_output_local.pdf", base64decode(out["output"]))
    write("test_output_local.stdout", out["stdout"])
end


# Test latex on Lambda.
function test()
    @time out = invoke_lambda("latex"; input=test_zip)
    write("test_output_lambda.pdf", base64decode(out[:output]))
    write("test_output_lambda.stdout", out[:stdout])
end

# Test latex on Lambda.
function s3test()
    @time out = invoke_lambda("latex"; input_bucket="cinchtmp.tex",
                                       input_key="Archive.zip",
                                       output_bucket="cinchtmp.pdf",
                                       output_key="test.pdf")
    write("test_output_lambda.stdout", out[:stdout])
end



# Remove intermediate files.
function clean()
    rm("test_output.json", force=true)
    rm("test_output_local.stdout", force=true)
    rm("test_output_local.pdf", force=true)
    rm("test_output_lambda.stdout", force=true)
    rm("test_output_lambda.pdf", force=true)
    rm("latexlambda.zip", force=true)
end


if length(ARGS) == 0
    all()
else
    include_string("$(ARGS[1])()")
end


end # module MakeLatexLambda