mirror of https://github.com/zellyn/diskii.git
70 lines
2.1 KiB
Go
70 lines
2.1 KiB
Go
// Copyright © 2016 Zellyn Hunter <zellyn@gmail.com>
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/zellyn/diskii/disk"
|
|
"github.com/zellyn/diskii/helpers"
|
|
"github.com/zellyn/diskii/types"
|
|
)
|
|
|
|
// PutCmd is the kong `put` command.
|
|
type PutCmd struct {
|
|
Order types.DiskOrder `kong:"default='auto',enum='auto,do,po',help='Logical-to-physical sector order.'"`
|
|
System string `kong:"default='auto',enum='auto,dos3',help='DOS system used for image.'"`
|
|
FiletypeName string `kong:"default='B',help='Type of file (“diskii filetypes” to list).'"`
|
|
Overwrite bool `kong:"short='f',help='Overwrite existing file?'"`
|
|
Address uint16 `kong:"type='anybaseuint16',default='0x6000',help='For filetypes where it is appropriate, address to load the code at.'"`
|
|
|
|
DiskImage string `kong:"arg,required,type='existingfile',help='Disk image to modify.'"`
|
|
TargetFilename string `kong:"arg,required,help='Filename to use on disk.'"`
|
|
SourceFilename string `kong:"arg,required,type='existingfile',help='Name of file containing data to put.'"`
|
|
}
|
|
|
|
// Help displays extended help and examples.
|
|
func (p PutCmd) Help() string {
|
|
return `Examples:
|
|
# Put file gremlins.o onto disk image games.dsk, using the filename GREMLINS.
|
|
diskii put games.dsk GREMLINS gremlins.o`
|
|
}
|
|
|
|
// Run the `put` command.
|
|
func (p *PutCmd) Run(globals *types.Globals) error {
|
|
if p.DiskImage == "-" {
|
|
if p.SourceFilename == "-" {
|
|
return fmt.Errorf("cannot get both disk image and file contents from stdin")
|
|
}
|
|
}
|
|
op, order, err := disk.OpenFilename(p.DiskImage, p.Order, p.System, globals.DiskOperatorFactories, globals.Debug)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
contents, err := helpers.FileContentsOrStdIn(p.SourceFilename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
filetype, err := types.FiletypeForName(p.FiletypeName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fileInfo := types.FileInfo{
|
|
Descriptor: types.Descriptor{
|
|
Name: p.TargetFilename,
|
|
Length: len(contents),
|
|
Type: filetype,
|
|
},
|
|
Data: contents,
|
|
StartAddress: p.Address,
|
|
}
|
|
_, err = op.PutFile(fileInfo, p.Overwrite)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return disk.WriteBack(p.DiskImage, op, order, true)
|
|
}
|