''' Plugin for CudaText editor
Authors:
    Andrey Kvichansky (kvichans on github.com)
    Alexey Torgashin (CudaText)
Version:
    '1.0.2 2024-08-24'
'''

import  os
import  cudatext            as app
from    cudatext        import ed
import  cudatext_cmd        as cmds
import  cudax_lib           as apx
from    .cd_plug_lib    import *

# I18N
_       = get_translation(__file__)

pass;                           LOG     = (-1==-1)  # Do or don't write log

class Command:
    def __init__(self):
        self.pair4lex = {}
        #def __init__

    def dlg_config(self):
        save_bd_col = apx.get_opt('comment_save_column'         , False)
        at_min_bd   = apx.get_opt('comment_equal_column'        , False)
        move_down   = apx.get_opt('comment_move_down'           , True)
        skip_blank  = apx.get_opt('comment_skip_blank'          , False)
        by_1st      = apx.get_opt('comment_toggle_by_nonempty'  , False)

        save_s      = _('(Line commands) Try to keep text position after (un)commenting')
        save_h      = _('Try to replace only blank(s) to keep text positions:'
                        '\rUncommented lines:'
                        '\r····foo1'
                        '\r····foo2'
                        '\rCommented lines:'
                        '\r#···foo1'
                        '\r···#foo2'
                        )
        vert_s      = _('(Line "at non-space") If selected few lines, insert comment at maximal common indent')
        vert_h      = _('Use maximal common column of first non-blank char:'
                        '\rUncommented lines:'
                        '\r··foo1'
                        '\r····foo2'
                        '\r······foo3'
                        '\rCommented lines:'
                        '\r··#foo1'
                        '\r··#··foo2'
                        '\r··#····foo3'
                        )
        down_s      = _('(All) Move caret to next line')
        skip_s      = _('(Line commands) Skip blank lines')
        by1st_s     = _('"Toggle line comment" detects action by first non-blank line')

        aid,vals,chds   = dlg_wrapper(_('Configure commenting commands'), 610, 160,
             [dict(cid='save',tp='ch'   ,t=5    ,l=5    ,w=600      ,cap=save_s ,hint=save_h) #
             ,dict(cid='vert',tp='ch'   ,t=5+25 ,l=5    ,w=600      ,cap=vert_s ,hint=vert_h) #
             ,dict(cid='down',tp='ch'   ,t=5+50 ,l=5    ,w=600      ,cap=down_s             ) #
             ,dict(cid='skip',tp='ch'   ,t=5+75 ,l=5    ,w=600      ,cap=skip_s             ) #
             ,dict(cid='by1st',tp='ch'  ,t=5+100,l=5    ,w=600      ,cap=by1st_s            ) #
             ,dict(cid='!'   ,tp='bt'   ,t=130  ,l=610-165-5,w=80   ,cap=_('OK'), ex0='1'   ) # default
             ,dict(cid='-'   ,tp='bt'   ,t=130  ,l=610 -80-5,w=80   ,cap=_('Cancel')        )
             ], dict(save=save_bd_col
                    ,vert=at_min_bd
                    ,down=move_down
                    ,skip=skip_blank
                    ,by1st=by_1st
             ), focus_cid='save')
        if aid is None or aid=='-': return
        if vals['save'] != save_bd_col: apx.set_opt('comment_save_column'       , vals['save'])
        if vals['vert'] != at_min_bd:   apx.set_opt('comment_equal_column'      , vals['vert'])
        if vals['down'] != move_down:   apx.set_opt('comment_move_down'         , vals['down'])
        if vals['skip'] != skip_blank:  apx.set_opt('comment_skip_blank'        , vals['skip'])
        if vals['by1st'] != by_1st:     apx.set_opt('comment_toggle_by_nonempty', vals['by1st'])
       #def dlg_config

    def cmt_toggle_line_1st(self):
        return self.work('bgn', '1st')

    def cmt_add_line_1st(self):
        return self.work('add', '1st')

    def cmt_toggle_line_body(self):
        return self.work('bgn', 'bod')

    def cmt_add_line_body(self):
        return self.work('add', 'bod')

    def cmt_del_line(self):
        return self.work('del')

    def line_cmt_by_range_cmt(self, ed_, rng1, rng2, cmt_act, cmt_type):
        changed = 0
        carets = ed_.get_carets()
        for caret in reversed(carets):
            x, y, x1, y1 = caret
            if y1<0:
                indexes = [y]
            else:
                if (y, x)>(y1, x1):
                    x, y, x1, y1 = x1, y1, x, y
                if x1==0:
                    y1-=1
                indexes = range(y, y1+1)

            for index in indexes:
                line = ed_.get_text_line(index)
                line_x = line.lstrip()
                if not line_x:
                    continue
                indent = line[:len(line)-len(line_x)]
                commented1 = line.startswith(rng1) and line.endswith(rng2)
                commented2 = line_x.startswith(rng1) and line_x.endswith(rng2)
                if commented1 or commented2:
                    if cmt_act=='add':
                        continue
                    if commented1:
                        line_new = line[len(rng1): -len(rng2)]
                    elif commented2:
                        line_new = indent + line_x[len(rng1): -len(rng2)]
                else:
                    if cmt_act=='del':
                        continue
                    if cmt_type=='1st':
                        line_new = rng1+line+rng2
                    elif cmt_type=='bod':
                        line_new = indent+rng1+line_x+rng2
                    else:
                        continue
                ed_.set_text_line(index, line_new)
                changed += 1

        if changed:
            app.msg_status(_('Toggled commenting for %d line(s)')%changed)
            if len(carets)==1:
                x, y, x1, y1 = carets[0]
                if y1<0:
                    if apx.get_opt('comment_move_down', True):
                        apx._move_caret_down(x, y)
        else:
            app.msg_status(_('No commenting action was done'))

    def work(self, cmt_act, cmt_type='', ed_=ed):
        ''' Add/Remove line-comments
            Params
                cmt_act     'del' - uncomment all lines
                            'add' - comment all lines
                            'bgn' - comment or uncomment, detect it by the 1st selected line
                cmt_type    '1st' - at begin of line
                            'bod' - at first non-blank char
        '''
        lex = ed_.get_prop(app.PROP_LEXER_CARET)
        if not lex:
            return app.msg_status(_('Commenting requires an active lexer'))
        prop = app.lexer_proc(app.LEXER_GET_PROP, lex)
        if not prop:
            return
        cmt_sgn   = prop['c_line'].rstrip() # remove trailing space, we will force space anyway
        cmt_range = prop['c_str']
        pass; #log('cmt_type, lex, cmt_sgn={}', (cmt_type, lex, cmt_sgn))

        if not cmt_sgn:
            if cmt_range:
                self.line_cmt_by_range_cmt(ed_, cmt_range[0], cmt_range[1], cmt_act, cmt_type)
                return

            return app.msg_status(f(_('Lexer "{}" doesn\'t support "line comments"'), lex))

        # Analyze
        empty_sel   = False
        rWrks       = []
        use_rep_lines = True # use API replace_lines()
        y1,y2,lines = (-1, -1, []) if use_rep_lines else (None, None, None)
        pass;                  #LOG and log('ed_.get_sel_mode(),app.SEL_NORMAL,app.SEL_COLUMN={}', (ed_.get_sel_mode(),app.SEL_NORMAL,app.SEL_COLUMN))
        crts        = ed_.get_carets()
        if len(crts)>1:
            app.msg_status(_('Cannot handle multi-carets'))
            return
        if False:pass
        elif ed_.get_sel_mode() == app.SEL_NORMAL:
            empty_sel     = 1==len(crts) and -1==crts[0][3]
            for (cCrt, rCrt, cEnd, rEnd) in crts:
                # sort 2 pairs
                if rEnd>=0 and (rCrt, cCrt)>(rEnd, cEnd):
                    cCrt, rCrt, cEnd, rEnd = cEnd, rEnd, cCrt, rCrt
                # selection until start of line?
                if rEnd>0 and cEnd==0:
                    rEnd -= 1
                rWrks      += list(range(rCrt, rEnd+1))
            use_rep_lines  = use_rep_lines and 1==len(crts)
        elif ed_.get_sel_mode() == app.SEL_COLUMN:
            (cBgn
            ,rSelBgn
            ,cEnd
            ,rSelEnd)   = ed_.get_sel_rect()
            rWrks       = list(range(rSelBgn, rSelEnd+1))
        if not rWrks:
            rWrks       = [crts[0][1]]
        pass;                  #log('rWrks={}', (rWrks))
        y1,y2       = (rWrks[0],rWrks[-1]) if use_rep_lines else (y1,y2)
        pass;                  #LOG and log('y1,y2,lines={}', (y1,y2,lines))

        # read options
        save_bd_col = apx.get_opt('comment_save_column' , False)
        at_min_bd   = apx.get_opt('comment_equal_column', False)
        skip_blank  = apx.get_opt('comment_skip_blank', False)
        by_1st      = apx.get_opt('comment_toggle_by_nonempty', False)

        if by_1st:
            # find index of first non-blank line
            row1st = -1
            for i in range(y1, y2+1):
                sline = ed_.get_text_line(i)
                if sline and sline.strip():
                    row1st = i
                    break
            if row1st<0:
                app.msg_status(_('Cannot handle blank lines / multi-carets'))
                return
        else:
            row1st = rWrks[0]

        # do we need to 'comment' or 'uncomment'?
        do_uncmt    = ed_.get_text_line(row1st).lstrip().startswith(cmt_sgn) \
                        if cmt_act=='bgn' else \
                      True \
                        if cmt_act=='del' else \
                      False
        # work
        col_min_bd  = 1000 # infinity
        col_kept    = False # plugin applied the "Try to keep text position"
        if at_min_bd:
            for rWrk in rWrks:
                line        = ed_.get_text_line(rWrk)
                pos_body    = line.index(line.lstrip())
                pos_body    = len(line) if 0==len(line.lstrip()) else pos_body
                col_min_bd  = min(pos_body, col_min_bd)
                if 0==col_min_bd:
                    break # for rWrk
        blnks4cmt   = ' '*(len(cmt_sgn)+1) # +1 for space
        pass;                  #log('rWrks,do_uncmt, save_cols, at_min_bd, col_min_bd={}', (rWrks,do_uncmt,save_bd_col,at_min_bd,col_min_bd))

        for rWrk in rWrks:
            line    = ed_.get_text_line(rWrk)
            if skip_blank and not line.strip():
                lines += [line]
                continue
            pos_body= line.index(line.lstrip())
            pos_body= len(line) if 0==len(line.lstrip()) else pos_body
            pass;              #LOG and log('rWrk,pos_body,line={}', (rWrk,pos_body,line))
            if do_uncmt:
                # Uncomment!
                if not line[pos_body:].startswith(cmt_sgn):
                    # Already no comment
                    if use_rep_lines:
                        lines += [line]
                    continue    #for rWrk
                if False:pass
                elif len(line)==len(cmt_sgn): # and line.startswith(cmt_sgn)
                    line = ''
                elif save_bd_col and (' '==line[0] or
                                      ' '==line[pos_body+len(cmt_sgn)]):
                    # Before or after cmt_sgn must be blank
                    if line[pos_body:].startswith(cmt_sgn+' '):
                        line = line.replace(cmt_sgn+' ', blnks4cmt, 1)
                    else:
                        line = line.replace(cmt_sgn, blnks4cmt, 1)
                    col_kept = True
                else:
                    if line[pos_body:].startswith(cmt_sgn+' '):
                        line = line.replace(cmt_sgn+' ', '', 1)
                    else:
                        line = line.replace(cmt_sgn, '', 1)
            else:
                # Comment!
                if cmt_type=='bod' and line[pos_body:].startswith(cmt_sgn):
                    # Body comment already set - willnot double it
                    if use_rep_lines:
                        lines += [line]
                    continue    #for rWrk
                if False:pass
                elif cmt_type=='1st' and save_bd_col and line.startswith(blnks4cmt) :
                    line = line.replace(blnks4cmt, cmt_sgn+' ', 1)
                    col_kept = True
               #elif cmt_type=='1st' and save_bd_col #  !line.startswith(blnks4cmt) :
                elif cmt_type=='1st':#  !save_bd_col
                    line = cmt_sgn+' '+line
                elif cmt_type=='bod' and save_bd_col and line.startswith(blnks4cmt):
                    col_kept = True
                    pos_cmnt = col_min_bd if at_min_bd else pos_body
                    pass;          #LOG and log('pos_cmnt={}', (pos_cmnt))
                    if pos_cmnt>=len(cmt_sgn):
                        line = line[:pos_cmnt-len(cmt_sgn)]+cmt_sgn+' '+line[pos_cmnt:             ]
                    else:
                        line = line[:pos_cmnt             ]+cmt_sgn+' '+line[pos_cmnt+len(cmt_sgn):]
                   #line = line[:pos_cmnt-len(cmt_sgn)]+cmt_sgn+line[pos_cmnt:]
                   #line = line[:pos_body-len(cmt_sgn)]+cmt_sgn+line[pos_body:]
               #elif cmt_type=='bod' and save_bd_col #  !line.startswith(blnks4cmt) :
                elif cmt_type=='bod':#  !save_bd_col
                    pos_cmnt = col_min_bd if at_min_bd else pos_body
                    pass;      #LOG and log('pos_cmnt={}', (pos_cmnt))
                    line = line[:pos_cmnt]             +cmt_sgn+' '+line[pos_cmnt:]
                   #line = line[:pos_body]             +cmt_sgn+' '+line[pos_body:]

            pass;              #LOG and log('new line={}', (line))
            if use_rep_lines:
                lines += [line]
            else:
                pass;           #log('line={}',(line))
                ed_.set_text_line(rWrk, line)
            #for rWrk
        if use_rep_lines:
            pass;              #log('y1, y2, len(lines), lines={}',(y1, y2, len(lines), lines))
            if y1==y2:
                ed_.set_text_line(y1, lines[0])
            else:
                ed_.replace_lines(y1, y2, lines)
        # move caret down
        (cCrt, rCrt, cEnd, rEnd) = crts[0]
        move_down = apx.get_opt('comment_move_down', True) and (rCrt+1 < ed_.get_line_count())
        if empty_sel and move_down:
            apx._move_caret_down(cCrt, rCrt)
        # shift caret horizontally if it's on the same line
        if not move_down and empty_sel and not col_kept:
            dx = len(cmt_sgn)+1
            if do_uncmt:
                dx = -dx
            cCrt = max(0, cCrt+dx)
            ed_.set_caret(cCrt, rCrt)
       #def work

    def cmt_toggle_stream(self):
        '''Toggle stream comment'''
        if ed.get_sel_mode() != app.SEL_NORMAL:
            return app.msg_status(f(_('{} works only with normal selection'), _('Commenting')))
        lex     = ed.get_prop(app.PROP_LEXER_CARET)
        if not lex:
            return app.msg_status(_('Commenting requires an active lexer'))
        ((bgn_sgn, end_sgn), bOnlyLn) = self._get_cmt_pair(lex)
        if not bgn_sgn:
            return app.msg_status(f(_('No stream comment for lexer "{}"'), lex))
        crts    = ed.get_carets()
        pass;                  #LOG and log('lex, get_carets()={}', (lex, crts))
        pass;                  #LOG and log('(bgn_sgn,end_sgn),bOnlyLn,bUseFLn={}', ((bgn_sgn,end_sgn),bOnlyLn,bUseFLn))
        for icrt, (cCrt, rCrt, cEnd, rEnd) in enumerate(crts):
            pass;              #LOG and log('(cCrt, rCrt), (cEnd, rEnd)={}', ((cCrt, rCrt), (cEnd, rEnd)))
            empty_sel     = -1==rEnd
            bDrtSel     = -1==rEnd or (rCrt, cCrt)>(rEnd, cEnd)
            bEntireLn   = (rEnd>=0) and (cEnd==0) and (cCrt==0)
            bEntireLn1  = bEntireLn and abs(rEnd-rCrt)==1
            bEntireLn2  = bEntireLn and abs(rEnd-rCrt)>1
            if False:pass
            elif empty_sel:
                # Use full line
                line        = ed.get_text_line(rCrt)
                (cTx1, rTx1), (cTx2, rTx2) = (0, rCrt), (len(line), rCrt)
            elif bOnlyLn: # and not empty_sel
                # Only full lines
                rTx1, rTx2  = apx.minmax(rCrt, rEnd)
                line    = ed.get_text_line(rTx2)
                (cTx1, rTx1), (cTx2, rTx2) = (0, rTx1), (len(line), rTx2)
            elif empty_sel: # and not bUseFLn and not bOnlyLn
                continue
            else:
                (rTx1, cTx1), (rTx2, cTx2) = apx.minmax((rCrt, cCrt), (rEnd, cEnd))
            selTx   = ed.get_text_substr(cTx1, rTx1, cTx2, rTx2)
            pass;              #LOG and log('(rTx1, cTx1), (rTx2, cTx2), selTx={}', ((rTx1, cTx1), (rTx2, cTx2), repr(selTx)))
            do_uncmt= selTx.startswith(bgn_sgn) #and selTx.endswith(end_sgn)
                # don't check for ending of selection - for HTML and entire selected line(s)
            pass;              #LOG and log('do_uncmt={}', (do_uncmt))

            cNSel1, rNSel1, cNSel2, rNSel2 = None, None, None, None

            if False:pass
            elif not do_uncmt and bOnlyLn:
                # Comment whole lines
                ed.insert(0, rTx2+1, end_sgn+'\n')
                ed.insert(0, rTx1,   bgn_sgn+'\n')
                (cNSel1, rNSel1
                ,cNSel2, rNSel2)    = 0, rTx1, len(end_sgn), rTx2+2

            elif not do_uncmt:
                # Comment some block
                if bEntireLn1:
                    s = ed.get_text_line(rTx1)
                    ed.set_text_line(rTx1, bgn_sgn+' '+s+' '+end_sgn)
                    (cNSel1, rNSel1
                    ,cNSel2, rNSel2) = (0, rTx1, 0, rTx2)

                elif bEntireLn2:
                    ed.insert(0, rTx2, end_sgn+'\n')
                    ed.insert(0, rTx1, bgn_sgn+'\n')
                    (cNSel1, rNSel1
                    ,cNSel2, rNSel2) = (0, rTx1, 0, rTx2+2)

                else:
                    ed.insert(cTx2, rTx2, ' '+end_sgn)
                    ed.insert(cTx1, rTx1, bgn_sgn+' ')
                    if False:pass
                    elif rTx1==rTx2:
                        # sel into one row
                        (cNSel1, rNSel1
                        ,cNSel2, rNSel2)    = cTx1, rTx1, cTx2+len(bgn_sgn)+len(end_sgn)+2, rTx2
                    elif rTx1!=rTx2:
                        # sel ends on diff rows
                        (cNSel1, rNSel1
                        ,cNSel2, rNSel2)    = cTx1, rTx1, cTx2             +len(end_sgn)+1, rTx2

            elif do_uncmt and bOnlyLn:
                # UnComment
                ed.delete(0, rTx2, 0, rTx2+1)
                ed.delete(0, rTx1, 0, rTx1+1)
                (cNSel1, rNSel1
                ,cNSel2, rNSel2)    = 0, rTx1, len(ed.get_text_line(rTx2-2)), rTx2-2

            elif do_uncmt:
                # UnComment
                if selTx.endswith(end_sgn):
                    trail_space = ed.get_text_line(rTx2)[cTx2-len(end_sgn)-1]==' '
                    lead_space = ed.get_text_line(rTx1)[cTx1+len(bgn_sgn)]==' '
                    ed.delete(cTx2-len(end_sgn)-(1 if trail_space else 0), rTx2, cTx2, rTx2)
                    ed.delete(cTx1, rTx1, cTx1+len(bgn_sgn)+(1 if lead_space else 0), rTx1)
                    if False:pass
                    elif rTx1==rTx2:
                        # sel into one row
                        (cNSel1, rNSel1
                        ,cNSel2, rNSel2)    = cTx1, rTx1, cTx2-len(bgn_sgn)-len(end_sgn)-(1 if lead_space else 0)-(1 if trail_space else 0), rTx2
                    elif rTx1!=rTx2:
                        # sel ends on diff rows
                        (cNSel1, rNSel1
                        ,cNSel2, rNSel2)    = cTx1, rTx1, cTx2             -len(end_sgn)-(1 if trail_space else 0), rTx2

                elif bEntireLn1:
                    s = ed.get_text_line(rTx1)
                    if s.startswith(bgn_sgn):
                        s = s[len(bgn_sgn):]
                    if s.endswith(end_sgn):
                        s = s[:-len(end_sgn)]
                    ed.set_text_line(rTx1, s)
                    (cNSel1, rNSel1
                    ,cNSel2, rNSel2) = (0, rTx1, 0, rTx2)

                elif bEntireLn2:
                    ed.delete(0, rTx2-1, 0, rTx2)
                    ed.delete(0, rTx1, 0, rTx1+1)
                    (cNSel1, rNSel1
                    ,cNSel2, rNSel2) = (0, rTx1, 0, rTx2-2)

            pass;              #LOG and log('bDrtSel, (cNSel1, rNSel1), (cNSel2, rNSel2)={}', (bDrtSel, (cNSel1, rNSel1), (cNSel2, rNSel2)))
            if cNSel1 is not None:
                if bDrtSel:
                    ed.set_caret(cNSel2, rNSel2, cNSel1, rNSel1, app.CARET_SET_INDEX+icrt)
                else:
                    ed.set_caret(cNSel1, rNSel1, cNSel2, rNSel2, app.CARET_SET_INDEX+icrt)
           #for icrt

        move_down = apx.get_opt('comment_move_down', True)
        if len(crts)==1 and empty_sel and move_down:
            apx._move_caret_down(cCrt, rCrt)
            if bOnlyLn and not do_uncmt:
                crt=ed.get_carets()[0]; apx._move_caret_down(crt[0], crt[1])
                crt=ed.get_carets()[0]; apx._move_caret_down(crt[0], crt[1])
        #def cmt_toggle_stream

    def _get_cmt_pair(self, lex):
        ''' Return ((begin_sign, end_sign), only_lines)
                begin_sign    as '/*'
                end_sign      as '*/'
                only_lines    True if each of *_sign must be whole line
        '''
        if lex not in self.pair4lex:
            only_ln = False
            prop = app.lexer_proc(app.LEXER_GET_PROP, lex)
            pair1 = prop['c_str'] if prop else None
            pair2 = prop['c_lined'] if prop else None
            if pair1 is not None:
                pair = pair1
            elif pair2 is not None:
                pair = pair2
                only_ln = True
            else:
                pair = ('', '')
            self.pair4lex[lex] = (pair, only_ln)
        return self.pair4lex[lex]
       #def _get_cmt_pair
