import Foundation
import Aztec


// MARK: - VideoAttachment
//
extension VideoAttachment {

    @objc var videoPressID: String? {
        get {
            return extraAttributes[VideoShortcodeProcessor.videoPressHTMLAttribute]?.toString()
        }
        set {
            if let nonNilValue = newValue {
                extraAttributes[VideoShortcodeProcessor.videoPressHTMLAttribute] = .string(nonNilValue)
            } else {
                extraAttributes.remove(named: VideoShortcodeProcessor.videoPressHTMLAttribute)
            }
        }
    }

    @objc var isShortcode: Bool {
        get {
            return extraAttributes[VideoShortcodeProcessor.videoWPShortcodeHTMLAttribute]?.toString() == "true"
        }
        set {
            if newValue {
                extraAttributes[VideoShortcodeProcessor.videoWPShortcodeHTMLAttribute] = .string(String("true"))
            } else {
                extraAttributes.remove(named: VideoShortcodeProcessor.videoWPShortcodeHTMLAttribute)
            }
        }
    }
}

import Foundation
import Aztec


// MARK: - VideoAttachment
//
extension VideoAttachment {

    @objc var videoPressID: String? {
        get {
            return extraAttributes[VideoShortcodeProcessor.videoPressHTMLAttribute]?.toString()
        }
        set {
            if let nonNilValue = newValue {
                extraAttributes[VideoShortcodeProcessor.videoPressHTMLAttribute] = .string(nonNilValue)
            } else {
                extraAttributes.remove(named: VideoShortcodeProcessor.videoPressHTMLAttribute)
            }
        }
    }
}

--- 
+++ 
@@ -18,4 +18,17 @@
             }
         }
     }
+
+    @objc var isShortcode: Bool {
+        get {
+            return extraAttributes[VideoShortcodeProcessor.videoWPShortcodeHTMLAttribute]?.toString() == "true"
+        }
+        set {
+            if newValue {
+                extraAttributes[VideoShortcodeProcessor.videoWPShortcodeHTMLAttribute] = .string(String("true"))
+            } else {
+                extraAttributes.remove(named: VideoShortcodeProcessor.videoWPShortcodeHTMLAttribute)
+            }
+        }
+    }
 }
Add helper method to detect if video is a WP shortcode.
