#region Properties

/// <summary>

/// File name being parsed for credentials

/// </summary>

internal string FileName { get; }

#endregion

#region Construction

/// <summary>

/// Initializes a new instance of the <see cref="PgPassFile"/> class

/// </summary>

/// <param name="fileName"></param>

public PgPassFile(string fileName)

=> FileName = fileName;

#endregion

/// <summary>

/// Parses file content and gets all credentials from the file

/// </summary>

/// <returns><see cref="IEnumerable{PgPassEntry}"/> corresponding to all lines in the .pgpass file</returns>

internal IEnumerable<Entry> Entries

{

get

{

var bytes = File.ReadAllBytes(FileName);

var mem = new MemoryStream(bytes);

using var reader = new StreamReader(mem);

while (reader.ReadLine() is { } l)

{

var line = l.Trim();

if (line.Length > 0 && line[0] != '#')

yield return Entry.Parse(line);

}

}

}

/// <summary>

/// Searches queries loaded from .PGPASS file to find first entry matching the provided parameters.

/// </summary>

/// <param name="host">Hostname to query. Use null to match any.</param>

/// <param name="port">Port to query. Use null to match any.</param>

/// <param name="database">Database to query. Use null to match any.</param>

/// <param name="username">User name to query. Use null to match any.</param>

/// <returns>Matching <see cref="Entry"/> if match was found. Otherwise, returns null.</returns>

internal Entry? GetFirstMatchingEntry(string? host = null, int? port = null, string? database = null, string? username = null)

{

foreach (var entry in Entries)

if (entry.IsMatch(host, port, database, username))

return entry;

return null;

}

/// <summary>

/// Represents a hostname, port, database, username, and password combination that has been retrieved from a .pgpass file

/// </summary>

internal sealed class Entry

{

#region Fields and Properties

/// <summary>

/// Hostname parsed from the .pgpass file

/// </summary>

internal string? Host { get; }

/// <summary>

/// Port parsed from the .pgpass file

/// </summary>

internal int? Port { get; }

/// <summary>

/// Database parsed from the .pgpass file

/// </summary>

internal string? Database { get; }

/// <summary>

/// User name parsed from the .pgpass file

/// </summary>

internal string? Username { get; }

/// <summary>

/// Password parsed from the .pgpass file

/// </summary>

internal string? Password { get; }

#endregion

#region Construction / Initialization

/// <summary>

/// This class represents an entry from the .pgpass file

/// </summary>

/// <param name="host">Hostname parsed from the .pgpass file</param>

/// <param name="port">Port parsed from the .pgpass file</param>

/// <param name="database">Database parsed from the .pgpass file</param>

/// <param name="username">User name parsed from the .pgpass file</param>

/// <param name="password">Password parsed from the .pgpass file</param>

Entry(string? host, int? port, string? database, string? username, string? password)

{

Host = host;

Port = port;

Database = database;

Username = username;

Password = password;

}

/// <summary>

/// Creates new <see cref="Entry"/> based on string in the format hostname:port:database:username:password. The : and \ characters should be escaped with a \.

/// </summary>

/// <param name="serializedEntry">string for the entry from the pgpass file</param>

/// <returns>New instance of <see cref="Entry"/> for the string</returns>

/// <exception cref="FormatException">Entry is not formatted as hostname:port:database:username:password or non-wildcard port is not a number</exception>

internal static Entry Parse(string serializedEntry)

{

var parts = new List<string?>(5);

var builder = new StringBuilder();

for (var pos = 0; pos < serializedEntry.Length; pos++)

{

var c = serializedEntry[pos];

switch (c)

{

case '\\' when pos < serializedEntry.Length - 1:

// Strip backslash before colon or backslash, otherwise preserve it

c = serializedEntry[++pos];

if (c is not (':' or '\\'))

{

builder.Append('\\');

}

builder.Append(c);

continue;

case ':':

var part = builder.ToString();

parts.Add(part == "*" ? null : part);

builder.Clear();

continue;

default:

builder.Append(c);

continue;

}

}

var lastPart = builder.ToString();

parts.Add(lastPart == "*" ? null : lastPart);

if (parts.Count != 5)

throw new FormatException("pgpass entry was not well-formed. Please ensure all non-comment entries are formatted as hostname:port:database:username:password. If colon is included, it must be escaped like \\:.");

int? port = null;

if (parts[1] != null)

{

if (!int.TryParse(parts[1], out var tempPort))

throw new FormatException("pgpass entry was not formatted correctly. Port must be a valid integer.");

port = tempPort;

}

return new Entry(parts[0], port, parts[2], parts[3], parts[4]);

}

#endregion

/// <summary>

/// Checks whether this <see cref="Entry"/> matches the parameters supplied

/// </summary>

/// <param name="host">Hostname to check against this entry</param>

/// <param name="port">Port to check against this entry</param>

/// <param name="database">Database to check against this entry</param>

/// <param name="username">Username to check against this entry</param>

/// <returns>True if the entry is a match. False otherwise.</returns>

internal bool IsMatch(string? host, int? port, string? database, string? username) =>

AreValuesMatched(host, Host) && AreValuesMatched(port, Port) && AreValuesMatched(database, Database) && AreValuesMatched(username, Username);

/// <summary>

/// Checks if 2 strings are a match for a <see cref="Entry"/> considering that either value can be a wildcard (*)

/// </summary>

/// <param name="query">Value being searched</param>

/// <param name="actual">Value from the PGPASS entry</param>

/// <returns>True if the values are a match. False otherwise.</returns>

bool AreValuesMatched(string? query, string? actual)

=> query == actual || actual == null || query == null;

bool AreValuesMatched(int? query, int? actual)

=> query == actual || actual == null || query == null;

}