sourcecode.sty

\NeedsTeXFormat{LaTeX2e}[1995/06/01]
\ProvidesPackage{sourcecode}
    [2021/04/08 v0.2.1 Sourcecode Lua script]

% setup listings
\lstdefinestyle{sourcecodelua}{
    language={[5.3]Lua},
    basicstyle=\small\ttfamily,
    keywordstyle=\color{blue},
    stringstyle=\color{violet},
    commentstyle=\color{gray}
}

\lstdefinestyle{sourcecodeterm}{
    frame=tb,
    basicstyle=\small\ttfamily
}

\definecolor{colKeys}{rgb}{0,0,1}
\definecolor{colIdentifier}{rgb}{0,0,0}
\definecolor{colString}{rgb}{0.6,0.1,0.1}
\definecolor{mydarkblue}{rgb}{0.07,0.08,0.4}

\lstdefinestyle{sourcecodelatex}{
    language={[latex]tex},
    basicstyle=\small\ttfamily,
    identifierstyle=\color{colIdentifier},
    keywordstyle=\color{colKeys}\bfseries,
    stringstyle=\color{violet},
    commentstyle=\itshape\color{black!70},
    morekeywords=[2]{directlua,newcommand,approx,texttt,textbf},
    morestring=[b]",
    emph=[2]{document,tabular,article,tikzpiture},
    emphstyle=[2]{\color{mydarkblue}\bfseries},
    texcl=false
}

\directlua{
sourcecodelib = require [[sourcecode]]
sourcecodelib._tkprint = tex.print
}

% option run = true via attributes #[run]
% option tex via attributes #[tex]
% option lua via attributes #[lua]
% option index file attributes #[indexfile=<path>]
\newenvironment{lines}{\directlua{
    luatexbase.add_to_callback(
        'process_input_buffer',
        sourcecodelib.process_input_buffer,
        'sourcecodelib:lines',
        1
    )
}}{\directlua{
    luatexbase.remove_from_callback(
        'process_input_buffer',
        'sourcecodelib:lines'
    )
    sourcecodelib
        :trim_tail()
        :typeset()
}}

\newcommand\sourcecode[1]{\directlua{
    sourcecodelib
        :option{#1}
        :index_file()
        :load_file()
        :trim_tail()
        :typeset()
}}

\endinput