feat: add oas 2.0 badge

src/core/components/info.jsx

@@ -56,6 +56,7 @@ class Info extends React.Component {
     getComponent: PropTypes.func.isRequired,
     oas3selectors: PropTypes.func,
     selectedServer: PropTypes.string,
+    oasVersion: PropTypes.string,
   }
 
   render() {
@@ -68,6 +69,7 @@ class Info extends React.Component {
       externalDocs,
       selectedServer,
       url: specUrl,
+      oasVersion
     } = this.props
     const version = info.get("version")
     const description = info.get("description")
@@ -99,7 +101,7 @@ class Info extends React.Component {
         <hgroup className="main">
           <h2 className="title">
             {title}
-            {version && <VersionStamp version={version}></VersionStamp>}
+            {version && <VersionStamp version={version} oasVersion={oasVersion}></VersionStamp>}
           </h2>
           {host || basePath ? (
             <InfoBasePath host={host} basePath={basePath} />

src/core/components/version-stamp.jsx

@@ -1,12 +1,22 @@
 import React from "react"
 import PropTypes from "prop-types"
 
-const VersionStamp = ({ version }) => {
-  return <small><pre className="version"> { version } </pre></small>
-}
+
+const VersionStamp = ({ version, oasVersion }) => (
+    <>
+      <small>
+        <pre className="version"> {version} </pre>
+    </small>
+    {oasVersion &&
+      <small className="version-stamp">
+        <pre className="version">OAS {oasVersion}</pre>
+      </small>}
+    </>
+)
 
 VersionStamp.propTypes = {
-  version: PropTypes.string.isRequired
+  version: PropTypes.string.isRequired,
+  oasVersion: PropTypes.string
 }
 
 export default VersionStamp

src/core/containers/info.jsx

@@ -19,14 +19,18 @@ export default class InfoContainer extends React.Component {
     const host = specSelectors.host()
     const externalDocs = specSelectors.externalDocs()
     const selectedServer = oas3Selectors.selectedServer()
+    const oasVersion = () => {
+    if (specSelectors.isSwagger2()) return "2.0"
+    if (specSelectors.isOAS30()) return "3.0"
+  }
 
     const Info = getComponent("info")
 
     return (
       <div>
         {info && info.count() ? (
           <Info info={info} url={url} host={host} basePath={basePath} externalDocs={externalDocs}
-                getComponent={getComponent} selectedServer={selectedServer} />
+                getComponent={getComponent} selectedServer={selectedServer} oasVersion={oasVersion()}/>
         ) : null}
       </div>
     )

src/core/plugins/oas3/wrap-components/index.js

@@ -1,6 +1,5 @@
 import Markdown from "./markdown"
 import AuthItem from "./auth-item"
-import VersionStamp from "./version-stamp"
 import OnlineValidatorBadge from "./online-validator-badge"
 import Model from "./model"
 import JsonSchema_string from "./json-schema-string"
@@ -9,7 +8,6 @@ export default {
   Markdown,
   AuthItem,
   JsonSchema_string,
-  VersionStamp,
   model: Model,
   onlineValidatorBadge: OnlineValidatorBadge,
 }

src/core/plugins/oas31/components/info.jsx

@@ -34,7 +34,7 @@ const Info = ({ getComponent, specSelectors }) => {
       <hgroup className="main">
         <h2 className="title">
           {title}
-          {version && <VersionStamp version={version}></VersionStamp>}
+          {version && <VersionStamp version={version} oasVersion="3.1"/>}
         </h2>
 
         {(host || basePath) && <InfoBasePath host={host} basePath={basePath} />}

src/core/plugins/oas31/index.js

@@ -15,7 +15,6 @@ import InfoWrapper from "./wrap-components/info"
 import ModelWrapper from "./wrap-components/model"
 import ModelsWrapper from "./wrap-components/models"
 import VersionPragmaFilterWrapper from "./wrap-components/version-pragma-filter"
-import VersionStampWrapper from "./wrap-components/version-stamp"
 import {
   isOAS31 as isOAS31Fn,
   createOnlyOAS31Selector as createOnlyOAS31SelectorFn,
@@ -91,7 +90,6 @@ const OAS31Plugin = ({ fn }) => {
       License: LicenseWrapper,
       Contact: ContactWrapper,
       VersionPragmaFilter: VersionPragmaFilterWrapper,
-      VersionStamp: VersionStampWrapper,
       Model: ModelWrapper,
       Models: ModelsWrapper,
       JSONSchema202012KeywordDescription:

test/e2e-cypress/e2e/features/oas3-badge.cy.js → test/e2e-cypress/e2e/features/oas-badge.cy.js

@@ -1,14 +1,24 @@
-describe("OpenAPI 3.x.y Badge", () => {
+describe("OpenAPI Badge", () => {
+  it("should display light green badge with version indicator for Swagger 2.0", () => {
+    cy.visit("/?url=/documents/features/info-openAPI2.yaml")
+      .get("#swagger-ui")
+      .get('*[class^="version-stamp"]')
+      .get("pre.version")
+      .contains("OAS 2.0")
+  })
+
   it("should display light green badge with version indicator for OpenAPI 3.0.x", () => {
     cy.visit("/?url=/documents/petstore-expanded.openapi.yaml")
       .get("#swagger-ui")
+      .get('*[class^="version-stamp"]')
       .get("pre.version")
       .contains("OAS 3.0")
   })
 
   it("should display light green badge with version indicator for OpenAPI 3.1.0", () => {
     cy.visit("/?url=/documents/features/info-openAPI31.yaml")
       .get("#swagger-ui")
+      .get('*[class^="version-stamp"]')
       .get("pre.version")
       .contains("OAS 3.1")
   })

test/unit/components/info-wrapper.jsx

@@ -14,7 +14,9 @@ describe("<InfoContainer/>", function () {
       url () {},
       basePath () {},
       host () {},
-      externalDocs () {},
+      externalDocs() {},
+      isSwagger2 () {},
+      isOAS30 () {}
     },
     oas3Selectors: {
       selectedServer () {},