init

Author: tid

HTMLNode.swift

@@ -0,0 +1,210 @@
+/**@file
+ * @brief Swift-HTML-Parser
+ * @author _tid_
+ */
+
+/**
+ * HTMLNode
+ */
+class HTMLNode {
+    enum HTMLNodeType : String {
+        case HTMLUnkownNode     = ""
+        case HTMLHrefNode       = "href"
+        case HTMLTextNode       = "text"
+        case HTMLCodeNode       = "code"
+        case HTMLSpanNode       = "span"
+        case HTMLPNode          = "p"
+        case HTMLLiNode         = "li"
+        case HTMLUiNode         = "ui"
+        case HTMLImageNode      = "image"
+        case HTMLOlNode         = "ol"
+        case HTMLStrongNode     = "strong"
+        case HTMLPreNode        = "pre"
+        case HTMLBlockQuoteNode = "blockquote"
+    }
+    
+    var doc       : htmlDocPtr
+    var node      : xmlNode?
+    let nodeType  : HTMLNodeType = HTMLNodeType.HTMLUnkownNode
+    
+    /**
+     * 親ノード
+     */
+    var parent    : HTMLNode? {
+        if let p = self.node?.parent {
+            return HTMLNode(doc: self.doc, node: p)
+        }
+        return nil
+    }
+
+    /**
+    * 次ノード
+    */
+    var next : HTMLNode? {
+        if let n : UnsafePointer<xmlNode> = node?.next {
+            if n {
+                return HTMLNode(doc: doc, node: n)
+            }
+        }
+        return nil
+    }
+
+    /**
+     * 子ノード
+     */
+    var child     : HTMLNode? {
+        if let c = node?.children {
+            if c {
+                return HTMLNode(doc: doc, node: c)
+            }
+        }
+        return nil
+    }
+
+    /**
+     * クラス名
+     */
+    var className : String {
+        return getAttributeNamed("class")
+    }
+
+    /**
+    * タグ名
+    */
+    var tagName : String {
+        if node {
+            return ConvXmlCharToString(self.node!.name)
+        }
+        return ""
+    }
+    
+    /**
+     * コンテンツ
+     */
+    var contents : String {
+        if node {
+            var n = self.node!.children
+            if n {
+                return ConvXmlCharToString(n.memory.content)
+            }
+        }
+        return ""
+    }
+
+    /**
+     * Initializer
+     * @param[in] doc xmlDoc
+     */
+    init(doc: htmlDocPtr = nil) {
+        self.doc  = doc
+        var node = xmlDocGetRootElement(doc)
+        if node {
+            self.node = node.memory
+        }
+    }
+    
+    init(doc: htmlDocPtr, node: UnsafePointer<xmlNode>) {
+        self.doc  = doc
+        self.node = node.memory
+
+        if let type = HTMLNodeType.fromRaw(tagName) {
+            nodeType = type
+        }
+    }
+
+    /**
+     * 属性名を取得する
+     * @param[in] name 属性
+     * @return 属性名
+     */
+    func getAttributeNamed(name: String) -> String {
+        for var attr : xmlAttrPtr = node!.properties; attr; attr = attr.memory.next {
+            if attr {
+                var mem = attr.memory
+
+                if name == ConvXmlCharToString(mem.name) {
+                    return ConvXmlCharToString(mem.children.memory.content)
+                }
+            }
+        }
+        
+        return ""
+    }
+    
+    /**
+     * タグ名に一致する全ての子ノードを探す
+     * @param[in] tagName タグ名
+     * @return 子ノードの配列
+     */
+    func findChildTags(tagName: String) -> HTMLNode[] {
+        var nodes : HTMLNode[] = []
+        
+        return findChildTags(tagName, node: self.child, retAry: &nodes)
+    }
+    
+    func findChildTags(tagName: String, node: HTMLNode?, inout retAry: HTMLNode[] ) -> HTMLNode[] {
+        if node {
+            for n in node! {
+                if n.tagName == tagName {
+                    retAry.append(n)
+                }
+            
+                findChildTags(tagName, node: n.child, retAry: &retAry)
+            }
+        }
+        return retAry
+    }
+    
+    /**
+     * タグ名で子ノードを探す
+     * @param[in] tagName タグ名
+     * @return 子ノード。見つからなければnil
+     */
+    func findChildTag(tagName: String) -> HTMLNode? {
+        return findChildTag(tagName, node: self)
+    }
+
+    func findChildTag(tagName: String, node: HTMLNode?) -> HTMLNode? {
+        if node == nil {
+            return nil
+        }
+        
+        for curNode in node! {
+            if tagName == curNode.tagName {
+                return curNode
+            }
+            
+            if let c = curNode.child {
+                if let n = findChildTag(tagName, node: c) {
+                    return n
+                }
+            }
+        }
+
+        return nil
+    }
+}
+
+extension HTMLNode : Sequence {
+    func generate() -> HTMLNodeGenerator {
+        return HTMLNodeGenerator(node: self)
+    }
+}
+
+/**
+* HTMLNodeGenerator
+*/
+class HTMLNodeGenerator : Generator {
+    var node : HTMLNode?
+    
+    init(node: HTMLNode?) {
+        self.node = node
+    }
+    
+    func next() -> HTMLNode? {
+        var temp = node?
+        node = node?.next
+        return temp
+    }
+}
+

