/***************************************************************************
                    ansigenerator.cpp  -  description
                             -------------------
    begin                : Jul 5 2004
    copyright            : (C) 2004-2026 by Andre Simon
    email                : a.simon@mailbox.org
 ***************************************************************************/


/*
This file is part of Highlight.

Highlight is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

Highlight is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with Highlight.  If not, see <http://www.gnu.org/licenses/>.
*/


#include <string_view>

#include "ansigenerator.h"

using std::string;
using std::string_view;
using std::to_string;

namespace highlight
{

string  AnsiGenerator::getOpenTag ( string_view font,
                                    string_view fgCol, string_view bgCol )
{
    string s = "\033[";
    s += font;
    if ( !fgCol.empty() ) {
        s += ";";
        s += fgCol;
    }
    if ( !bgCol.empty() ) {
        s += ";";
        s += bgCol;
    }
    s += "m";
    return s;
}


string  AnsiGenerator::getOpenTag ( const ElementStyle &elem,
                                    string_view font, string_view fgCol )
{
    int idx = elem.getAnsiIndex();
    if ( idx < 0 ) return getOpenTag ( font, fgCol );
    return getOpenTag ( font, to_string ( idx<8 ? 30+idx : 90+idx-8 ) );
}


AnsiGenerator::AnsiGenerator() : CodeGenerator ( ESC_ANSI )
{
    newLineTag = "\n";
    spacer = initialSpacer = " ";
}

void AnsiGenerator::initOutputTags()
{
    openTags = {
        getOpenTag ( docStyle.getDefaultStyle(), "00", "39" ),
        getOpenTag ( docStyle.getStringStyle(), "00", "31" ), //str
        getOpenTag ( docStyle.getNumberStyle(), "00", "34" ), //number
        getOpenTag ( docStyle.getSingleLineCommentStyle(), "00", "34" ), //sl comment
        getOpenTag ( docStyle.getCommentStyle(), "00", "34" ), //ml comment
        getOpenTag ( docStyle.getEscapeCharStyle(), "00", "35" ), //escapeChar
        getOpenTag ( docStyle.getPreProcessorStyle(), "00", "35" ), //directive
        getOpenTag ( docStyle.getPreProcStringStyle(), "00", "31" ), //directive string
        getOpenTag ( docStyle.getLineStyle(), "00", "39" ), //linenum
        getOpenTag ( docStyle.getOperatorStyle(), "00", "39" ), //symbol
        getOpenTag ( docStyle.getInterpolationStyle(), "00", "35" ), //interpolation
        getOpenTag ( docStyle.getErrorStyle(), "01", "31" ), //error
        getOpenTag ( docStyle.getErrorMessageStyle(), "01", "31" )  //warning
    };

    closeTags.assign ( NUMBER_BUILTIN_STATES, "\033[m" );
    closeTags[0] = "";

    // basic ANSI has no theme-driven RGB guide colour, so fall back to a
    // faint/dim variant of the default foreground colour
    guideTagOpen = getOpenTag ( "02", "39" );
    guideTagClose = "\033[m";
}

string AnsiGenerator::getHeader()
{
    return string();
}

void AnsiGenerator::printBody()
{
    processRootState();
}

string AnsiGenerator::getFooter()
{
    return string();
}

string AnsiGenerator::maskCharacter ( unsigned char c )
{
    switch ( c ) {
        case 0x1b :
            return "";
        case 0x07 :
            return "";
        default:
            return string ( 1, c );
    }
}

string AnsiGenerator::getKeywordOpenTag ( unsigned int styleID )
{
    return getOpenTag ( docStyle.getKeywordStyle ( currentSyntax->getKeywordClasses() [styleID] ),
                        "00", ( styleID ) ?"32" :"33" );
}

string AnsiGenerator::getKeywordCloseTag ( unsigned int styleID )
{
    return "\033[m";
}

}