summaryrefslogtreecommitdiff
path: root/tex/context/base/mkiv/util-sql-tracers.lua
blob: 44e32d256c6061d82f41b8034e7ad7c18862844e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
if not modules then modules = { } end modules ['util-sql-tracers'] = {
    version   = 1.001,
    comment   = "companion to m-sql.mkiv",
    author    = "Hans Hagen, PRAGMA-ADE, Hasselt NL",
    copyright = "PRAGMA ADE / ConTeXt Development Team",
    license   = "see context related readme files"
}

local sql     = utilities.sql
local tracers = { }
sql.tracers   = tracers

sql.setmethod("swiglib")

local gsub, lower = string.gsub, string.lower

local t_names = {
    mysql = [[SHOW TABLES FROM `%database%`]],
    mssql = [[SELECT table_name FROM %database%.information_schema.tables;]],
    mssql = [[SELECT "name" FROM "%database%"."sys"."databases" ORDER BY "name";]],
    mssql = [[SELECT name FROM "%database%"."sys"."objects" WHERE "type" IN ('P', 'U', 'V', 'TR', 'FN', 'TF');]],
}

local t_fields = {
    mysql = [[SHOW FIELDS FROM `%database%`.`%table%` ]],
    mssql = [[SELECT column_name "field", data_type "type", column_default "default", is_nullable "null" FROM %database%.information_schema.columns WHERE table_name='%table%']],
}

function sql.tracers.gettables(presets)
    local servertype = sql.getserver()

    local results, keys = sql.execute {
        presets   = presets,
        template  = t_names[servertype],
        variables = {
            database = presets.database,
        },
    }

    local key    = keys and keys[1]
    local tables = { }

    if keys then
        for i=1,#results do
            local name = results[i][key]
            local results, keys = sql.execute {
                presets   = presets,
                template  = t_fields[servertype],
                variables = {
                    database = presets.database,
                    table    = name
                },
            }
            if #results > 0 then
                for i=1,#results do
                    local result = table.loweredkeys(results[i])
                    -- ms cleanup
                    result.default = gsub(result.default,"^[%(']+(.-)[%)']+$","%1")
                    result.null    = lower(result.null)
                    --
                    results[i] = result
                end
                tables[name] = results
            else
                -- a view
            end
        end
    end

    return tables
end