PSDText.cs 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
  1. using System;
  2. using System.Collections.Generic;
  3. using System.IO;
  4. using System.Linq;
  5. using System.Text;
  6. using System.Text.Json;
  7. using System.Xml;
  8. using System.Xml.Serialization;
  9. namespace PSDText
  10. {
  11. public class PSDText
  12. {
  13. private readonly string _xmlData;
  14. private readonly XmlNamespaceManager _ns = new(new NameTable());
  15. public List<TextData> ExtractedStrings;
  16. private static string Readxmpmeta(string path)
  17. {
  18. var sb = new StringBuilder(1000);
  19. using (var sr = new StreamReader(path))
  20. {
  21. var read = false;
  22. while (true)
  23. {
  24. var line = sr.ReadLine();
  25. if (string.IsNullOrWhiteSpace(line))
  26. return sb.ToString();
  27. if (line.StartsWith("<x:xmpmeta"))
  28. read = true;
  29. if (read) sb.Append(line);
  30. if (line.StartsWith("</x:xmpmeta>"))
  31. break;
  32. }
  33. }
  34. return sb.ToString();
  35. }
  36. private void AddXMLNamespaces()
  37. {
  38. _ns.AddNamespace("x", "adobe:ns:meta/");
  39. _ns.AddNamespace("rdf", "http://www.w3.org/1999/02/22-rdf-syntax-ns#");
  40. _ns.AddNamespace("photoshop", "http://ns.adobe.com/photoshop/1.0/");
  41. }
  42. private XmlDocument ReadXML()
  43. {
  44. var xml = new XmlDocument();
  45. using (var sr = new StringReader(_xmlData))
  46. using (var reader = XmlReader.Create(sr))
  47. {
  48. xml.Load(reader);
  49. }
  50. return xml;
  51. }
  52. private List<TextData> GetTextData()
  53. {
  54. var xml = ReadXML();
  55. var data = new List<TextData>();
  56. var textNodes =
  57. xml.SelectNodes("/x:xmpmeta/rdf:RDF/rdf:Description/photoshop:TextLayers/rdf:Bag/rdf:li", _ns);
  58. foreach (XmlNode textNode in textNodes)
  59. {
  60. var name = textNode.SelectSingleNode("./photoshop:LayerName", _ns)?.InnerText;
  61. if (string.IsNullOrWhiteSpace(name))
  62. continue;
  63. var text = textNode.SelectSingleNode("./photoshop:LayerText", _ns)?.InnerText;
  64. if (string.IsNullOrWhiteSpace(text))
  65. continue;
  66. data.Add(new TextData { Name = name, Text = text });
  67. }
  68. return data;
  69. }
  70. public PSDText(string path)
  71. {
  72. if (!File.Exists(path))
  73. throw new FileNotFoundException($"File {path} doesn't exist!");
  74. _xmlData = Readxmpmeta(path);
  75. if (string.IsNullOrWhiteSpace(_xmlData))
  76. throw new Exception("No data was read!");
  77. AddXMLNamespaces();
  78. ExtractedStrings = GetTextData();
  79. if (!ExtractedStrings.Any())
  80. throw new Exception("Nothing was read from XML!");
  81. }
  82. /// <summary>
  83. /// Serializes text layers as XML.
  84. /// </summary>
  85. /// <param name="path">Output XML path.</param>
  86. public void SaveAsXML(string path)
  87. {
  88. if (string.IsNullOrWhiteSpace(path))
  89. throw new ArgumentNullException(nameof(path));
  90. var xmlSerializer = new XmlSerializer(typeof(List<TextData>));
  91. using var fs = File.OpenWrite(path);
  92. xmlSerializer.Serialize(fs, ExtractedStrings);
  93. }
  94. /// <summary>
  95. /// Serializes text layers as JSON.
  96. /// </summary>
  97. /// <param name="path">Output JSON path.</param>
  98. public void SaveAsJSON(string path)
  99. {
  100. if (string.IsNullOrWhiteSpace(path))
  101. throw new ArgumentNullException(nameof(path));
  102. using var fs = File.OpenWrite(path);
  103. JsonSerializer.Serialize(fs, ExtractedStrings, typeof(List<TextData>));
  104. }
  105. }
  106. }