public async Task <ActionResult <HostsEntryDto> > Get([FromRoute] string name)
{
using (var fileStream = System.IO.File.Open(Program.PathToHostsFile, FileMode.Open, FileAccess.Read, FileShare.Read))
{
string fileText;
var utf8EncodingWithoutBom = new UTF8Encoding(false);
using (var sr = new StreamReader(fileStream, utf8EncodingWithoutBom, false, 4096, true))
{
fileText = await sr.ReadToEndAsync();
}
var hostRegex = new Regex(@"^\s*([^#][^\s^#]+)\s*([^#][^\s^#]+)", RegexOptions.Multiline); // captures both IP and hostname
var ipAddresses = hostRegex.Matches(fileText)
.Select(m => m.Captures[0].Value)
.ToArray();
if (ipAddresses.Length > 0)
{
var result = new HostsEntryDto {
Host = name, IpAddresses = ipAddresses
};
return(Ok(result));
}
return(NotFound());
}
}
public void ToModelsCreatesModels()
{
HostsEntryDto dto = GetTestDto();
HostsEntry[] models = HostsEntryConverter.ToModels(dto);
Assert.Equal(dto.Host, models[0].Host);
Assert.Equal(dto.IpAddresses[0], models[0].IpAddress.ToString());
}
public void ToDtoCreatesADto()
{
HostsEntry model = GetTestModel();
HostsEntryDto dto = HostsEntryConverter.ToDto(new[] { model });
Assert.Equal(model.Host, dto.Host);
Assert.Equal(model.IpAddress, IPAddress.Parse(dto.IpAddresses[0]));
}
public async Task <ActionResult <HostsEntryDto> > Put([FromRoute] string name, [FromBody] HostsEntryDto hostsEntry)
{
if (name != hostsEntry.Host)
{
return(BadRequest());
}
using (var fileStream = System.IO.File.Open(Program.PathToHostsFile, FileMode.Open, FileAccess.ReadWrite, FileShare.Read))
{
string[] fileLines;
var utf8EncodingWithoutBom = new UTF8Encoding(false);
using (var sr = new StreamReader(fileStream, utf8EncodingWithoutBom, false, 4096, true))
{
fileLines = (await sr.ReadToEndAsync())
.Split('\n')
.Select(s => s.Trim('\r'))
.ToArray();
}
//search through each line in the HOSTS file to find the host to edit
var hostRegex = new Regex(@"^\s*([^#][^\s^#]+)\s*" + name, RegexOptions.IgnoreCase);
//remove all matched host lines
fileLines = fileLines.Where(l => !hostRegex.IsMatch(l)).ToArray();
// remove any blank lines at the end of the file
int i;
for (i = fileLines.Length - 1; i >= 0; i--)
{
if (!string.IsNullOrWhiteSpace(fileLines[i]))
{
break;
}
}
fileLines = fileLines
.Take(i + 1)
.ToArray();
// add new lines and append a single blank line at the end of the file
fileLines = fileLines
.Concat(hostsEntry.IpAddresses.Select(ip => $"\t{ip}\t\t{name}"))
.Append("")
.ToArray();
//write out file data
fileStream.Position = 0;
using (var sw = new StreamWriter(fileStream, utf8EncodingWithoutBom, 4096, false))
{
await sw.WriteAsync(string.Join(Environment.NewLine, fileLines));
await fileStream.FlushAsync();
}
}
return(CreatedAtAction(nameof(Get), new { name = name }, hostsEntry));
}
