prompt-extractor.user.js

이미지로부터 Exif 정보를 추출해 사용자에게 보여줍니다

当前为 2023-02-27 提交的版本,查看 最新版本

  1. // ==UserScript==
  2. // @name prompt-extractor.user.js
  3. // @namespace https://github.com/toriato/userscripts/prompt-extractor.user.js
  4. // @version 0.1.0
  5. // @description 이미지로부터 Exif 정보를 추출해 사용자에게 보여줍니다
  6. // @author Sangha Lee <totoriato@gmail.com>
  7. // @license MIT
  8. // @match https://arca.live/b/*
  9. // @run-at document-start
  10. // @grant GM_xmlhttpRequest
  11. // @grant GM_addStyle
  12. // @require https://cdn.jsdelivr.net/npm/exifreader@4.9.2/dist/exif-reader.min.js
  13. // ==/UserScript==
  14.  
  15. GM_addStyle(/*css*/`
  16. @keyframes spin {
  17. from { transform:rotate(0deg) }
  18. to { transform:rotate(360deg) }
  19. }
  20.  
  21. figure.params {
  22. margin: 0;
  23. position: relative;
  24. display: table;
  25. }
  26. figure.params img {
  27. max-width: 100%;
  28. }
  29.  
  30. /* 우측 상단 상태 아이콘 */
  31. figure.params:not([data-params=""])::after {
  32. position: absolute;
  33. right: 0;
  34. top: 0;
  35. margin: .5em;
  36. font-size: 2rem;
  37. text-shadow: 0 0 4px black;
  38. content: '❤️'
  39. }
  40. figure.params.loading::after {
  41. animation: spin 1s infinite linear;
  42. content: '🌀'
  43. }
  44. figure.params:not(.loading):not([data-params=""]):hover::after {
  45. display: none;
  46. }
  47.  
  48. figure.params figcaption {
  49. transition: transform .25s, opacity .25s;
  50. transform: scaleY(0);
  51. transform-origin: top;
  52. position: absolute;
  53. left: 0;
  54. top: 0;
  55. overflow-y: auto;
  56. max-height: 50%;
  57. padding: .5em;
  58. opacity: 0;
  59. background-color: rgba(0, 0, 0, 0.5);
  60. text-align: left;
  61. pointer-events: none;
  62. }
  63. figure.params:not(.loading):not([data-params=""]):hover figcaption {
  64. transform: scaleY(1);
  65. opacity: 1;
  66. pointer-events: inherit;
  67. }
  68. `)
  69.  
  70. /**
  71. * UPNG.js - JS PNG Decoder/Encoder
  72. * https://github.com/photopea/UPNG.js
  73. * MIT License
  74. */
  75. class UPNG {
  76. static bin = {
  77. nextZero: (data, p) => {
  78. while (data[p] != 0) p++
  79. return p
  80. },
  81. readUshort: (buff, p) =>
  82. (buff[p] << 8) | buff[p + 1],
  83. writeUshort: (buff, p, n) => {
  84. buff[p] = (n >> 8) & 255
  85. buff[p + 1] = n & 255
  86. },
  87. readUint: (buff, p) =>
  88. (buff[p] * (256 * 256 * 256)) + ((buff[p + 1] << 16) | (buff[p + 2] << 8) | buff[p + 3]),
  89. writeUint: (buff, p, n) => {
  90. buff[p] = (n >> 24) & 255
  91. buff[p + 1] = (n >> 16) & 255
  92. buff[p + 2] = (n >> 8) & 255
  93. buff[p + 3] = n & 255
  94. },
  95. readASCII: (buff, p, l) => {
  96. let s = ''
  97. for (let i = 0; i < l; i++)
  98. s += String.fromCharCode(buff[p + i])
  99. return s
  100. },
  101. writeASCII: (data, p, s) => {
  102. for (let i = 0; i < s.length; i++)
  103. data[p + i] = s.charCodeAt(i)
  104. },
  105. readBytes: (buff, p, l) => {
  106. const arr = []
  107. for (let i = 0; i < l; i++)
  108. arr.push(buff[p + i]);
  109. return arr
  110. },
  111. pad: (n) =>
  112. n.length < 2 ? '0' + n : n,
  113. readUTF8: function (buff, p, l) {
  114. let s = ''
  115. let ns;
  116. for (var i = 0; i < l; i++) s += "%" + UPNB.bin.pad(buff[p + i].toString(16));
  117. try { ns = decodeURIComponent(s); }
  118. catch (e) { return UPNG.bin.readASCII(buff, p, l); }
  119. return ns;
  120. }
  121. }
  122.  
  123. static magicNumbers = [0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a]
  124.  
  125. static decode(buff) {
  126. const bin = UPNG.bin
  127. const data = new Uint8Array(buff)
  128. const texts = {}
  129.  
  130. for (let i = 0; i < 8; i++) {
  131. if (data[i] !== UPNG.magicNumbers[i]) {
  132. throw Error('Input file is not a PNG')
  133. }
  134. }
  135.  
  136. let offset = 8
  137. while (offset < data.length) {
  138. const len = bin.readUint(data, offset); offset += 4;
  139. const type = bin.readASCII(data, offset, 4); offset += 4;
  140.  
  141. if (type == 'tEXt') {
  142. const nz = bin.nextZero(data, offset);
  143. const key = bin.readASCII(data, offset, nz - offset);
  144. const textLen = offset + len - nz - 1;
  145. texts[key] = bin.readASCII(data, nz + 1, textLen)
  146. }
  147.  
  148. offset += len + 4;
  149. }
  150.  
  151. return texts
  152. }
  153. }
  154.  
  155. /**
  156. * @param {UIEvent} event
  157. */
  158. function onLoad(event) {
  159. /** @type {HTMLImageElement} */
  160. const node = event.target
  161.  
  162. // 작은 이미지는 메타데이터 확인하지 않기
  163. const rect = node.getBoundingClientRect()
  164. if (rect.width < 128 || rect.height < 128) {
  165. return
  166. }
  167.  
  168. let src = new URL(node.src)
  169.  
  170. // 아카라이브에선 원본 이미지에만 Exif 데이터가 존재함
  171. if (src.host.endsWith('namu.la')) {
  172. src.searchParams.set('type', 'orig')
  173. }
  174.  
  175. const $figure = document.createElement('figure')
  176. $figure.classList.add('params', 'loading')
  177. $figure.innerHTML = /*html*/`
  178. ${node.closest('p').innerHTML}
  179. <figcaption></figcaption>
  180. `
  181.  
  182. node.replaceWith($figure)
  183.  
  184. // Exif 로부터 파라미터 문자열 가져오기
  185. let params = ''
  186.  
  187. new Promise((resolve, reject) => {
  188. GM_xmlhttpRequest({
  189. url: src.toString(),
  190. responseType: 'arraybuffer',
  191. onload: resolve,
  192. onerror: reject
  193. })
  194. })
  195. .then(res => {
  196. const headers = Object.fromEntries(
  197. res.responseHeaders
  198. .split(/\r?\n/)
  199. .map(v => {
  200. const [key, value] = v.split(':', 2).map(v => v.trim())
  201. return [key.toLowerCase(), value]
  202. })
  203. )
  204.  
  205. const contentType = headers['content-type']
  206. switch (contentType) {
  207. case 'image/png':
  208. const texts = UPNG.decode(res.response)
  209. params = texts['parameters'] ?? ''
  210. break
  211.  
  212. default:
  213. // 반환 받은 파일이 이미지가 아니라면 무시하기
  214. if (!contentType.startsWith('image/')) {
  215. return
  216. }
  217.  
  218. try {
  219. const tags = ExifReader.load(res.response)
  220.  
  221. if (tags?.UserComment?.value) {
  222. params = String.fromCharCode(
  223. // 첫 8바이트는 인코딩 타입이므로 디코딩 할 필요 없음
  224. // https://www.awaresystems.be/imaging/tiff/tifftags/privateifd/exif/usercomment.html
  225. ...tags.UserComment.value.slice(8).filter(v => v !== 0)
  226. )
  227. }
  228.  
  229. } catch (err) {
  230. // 메타데이터가 존재하지 않는 이미지라면 무시하기
  231. if (err.name !== 'MetadataMissingError') {
  232. throw err
  233. }
  234. }
  235. }
  236. })
  237. .finally(() => {
  238. $figure.classList.remove('loading')
  239. $figure.dataset.params = params
  240. $figure.querySelector('figcaption').innerHTML = params
  241. })
  242. }
  243.  
  244. const observer = new MutationObserver(mutations => {
  245. for (let mutation of mutations) {
  246. for (let node of mutation.addedNodes) {
  247. // 노드가 이미지 태그가 아니라면 무시하기
  248. if (!(node instanceof HTMLImageElement)) {
  249. continue
  250. }
  251.  
  252. node.addEventListener('load', onLoad)
  253. }
  254. }
  255. })
  256.  
  257. observer.observe(
  258. document,
  259. {
  260. attributes: true,
  261. childList: true,
  262. subtree: true
  263. }
  264. )