1
1
import React , { useContext } from 'react' ;
2
2
import { UIRuntime } from '@teambit/ui' ;
3
+ import { useQuery } from '@teambit/ui-foundation.ui.react-router.use-query' ;
3
4
import { ComponentModel } from '@teambit/component' ;
4
5
import { DocsAspect , DocsUI } from '@teambit/docs' ;
5
6
import { ComponentSize } from '@teambit/component.ui.component-size' ;
@@ -22,15 +23,20 @@ export class SizerUIRuntime {
22
23
docs . registerTitleBadge ( {
23
24
component : function badge ( { legacyComponentModel } : { legacyComponentModel : ComponentModel } ) {
24
25
const workspace = useContext ( WorkspaceContext ) ;
25
- const workspaceComponent = workspace ?. getComponent ( legacyComponentModel . id ) ;
26
+ const query = useQuery ( ) ;
27
+ const workspaceComponent = workspace ?. components . find ( ( component ) =>
28
+ component . id . isEqual ( legacyComponentModel . id )
29
+ ) ;
30
+ const componentVersionFromUrl = query . get ( 'version' ) ;
31
+ const isWorkspaceVersion = Boolean ( workspaceComponent && ! componentVersionFromUrl ) ;
26
32
const size = legacyComponentModel . size ;
27
33
const isModified = Boolean (
28
34
workspaceComponent ?. status ?. modifyInfo ?. hasModifiedFiles ||
29
35
workspaceComponent ?. status ?. modifyInfo ?. hasModifiedDependencies
30
36
) ;
31
37
const sizeExistsBuComponentModified = Boolean ( size && isModified ) ;
32
38
33
- if ( sizeExistsBuComponentModified ) {
39
+ if ( isWorkspaceVersion && sizeExistsBuComponentModified ) {
34
40
return (
35
41
< Tooltip
36
42
className = { styles . componentSizeTooltip }
0 commit comments