You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.4 KiB
91 lines
2.4 KiB
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"unicode"
|
|
)
|
|
|
|
func main() {
|
|
// Define the shift and hex flags
|
|
shift := flag.Int("s", 0, "Shift value for alphabet positions (can be negative)")
|
|
useHex := flag.Bool("x", false, "Encode numbers in hexadecimal")
|
|
flag.Parse()
|
|
|
|
// Check if input is being piped or passed as an argument
|
|
stat, _ := os.Stdin.Stat()
|
|
if (stat.Mode() & os.ModeCharDevice) != 0 {
|
|
// No input, display help text
|
|
flag.Usage()
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Read input from stdin
|
|
inputBytes, err := ioutil.ReadAll(os.Stdin)
|
|
if err != nil {
|
|
fmt.Println("Error reading input:", err)
|
|
os.Exit(1)
|
|
}
|
|
input := strings.TrimSpace(string(inputBytes))
|
|
|
|
// Convert input to uppercase and prepare the result slice
|
|
input = strings.ToUpper(input)
|
|
var result []string
|
|
var currentNumber string
|
|
|
|
for _, char := range input {
|
|
if unicode.IsLetter(char) {
|
|
// If there was a number being processed, add it to the result first
|
|
if currentNumber != "" {
|
|
appendNumber(&result, currentNumber, *useHex)
|
|
currentNumber = ""
|
|
}
|
|
// Get the position of the character in the alphabet (A = 1, B = 2, ..., Z = 26)
|
|
position := int(char - 'A' + 1)
|
|
// Apply the shift, wrapping around the alphabet
|
|
shiftedPosition := (position + *shift - 1) % 26
|
|
if shiftedPosition < 0 {
|
|
shiftedPosition += 26
|
|
}
|
|
shiftedPosition += 1
|
|
|
|
// Append the shifted position to the result slice
|
|
result = append(result, fmt.Sprintf("%02d", shiftedPosition))
|
|
} else if unicode.IsDigit(char) {
|
|
// Collect digits to process as a whole number
|
|
currentNumber += string(char)
|
|
} else {
|
|
// If there was a number being processed and now a non-digit/non-letter appears, append it
|
|
if currentNumber != "" {
|
|
appendNumber(&result, currentNumber, *useHex)
|
|
currentNumber = ""
|
|
}
|
|
}
|
|
}
|
|
|
|
// If there's any remaining number at the end of the input, append it
|
|
if currentNumber != "" {
|
|
appendNumber(&result, currentNumber, *useHex)
|
|
}
|
|
|
|
// Join the result slice into a space-separated string and print it
|
|
fmt.Println(strings.Join(result, " "))
|
|
}
|
|
|
|
// appendNumber appends a formatted number to the result slice
|
|
func appendNumber(result *[]string, number string, useHex bool) {
|
|
if useHex {
|
|
num, err := strconv.Atoi(number)
|
|
if err == nil {
|
|
*result = append(*result, fmt.Sprintf("0x%X", num))
|
|
}
|
|
} else {
|
|
*result = append(*result, "0d"+number)
|
|
}
|
|
}
|
|
|