Hans Leidekker : mshtml: Implement HTMLStorage_setItem().

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 076281c016fdf5bc2955f62cff6090475c992bee
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=076281c016fdf5bc2955f62cff6090475c992bee

Author: Hans Leidekker <hans at codeweavers.com>
Date:   Mon May 30 10:29:47 2022 +0200

mshtml: Implement HTMLStorage_setItem().

Signed-off-by: Hans Leidekker <hans at codeweavers.com>
Signed-off-by: Jacek Caban <jacek at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/mshtml/htmlstorage.c | 113 +++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 111 insertions(+), 2 deletions(-)

diff --git a/dlls/mshtml/htmlstorage.c b/dlls/mshtml/htmlstorage.c
index 8ef92453b95..f7b92114622 100644
--- a/dlls/mshtml/htmlstorage.c
+++ b/dlls/mshtml/htmlstorage.c
@@ -338,11 +338,120 @@ static HRESULT WINAPI HTMLStorage_getItem(IHTMLStorage *iface, BSTR bstrKey, VAR
     return hres;
 }
 
+static HRESULT set_attribute(IXMLDOMElement *elem, const WCHAR *name, BSTR value)
+{
+    BSTR str;
+    VARIANT var;
+    HRESULT hres;
+
+    str = SysAllocString(name);
+    if(!str)
+        return E_OUTOFMEMORY;
+    V_VT(&var) = VT_BSTR;
+    V_BSTR(&var) = value;
+
+    hres = IXMLDOMElement_setAttribute(elem, str, var);
+    SysFreeString(str);
+    return hres;
+}
+
+static HRESULT save_document(IXMLDOMDocument *doc, const WCHAR *filename)
+{
+    VARIANT var;
+    HRESULT hres;
+
+    V_VT(&var) = VT_BSTR;
+    V_BSTR(&var) = SysAllocString(filename);
+    if(!V_BSTR(&var))
+        return E_OUTOFMEMORY;
+
+    hres = IXMLDOMDocument_save(doc, var);
+    SysFreeString(V_BSTR(&var));
+    return hres;
+}
+
+static HRESULT set_item(const WCHAR *filename, BSTR key, BSTR value)
+{
+    IXMLDOMDocument *doc;
+    IXMLDOMNode *root = NULL, *node = NULL;
+    IXMLDOMElement *elem = NULL;
+    BSTR query = NULL;
+    HRESULT hres;
+
+    hres = open_document(filename, &doc);
+    if(hres != S_OK)
+        return hres;
+
+    hres = get_root_node(doc, &root);
+    if(hres != S_OK)
+        goto done;
+
+    query = build_query(key);
+    if(!query) {
+        hres = E_OUTOFMEMORY;
+        goto done;
+    }
+
+    hres = IXMLDOMNode_selectSingleNode(root, query, &node);
+    if(hres == S_OK) {
+        hres = IXMLDOMNode_QueryInterface(node, &IID_IXMLDOMElement, (void**)&elem);
+        if(hres != S_OK)
+            goto done;
+
+        hres = set_attribute(elem, L"value", value);
+        if(hres != S_OK)
+            goto done;
+    }else {
+        BSTR str = SysAllocString(L"item");
+        hres = IXMLDOMDocument_createElement(doc, str, &elem);
+        SysFreeString(str);
+        if(hres != S_OK)
+            goto done;
+
+        hres = set_attribute(elem, L"name", key);
+        if(hres != S_OK)
+            goto done;
+
+        hres = set_attribute(elem, L"value", value);
+        if(hres != S_OK)
+            goto done;
+
+        hres = IXMLDOMNode_appendChild(root, (IXMLDOMNode*)elem, NULL);
+        if(hres != S_OK)
+            goto done;
+    }
+
+    hres = save_document(doc, filename);
+
+done:
+    SysFreeString(query);
+    if(root)
+        IXMLDOMNode_Release(root);
+    if(node)
+        IXMLDOMNode_Release(node);
+    if(elem)
+        IXMLDOMElement_Release(elem);
+    IXMLDOMDocument_Release(doc);
+    return hres;
+}
+
 static HRESULT WINAPI HTMLStorage_setItem(IHTMLStorage *iface, BSTR bstrKey, BSTR bstrValue)
 {
     HTMLStorage *This = impl_from_IHTMLStorage(iface);
-    FIXME("(%p)->(%s %s)\n", This, debugstr_w(bstrKey), debugstr_w(bstrValue));
-    return E_NOTIMPL;
+    HRESULT hres;
+
+    TRACE("(%p)->(%s %s)\n", This, debugstr_w(bstrKey), debugstr_w(bstrValue));
+
+    if(!This->filename) {
+        FIXME("session storage not supported\n");
+        return E_NOTIMPL;
+    }
+
+    WaitForSingleObject(This->mutex, INFINITE);
+    hres = set_item(This->filename, bstrKey, bstrValue);
+    ReleaseMutex(This->mutex);
+
+    return hres;
 }
 
 static HRESULT WINAPI HTMLStorage_removeItem(IHTMLStorage *iface, BSTR bstrKey)