#coding:utf-8

"""
ID:          issue-4862
ISSUE:       4862
TITLE:       Allow hiding source code of procedures and triggers in FB 3
DESCRIPTION:
JIRA:        CORE-4544
FBTEST:      bugs.core_4544
"""

import pytest
from firebird.qa import *

db = db_factory()

test_script = """
    recreate table test(id int primary key, x int);
    recreate table tlog(id int, x int);

    recreate sequence g;

    set term ^;
    create or alter procedure sp_test1 as
        declare c int;
    begin
        select count(*) from rdb$types into c;
    end
    ^

    create or alter function fn_test1 returns int as
    begin
      return 111;
    end
    ^

    create or alter trigger trg_test_1 active before insert on test as
    begin
        new.id = gen_id(g,1);
    end
    ^

    create or alter package pkg_test1 as
    begin
        procedure sp_test1;
        function fn_test1 returns int;
    end
    ^

    recreate package body pkg_test1 as
    begin
        procedure sp_test1 as
            declare c int;
        begin
            select count(*) from rdb$types into c;
        end

        function fn_test1 returns int as
        begin
           return 111;
        end
    end
    ^

    set term ;^
    commit;

    set blob all;
    set list on;

    select rdb$procedure_name, rdb$procedure_source as source_blob
    from rdb$procedures where rdb$system_flag is distinct from 1;

    select rdb$function_name, rdb$function_source as source_blob
    from rdb$functions where rdb$system_flag is distinct from 1;

    select rdb$trigger_name, rdb$trigger_source as source_blob
    from rdb$triggers where rdb$system_flag is distinct from 1;

    select rdb$package_name, rdb$package_body_source as source_blob
    from rdb$packages where rdb$system_flag is distinct from 1;

    set count on;
    set echo on;

    update rdb$procedures set rdb$procedure_source = null
    where
        rdb$system_flag is distinct from 1
        and rdb$procedure_source is not null
    ;

    update rdb$functions set rdb$function_source = null
    where
        rdb$system_flag is distinct from 1
        and rdb$function_source is not null
    ;

    update rdb$triggers set rdb$trigger_source = null
    where
        rdb$system_flag is distinct from 1
        and rdb$trigger_source is not null
    ;

    update rdb$packages set rdb$package_body_source = null
    where
        rdb$system_flag is distinct from 1
        and rdb$package_body_source is not null
    ;

    set echo off;
    commit;


    select rdb$procedure_name, rdb$procedure_source as source_blob
    from rdb$procedures
    where rdb$system_flag is distinct from 1 and rdb$procedure_source is not null;

    select rdb$function_name, rdb$function_source as source_blob
    from rdb$functions
    where rdb$system_flag is distinct from 1 and rdb$function_source is not null;

    select rdb$trigger_name, rdb$trigger_source as source_blob
    from rdb$triggers
    where rdb$system_flag is distinct from 1 and rdb$trigger_source is not null;

    select rdb$package_name, rdb$package_body_source as source_blob
    from rdb$packages where rdb$system_flag is distinct from 1 and rdb$package_body_source is not null;
"""

act = isql_act('db', test_script, substitutions=[('SOURCE_BLOB.*', '')])

expected_stdout = """
    RDB$PROCEDURE_NAME              SP_TEST1
    SOURCE_BLOB                     1a:1e0
    declare c int;
    begin
        select count(*) from rdb$types into c;
    end

    RDB$PROCEDURE_NAME              SP_TEST1
    SOURCE_BLOB                     <null>



    RDB$FUNCTION_NAME               FN_TEST1
    SOURCE_BLOB                     e:1e0
    begin
      return 111;
    end

    RDB$FUNCTION_NAME               FN_TEST1
    SOURCE_BLOB                     <null>



    RDB$TRIGGER_NAME                TRG_TEST_1
    SOURCE_BLOB                     c:3d0
    as
    begin
        new.id = gen_id(g,1);
    end



    RDB$PACKAGE_NAME                PKG_TEST1
    SOURCE_BLOB                     2a:1
    begin
        procedure sp_test1 as
            declare c int;
        begin
            select count(*) from rdb$types into c;
        end

        function fn_test1 returns int as
        begin
           return 111;
        end
    end



    update rdb$procedures set rdb$procedure_source = null
    where
        rdb$system_flag is distinct from 1
        and rdb$procedure_source is not null
    ;
    Records affected: 1

    update rdb$functions set rdb$function_source = null
    where
        rdb$system_flag is distinct from 1
        and rdb$function_source is not null
    ;
    Records affected: 1

    update rdb$triggers set rdb$trigger_source = null
    where
        rdb$system_flag is distinct from 1
        and rdb$trigger_source is not null
    ;
    Records affected: 1

    update rdb$packages set rdb$package_body_source = null
    where
        rdb$system_flag is distinct from 1
        and rdb$package_body_source is not null
    ;
    Records affected: 1

    set echo off;
    Records affected: 0
    Records affected: 0
    Records affected: 0
    Records affected: 0
"""

@pytest.mark.version('>=3.0')
def test_1(act: Action):
    act.expected_stdout = expected_stdout
    act.execute()
    assert act.clean_stdout == act.clean_expected_stdout