48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package system
|
||
|
||
import (
|
||
"os"
|
||
"time"
|
||
)
|
||
|
||
// SetMOTD 设置 /etc/motd 文件内容
|
||
// 参数: content - MOTD 文本内容
|
||
// 返回: error - 写入文件错误
|
||
func SetMOTD(content string) error {
|
||
if content == "" {
|
||
// 如果内容为空，不清除现有 MOTD，避免误操作
|
||
return nil
|
||
}
|
||
|
||
// 添加时间和系统信息
|
||
finalContent := "========================================\n"
|
||
finalContent += "SunHPC 集群管理系统\n"
|
||
finalContent += "时间: " + time.Now().Format("2006-01-02 15:04:05") + "\n"
|
||
finalContent += "========================================\n\n"
|
||
finalContent += content
|
||
|
||
// 确保行尾有换行
|
||
if content[len(content)-1] != '\n' {
|
||
finalContent += "\n"
|
||
}
|
||
|
||
return os.WriteFile("/etc/motd", []byte(finalContent), 0644)
|
||
}
|
||
|
||
// ClearMOTD 清空 MOTD
|
||
func ClearMOTD() error {
|
||
return os.WriteFile("/etc/motd", []byte{}, 0644)
|
||
}
|
||
|
||
// AppendToMOTD 追加内容到 MOTD
|
||
func AppendToMOTD(additional string) error {
|
||
f, err := os.OpenFile("/etc/motd", os.O_APPEND|os.O_WRONLY, 0644)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer f.Close()
|
||
|
||
_, err = f.WriteString(additional + "\n")
|
||
return err
|
||
}
|