37 lines
1.0 KiB
TypeScript
37 lines
1.0 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import React, { useEffect, useState } from 'react';
|
|
import { Box, Text } from 'ink';
|
|
import { Colors } from '../colors.js';
|
|
import process from 'node:process';
|
|
import { formatMemoryUsage } from '../utils/formatters.js';
|
|
|
|
export const MemoryUsageDisplay: React.FC = () => {
|
|
const [memoryUsage, setMemoryUsage] = useState<string>('');
|
|
const [memoryUsageColor, setMemoryUsageColor] = useState<string>(Colors.Gray);
|
|
|
|
useEffect(() => {
|
|
const updateMemory = () => {
|
|
const usage = process.memoryUsage().rss;
|
|
setMemoryUsage(formatMemoryUsage(usage));
|
|
setMemoryUsageColor(
|
|
usage >= 2 * 1024 * 1024 * 1024 ? Colors.AccentRed : Colors.Gray,
|
|
);
|
|
};
|
|
const intervalId = setInterval(updateMemory, 2000);
|
|
updateMemory(); // Initial update
|
|
return () => clearInterval(intervalId);
|
|
}, []);
|
|
|
|
return (
|
|
<Box>
|
|
<Text color={Colors.Gray}>| </Text>
|
|
<Text color={memoryUsageColor}>{memoryUsage}</Text>
|
|
</Box>
|
|
);
|
|
};
|