src/WixBuildTools.TestSupport/Query.cs


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

// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.

namespace WixBuildTools.TestSupport
{
    using System;
    using System.Collections.Generic;
    using System.Text;
    using WixToolset.Dtf.WindowsInstaller;

    public class Query
    {
        public static string[] QueryDatabase(string path, string[] tables)
        {
            var results = new List<string>();

            if (tables?.Length > 0)
            {
                var sb = new StringBuilder();
                using (var db = new Database(path))
                {
                    foreach (var table in tables)
                    {
                        if (!db.IsTablePersistent(table))
                        {
                            continue;
                        }

                        using (var view = db.OpenView($"SELECT * FROM `{table}`"))
                        {
                            view.Execute();

                            Record record;
                            while ((record = view.Fetch()) != null)
                            {
                                sb.Clear();
                                sb.AppendFormat("{0}:", table);

                                using (record)
                                {
                                    for (var i = 0; i < record.FieldCount; ++i)
                                    {
                                        if (i > 0)
                                        {
                                            sb.Append("\t");
                                        }

                                        sb.Append(record[i + 1]?.ToString());
                                    }
                                }

                                results.Add(sb.ToString());
                            }
                        }
                    }
                }
            }

            results.Sort();
            return results.ToArray();
        }
    }
}