add image to mitmproxy.contentview.image

2 files changed, 10 insertions, 4 deletions

diff --git a/web/src/js/components/ContentView/ContentViews.jsx b/web/src/js/components/ContentView/ContentViews.jsx
index 89e97267..a1bee54e 100644
--- a/web/src/js/components/ContentView/ContentViews.jsx
+++ b/web/src/js/components/ContentView/ContentViews.jsx
@@ -39,8 +39,11 @@ function ViewAuto({ message, flow, readonly, onChange }) {
     return <View message={message} flow={flow} readonly={readonly} onChange={onChange}/>
 }
 
-function ViewServer({content, contentView}){
+
+function ViewServer({content, contentView, message, flow}){
     let data = JSON.parse(content)
+    let showImage = isImage.test(MessageUtils.getContentType(message))
+
     return <div>
             {contentView != data.description &&
                 <div className="alert alert-warning">{data.description}</div>
@@ -56,6 +59,11 @@ function ViewServer({content, contentView}){
                     </div>
                 )}
             </pre>
+            {showImage &&
+                <div className="flowview-image">
+                    <img src={MessageUtils.getContentURL(flow, message)} alt="preview" className="img-thumbnail"/>
+                </div>
+            }
         </div>
 }
 
diff --git a/web/src/js/components/ContentView/ViewSelector.jsx b/web/src/js/components/ContentView/ViewSelector.jsx
index 423cc157..e031b51f 100644
--- a/web/src/js/components/ContentView/ViewSelector.jsx
+++ b/web/src/js/components/ContentView/ViewSelector.jsx
@@ -35,10 +35,8 @@ function ViewSelector({ message, contentViews }) {
     return (
         <div className="view-selector btn-group btn-group-xs">
 
-            <ViewButton name="ViewAuto">auto: {autoViewName}</ViewButton>
-
             {Object.keys(ContentViews).map(name =>
-                name !== "ViewAuto" && name !== "ViewServer" && 
+                name === "ViewRaw"  &&
                 <ViewButton key={name} name={name}>{name.toLowerCase().replace('view', '')}</ViewButton>
             )}