This commit is contained in:
Ian Adam Naval 2014-09-16 20:52:57 -04:00
parent b226effff0
commit 1d57896123
6 changed files with 167 additions and 167 deletions

View File

@ -13,8 +13,9 @@ type JshOutput struct {
// Size prefixes as integers, using binary representation
type Unit int
const (
B Unit = 2 ^ 0
B Unit = 2 ^ 0
KB Unit = 2 ^ 10
MB Unit = 2 ^ 20
GB Unit = 2 ^ 30

16
free.go
View File

@ -1,8 +1,8 @@
/* File for free-related structs and methods */
package jsh
type MemStat struct {
Stat int
Units Unit
}
/* File for free-related structs and methods */
package jsh
type MemStat struct {
Stat int
Units Unit
}

View File

@ -1,93 +1,93 @@
package main
import (
"bufio"
"errors"
"flag"
"fmt"
"jsh"
"regexp"
"os"
"strconv"
)
func convertUnit(stringUnit string) (jsh.Unit, error) {
switch stringUnit {
case "B":
return jsh.B, nil
case "kB":
return jsh.KB, nil
case "mB":
return jsh.MB, nil
case "gB":
return jsh.GB, nil
case "tB":
return jsh.TB, nil
default:
return 0, errors.New(fmt.Sprintln("Unknown unit %s", stringUnit))
}
}
func parseLine(line string) (string, jsh.MemStat, error) {
// Recognizes a alphanumeric or () word, the : character, whitespace, the number we're looking for,
// more whitespace, and possibly the unit
lineRegex := regexp.MustCompile("(?P<key>^[\\w()]+):\\s+(?P<val>\\d+)(\\s+)?(?P<unit>\\w+)?")
// Parse the line, and construct a map of the valid matches
parsedLine := lineRegex.FindStringSubmatch(line)
names := lineRegex.SubexpNames()
matchedVals := map[string]string{}
for i, n := range parsedLine {
matchedVals[names[i]] = n
}
key := matchedVals["key"]
val, err := strconv.Atoi(matchedVals["val"])
if err != nil {
return "", jsh.MemStat{}, err
}
unit, err := convertUnit(matchedVals["unit"])
if err != nil {
return "", jsh.MemStat{}, err
}
return key, jsh.MemStat{val, unit}, nil
}
func parseMemInfo() jsh.JshOutput {
file, err := os.Open("/proc/meminfo")
if err != nil {
return jsh.JshOutput{[]string{}, []error{err}}
}
defer file.Close()
scanner := bufio.NewScanner(file)
memInfo := make(map[string]jsh.MemStat)
errors := []error{}
// Read in each line of the meminfo file, and place it in the map
for scanner.Scan() {
key, val, err := parseLine(scanner.Text())
if err != nil {
errors = append(errors, err)
}
memInfo[key] = val
}
finalOut := jsh.JshOutput{memInfo, errors}
return finalOut
}
func runJsonMode() {
output := parseMemInfo()
fmt.Println("%s", *output.ToJson())
}
func main() {
jsonModePtr := flag.Bool("json", false, "whether to use json mode for output")
flag.Parse()
if !*jsonModePtr {
fmt.Printf("%s", jsh.Fallback("/usr/bin/free"))
} else {
runJsonMode()
}
}
package main
import (
"bufio"
"errors"
"flag"
"fmt"
"jsh"
"os"
"regexp"
"strconv"
)
func convertUnit(stringUnit string) (jsh.Unit, error) {
switch stringUnit {
case "B":
return jsh.B, nil
case "kB":
return jsh.KB, nil
case "mB":
return jsh.MB, nil
case "gB":
return jsh.GB, nil
case "tB":
return jsh.TB, nil
default:
return 0, errors.New(fmt.Sprintln("Unknown unit %s", stringUnit))
}
}
func parseLine(line string) (string, jsh.MemStat, error) {
// Recognizes a alphanumeric or () word, the : character, whitespace, the number we're looking for,
// more whitespace, and possibly the unit
lineRegex := regexp.MustCompile("(?P<key>^[\\w()]+):\\s+(?P<val>\\d+)(\\s+)?(?P<unit>\\w+)?")
// Parse the line, and construct a map of the valid matches
parsedLine := lineRegex.FindStringSubmatch(line)
names := lineRegex.SubexpNames()
matchedVals := map[string]string{}
for i, n := range parsedLine {
matchedVals[names[i]] = n
}
key := matchedVals["key"]
val, err := strconv.Atoi(matchedVals["val"])
if err != nil {
return "", jsh.MemStat{}, err
}
unit, err := convertUnit(matchedVals["unit"])
if err != nil {
return "", jsh.MemStat{}, err
}
return key, jsh.MemStat{val, unit}, nil
}
func parseMemInfo() jsh.JshOutput {
file, err := os.Open("/proc/meminfo")
if err != nil {
return jsh.JshOutput{[]string{}, []error{err}}
}
defer file.Close()
scanner := bufio.NewScanner(file)
memInfo := make(map[string]jsh.MemStat)
errors := []error{}
// Read in each line of the meminfo file, and place it in the map
for scanner.Scan() {
key, val, err := parseLine(scanner.Text())
if err != nil {
errors = append(errors, err)
}
memInfo[key] = val
}
finalOut := jsh.JshOutput{memInfo, errors}
return finalOut
}
func runJsonMode() {
output := parseMemInfo()
fmt.Println("%s", *output.ToJson())
}
func main() {
jsonModePtr := flag.Bool("json", false, "whether to use json mode for output")
flag.Parse()
if !*jsonModePtr {
fmt.Printf("%s", jsh.Fallback("/usr/bin/free"))
} else {
runJsonMode()
}
}

View File

@ -1,76 +1,75 @@
package main
import (
"io/ioutil"
"fmt"
"flag"
"log"
"syscall"
"strconv"
"flag"
"fmt"
"io/ioutil"
"log"
"strconv"
"syscall"
)
func get_fileinfo(f string, size bool, mode bool, inode bool) string{
var stat syscall.Stat_t
var ret string
ret = "{\"name\":\"" + f + "\""
if err := syscall.Stat(f, &stat); err != nil {
log.Fatal(err)
}
if(size){
ret = ret + ", \"size\":" + strconv.FormatInt(stat.Size,10)
}
if(mode){
ret = ret + ", \"mode\":" + strconv.Itoa(int(stat.Mode))
}
if(inode){
ret = ret + ", \"inode\":" + strconv.FormatUint(stat.Ino,10)
}
ret = ret + "}"
return ret
func get_fileinfo(f string, size bool, mode bool, inode bool) string {
var stat syscall.Stat_t
var ret string
ret = "{\"name\":\"" + f + "\""
if err := syscall.Stat(f, &stat); err != nil {
log.Fatal(err)
}
if size {
ret = ret + ", \"size\":" + strconv.FormatInt(stat.Size, 10)
}
if mode {
ret = ret + ", \"mode\":" + strconv.Itoa(int(stat.Mode))
}
if inode {
ret = ret + ", \"inode\":" + strconv.FormatUint(stat.Ino, 10)
}
ret = ret + "}"
return ret
}
func main() {
// here be the ls flags
// here be the ls flags
var a_flag bool // all files, even ones starting with .
var mode_flag bool // flags
var inode_flag bool // inode
var size_flag bool // size
var first = true
var a_flag bool // all files, even ones starting with .
var mode_flag bool // flags
var inode_flag bool // inode
var size_flag bool // size
var first = true
flag.BoolVar(&a_flag, "a", false, "lists all files in directory, even hidden ones")
flag.BoolVar(&mode_flag, "f", false, "include flags for file")
flag.BoolVar(&inode_flag, "i", false, "include flags for file")
flag.BoolVar(&size_flag, "s", false, "include flags for file")
// end ls flag
flag.BoolVar(&a_flag, "a", false, "lists all files in directory, even hidden ones")
flag.BoolVar(&mode_flag, "f", false, "include flags for file")
flag.BoolVar(&inode_flag, "i", false, "include flags for file")
flag.BoolVar(&size_flag, "s", false, "include flags for file")
// end ls flag
flag.Parse()
root := "."//flag.Arg(0)
dir,_ := ioutil.ReadDir(root)
flag.Parse()
root := "." //flag.Arg(0)
dir, _ := ioutil.ReadDir(root)
fmt.Printf("[\n")
if(!a_flag){
for _,entry := range dir{
if(entry.Name()[0]!='.'){
if(!first){
fmt.Printf(",")
}else{
first = false
fmt.Printf("[\n")
if !a_flag {
for _, entry := range dir {
if entry.Name()[0] != '.' {
if !first {
fmt.Printf(",")
} else {
first = false
}
fmt.Printf("%s\n", get_fileinfo(entry.Name(), size_flag, mode_flag, inode_flag))
}
}
} else {
for _, entry := range dir {
if !first {
fmt.Printf(",")
} else {
first = false
}
fmt.Printf("%s\n", get_fileinfo(entry.Name(), size_flag, mode_flag, inode_flag))
}
}
fmt.Printf("%s\n", get_fileinfo(entry.Name(), size_flag, mode_flag, inode_flag))
}
}
}else{
for _,entry := range dir{
if(!first){
fmt.Printf(",")
}else{
first = false
}
fmt.Printf("%s\n", get_fileinfo(entry.Name(), size_flag, mode_flag, inode_flag))
}
}
fmt.Printf("]\n")
fmt.Printf("]\n")
}

View File

@ -25,7 +25,7 @@ func PsOutputToProcesses(out string) *[]jsh.Process {
func runJsonMode() {
// Run procps-ng "ps" with full output
psOut := string(*jsh.FallbackWithArgs("/usr/bin/ps", []string{"auxww"}))
psOut := string(*jsh.FallbackWithArgs("/usr/bin/ps", []string{"auxww"}))
processesPtr := PsOutputToProcesses(psOut)
finalOut := jsh.JshOutput{*processesPtr, []string{}}

View File

@ -2,9 +2,9 @@ package jsh
import (
"math"
"unicode"
"os"
"os/exec"
"unicode"
)
// FieldsN slices s into substrings after each instance of a whitespace
@ -56,7 +56,7 @@ func FallbackWithArgs(program string, args []string) *[]byte {
if err != nil {
panic(err)
}
return &out
return &out
}
func Fallback(program string) *[]byte {