AsyncAPI.NET/src/ByteBard.AsyncAPI.Readers/Services/DefaultStreamLoader.cs at 7a2ea912e4c1b57cd70b36c1a979790bfd8a32e2 · ByteBardOrg/AsyncAPI.NET · GitHub

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
namespace ByteBard.AsyncAPI.Readers.Services
{
    using System;
    using System.IO;
    using System.Net.Http;
    using System.Threading.Tasks;
    using ByteBard.AsyncAPI.Readers.Exceptions;
    using ByteBard.AsyncAPI.Readers.Interface;

    internal class DefaultStreamLoader : IStreamLoader
    {
        private static readonly HttpClient HttpClient = new HttpClient();

        public Stream Load(Uri uri)
        {
            try
            {
                if (uri.IsAbsoluteUri)
                {
                    switch (uri.Scheme.ToLowerInvariant())
                    {
                        case "file":
                            return File.OpenRead(uri.LocalPath);
                        case "http":
                        case "https":
                            return HttpClient.GetStreamAsync(uri).GetAwaiter().GetResult();
                        default:
                            throw new ArgumentException("Unsupported scheme");
                    }
                }
                else
                {
                    return File.OpenRead(uri.OriginalString);
                }
            }
            catch (Exception ex)
            {
                throw new AsyncApiReaderException($"Something went wrong trying to fetch '{uri.OriginalString}'. {ex.Message}", ex);
            }
        }

        public async Task<Stream> LoadAsync(Uri uri)
        {
            try
            {
                if (uri.IsAbsoluteUri)
                {
                    switch (uri.Scheme.ToLowerInvariant())
                    {
                        case "file":
                            return File.OpenRead(uri.AbsolutePath);
                        case "http":
                        case "https":
                            return await HttpClient.GetStreamAsync(uri);
                        default:
                            throw new ArgumentException("Unsupported scheme");
                    }
                }
                else
                {
                    return File.OpenRead(uri.OriginalString);
                }
            }
            catch (Exception ex)
            {
                throw new AsyncApiReaderException($"Something went wrong trying to fetch '{uri.OriginalString}'. {ex.Message}", ex);
            }
        }
    }
}