HTMLParser.swift

@@ -0,0 +1,70 @@
+/**@file
+ * @brief Swift-HTML-Parser
+ * @author _tid_
+ */
+import Foundation
+
+func ConvXmlCharToString(str: UnsafePointer<xmlChar>) -> String {
+    if str {
+        return String.fromCString(UnsafePointer<CChar>(str))
+    }
+    return ""
+}
+
+/**
+ * HTMLParser
+ */
+class HTMLParser {
+    var _doc     : htmlDocPtr = nil
+    var rootNode : HTMLNode?
+    
+    /**
+     * HTML tag
+     */
+    var html : HTMLNode? {
+        return rootNode?.findChildTag("html")
+    }
+    
+    /**
+     * HEAD tag
+     */
+    var head : HTMLNode? {
+        return rootNode?.findChildTag("head")
+    }
+    
+    /**
+     * BODY tag
+     */
+    var body : HTMLNode? {
+        return rootNode?.findChildTag("body")
+    }
+    
+    /**
+     * @param[in] html  HTML文字列
+     * @param[in] error エラーがあれば返します
+     */
+    init(html: String, inout error: NSError?) {
+        if html.lengthOfBytesUsingEncoding(NSUTF8StringEncoding) > 0 {
+            var cfenc : CFStringEncoding = CFStringConvertNSStringEncodingToEncoding(NSUTF8StringEncoding)
+            var cfencstr : CFStringRef   = CFStringConvertEncodingToIANACharSetName(cfenc)
+        
+            var cur : CChar[]? = html.cStringUsingEncoding(NSUTF8StringEncoding)
+            var url : CString = ""
+            var enc : CString = CFStringGetCStringPtr(cfencstr, 0)
+            let optionHtml : CInt = 1
+        
+            if var ucur = cur {
+                var temp : CMutablePointer<CChar> = &ucur
+                var aaa = UnsafePointer<CUnsignedChar>(temp.value)
+                _doc = htmlReadDoc(aaa, url, enc, optionHtml)
+                rootNode  = HTMLNode(doc: _doc)
+            }
+        } else {
+            error = NSError(domain: "HTMLParserdomain", code: 1, userInfo: nil)
+        }
+    }
+    
+    deinit {
+        xmlFreeDoc(_doc)
+    }
+}

Swift-HTML-Parser-Bridging-Header.h

@@ -0,0 +1,5 @@
+//
+//  Use this file to import your target's public headers that you would like to expose to Swift.
+//
+
+#import <libxml/HTMLtree.h>

main.swift

@@ -0,0 +1,18 @@
+import Foundation
+
+let html = "<ul><li><input type='image' name='input1' value='string1value' class='abc' /></li><li><input type='image' name='input2' value='string2value' class='def' /></li></ul><span class='spantext'><b>Hello World 1</b></span><span class='spantext'><b>Hello World 2</b></span>"
+
+var err : NSError?
+var parser     = HTMLParser(html: html, error: &err)
+if err {
+    println(err)
+    exit(1)
+}
+
+var bodyNode   = parser.body
+
+if let inputNodes = bodyNode?.findChildTags("b") {
+    for node in inputNodes {
+        println(node.contents)
+    }
+}